1.创建unitWebtours.py

将unitWebtours.py放在test文件夹下

unitWebtours.py

代码实现

from selenium import webdriver
from selenium.webdriver.support.select import Select
from time import sleep
import unittest

driver = None

class Webtours(unittest.TestCase):
    @classmethod
    def setUpClass(cls):
        global driver
        driver = webdriver.Chrome()
        
    @classmethod
    def tearDownClass(cls):
        driver.quit()
    
    def test1OpenWeb(self):
        u'''打开网站'''
        driver.get("http://127.0.0.1:1080/WebTours/")
        # 等待3秒
        sleep(3)

        driver.switch_to.default_content()
        driver.switch_to_frame("body")
        driver.switch_to_frame("info")
        actual = driver.find_element_by_tag_name("body").text
        # expect = 'Welcome to the Web Tours site.'
        # # 精确匹配用in 否则用in 
        # if expect in actual:
        #     print("打开网站：成功")
        # else:
        #     print("打开网站：失败") 
    
    def test2Loginin(self):
        u'''登录'''
        driver.switch_to.default_content()
        driver.switch_to_frame("body")
        driver.switch_to_frame("navbar")

        # 登录
        yhm = 'admin'
        mm = '123456'
        # 输入用户名
        driver.find_element_by_name("username").send_keys(yhm)
        # 输入密码
        driver.find_element_by_name("password").send_keys(mm)
        # 点击登录
        driver.find_element_by_name("login").click()
        sleep(3)
        # 切换框架
        driver.switch_to.default_content()
        driver.switch_to_frame("body")
        driver.switch_to_frame("info")

        expect = "Welcome, "+yhm
        actual = driver.find_element_by_tag_name("body").text
        # # 精确匹配用in 否则用in 
        # if expect in actual:
        #     print("登录：成功")
        # else:
        #     print("登录：失败")
    
    def test3Search(self):
        u'''搜票'''
        driver.switch_to.default_content()
        driver.switch_to_frame("body")
        driver.switch_to_frame("navbar")
        driver.find_element_by_xpath('//img[@alt="Search Flights Button"]').click()
        sleep(3)
        
    def test4Book(self):
        u'''订票'''
        # 预定票
        driver.switch_to.default_content()
        driver.switch_to_frame("body")
        driver.switch_to_frame("info")
        Select(driver.find_element_by_name("depart")).select_by_visible_text="London"
        # select_by_value 值是看不见的
        Select(driver.find_element_by_name("arrive")).select_by_visible_text="Paris"
        driver.find_element_by_name("findFlights").click()
        sleep(1)
        
        # 查看预定的航班信息
        driver.find_elements_by_name("outboundFlight")[0].click()
        driver.find_element_by_name("reserveFlights").click()
        sleep(1)
        
        # 支付信息
        driver.find_element_by_name("buyFlights").click()
        sleep(1)


        # 切换框架

        expect = "Flight 240 leaves London for Paris"
        actual = driver.find_element_by_tag_name("body").text
        # 精确匹配用in 否则用in 
        # if expect in actual:
        #     print("订票：成功")
        # else:
        #     print("订票：失败")
            
        #预定下一个
        driver.find_element_by_name("Book Another").click()
        sleep(1)
    
    def test5Logoff(self):
        u'''注销'''
        # 切换框架
        driver.switch_to.default_content()
        driver.switch_to_frame("body")
        driver.switch_to_frame("navbar")
        
        driver.find_element_by_xpath('//img[@alt="SignOff Button"]').click()
        sleep(1)


        # 切换框架
        driver.switch_to.default_content()
        driver.switch_to_frame("body")
        driver.switch_to_frame("info")

        expect = "sign up now"
        actual = driver.find_element_by_tag_name("body").text
        # 精确匹配用in 否则用in 
        # if expect in actual:
        #     print("注销：成功")
        # else:
        #     print("注销：失败")

if __name__=="__main__":
    # verbosity=2 显示详细信息，可以省略，测试信息不详细
    unittest.main(verbosity=2)

2.使用测试套件运行测试函数

代码实现

import unittest

suite = unittest.TestSuite()
tests = unittest.defaultTestLoader.discover('./test',pattern='unit*.py')
suite.addTests(tests)
runner = unittest.TextTestRunner(verbosity=2)
runner.run(suite)