结合单元测试框架

coding:utf-8

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()