selenium+python笔记9

#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
@desc:        delete mail
我们多添加一些测试场景，比如：删除邮件，查找邮件，发送邮件等等
"""
import unittest
import os
import sys
from selenium import webdriver

cur_dir = os.getcwd()
sys.path.append(cur_dir.split(r'\test_case')[0])

from public import login

# from selenium.webdriver.common.keys import Keys
import time

class TestDel(unittest.TestCase):
    def setUp(self):
        self.driver = webdriver.Firefox()
        self.driver.implicitly_wait(30)
        self.base_url = "http://www.126.com/"
        self.verificationErrors = []
        self.accept_next_alert = True

    # 搜索邮件
    def test_del_mail(self):
        driver = self.driver
        driver.get(self.base_url)
        # 调用登录模块
        login.login(self, 'xxxx', 'xxxx')
        # 打开收件箱
        driver.find_element_by_class_name('nui-tree-item-text').click()
        time.sleep(2)
        driver.find_elements_by_xpath("//span[@class='nui-chk-symbol']/b").pop(1).click()
        try:
            spans = driver.find_elements_by_tag_name('span')
            for s in spans:
                if s.text == u'删 除':
                    s.click()
        except:
            pass
        # 断言是否已删除
        text = driver.find_element_by_css_selector("span.nui-tips-text>a").text
        self.assertEqual(text, u'已删除')
        # 退出
        login.logout(self)

    def tearDown(self):
        self.driver.quit()
        self.assertEqual([], self.verificationErrors)

if __name__ == "__main__":
    unittest.main()