引入setUpClass()、tearDownClass()测试固件,保证运行所有测试用例时浏览器只需要打开和关闭一次,可提高测试效率。
代码如下:
from selenium import webdriver
import unittest
class Test_Web_UI(unittest.TestCase):
@classmethod
def setUpClass(cls):
cls.driver = webdriver.Chrome()
@classmethod
def tearDownClass(cls):
cls.driver.quit()
# 测试用例1
def test_login(self):
self.driver.get("https://mail.qq.com/cgi-bin/loginpage")
print("网页一标题:", self.driver.title)
if self.assertEqual(self.driver.title, "登录QQ邮箱"):
print("用例通过")
else:
print("用例失败")
# self.assertEqual(self.driver.title, "登录QQ邮箱")
# 测试用例2
def test_movie(self):
self.driver.get("https://maoyan.com/")
print("网页二标题:", self.driver.title)
self.assertEqual(self.driver.title, "电影院票房购票_评分_选座_经典影视推荐-猫眼电影")
if __name__ == "__main__":
unittest.main()