from selenium import webdriver from selenium.webdriver.support.ui import WebDriverWait from selenium.webdriver.support import expected_conditions as EC from selenium.webdriver.common.by import By import time import unittest import ddt #单元测试框架unittest中进行数据驱动
path = r’D:\webdriver\chromedriver.exe’
@ddt.ddt class LoginTest(unittest.TestCase): “”“定义一个测试类LoginTest，继承unittest.TestCase”""
def setUp(self): """初始化操作""" self.wd = webdriver.Chrome(executable_path=path) self.wd.set_page_load_timeout(5) self.wd.get('http://192.168.1.251:9012/login') self.wd.implicitly_wait(5) time.sleep(3) def sub_login(self, usr, pwd): """登录的过程的封装""" ele_user = self.wd.find_element_by_id('userName') ele_user.clear() ele_user.send_keys(usr) ele_pwd = self.wd.find_element_by_id('passWord') ele_pwd.clear() ele_pwd.send_keys(pwd) ele_btn = self.wd.find_element_by_xpath('/html/body/form/div[4]/div/button') ele_btn.click() def check(self, word): """用来比较指定的字符串是否在页面源代码中""" # assert True == word in self.wd.page_source self.assertTrue(True, word in self.wd.page_source) def test_login_success(self): """测试登录成功时的情况""" self.sub_login('qwen', 'q123456') self.check('实训项目') @ddt.data(['qwen1', '1'], ['qwen2', '1'], ['qwen2', '1']) def test_login_fail(self, p): """登录失败时的情况""" self.sub_login(p[0], p[1]) self.check('密码') def tearDown(self): """测试结束时去执行""" self.wd.quit()if name == “main”: unittest.main()
