八.自动化测试模型-模块化驱动

文件1

baidu_search.py

from selenium import webdriver
from selenium.webdriver.common.by import By
from time import sleep
from webdriver_manager.chrome import ChromeDriverManager

class BaiduSearch():
    def __init__(self):
        self.driver = webdriver.Chrome(ChromeDriverManager().install())
    # 定义一个方法用来打开网页
    def get_url(self):
        print('开始打开百度')
        self.driver.get("https://baidu.com")
        sleep(5)

    # 定义一个方法封装搜索功能
    def search_keys(self):
        #操作
        print('开始搜索关键词神话')
        self.driver.find_element(By.ID,"kw").send_keys("神话")
        self.driver.find_element(By.ID,"su").click()
        sleep(5)

 文件2     

test_search.py

#定义测试类
import unittest
from baidu_search import BaiduSearch

class TestSearch(unittest.TestCase):
    # 定义类级别的夹具
    @classmethod
    #定义前置后置条件
    def setUpClass(clq):
        #生成实例对象
        clq.bs = BaiduSearch()
        print("开始执行类级别前置条件")
    # 定义类级别的夹具
    @classmethod
    def tearDownClass(clq):
        clq.bs.driver.quit()

    #定义用例
    def test_search01(self):
        #用来验证可以打开百度页面
        self.bs.get_url()
    #定义用例
    def test_search02(self):
        #用来验证可以打开百度页面
        self.bs.search_keys()
        print("开始执行类级别后置条件")

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