1、unittest框架
2、路径自动读取
3、截图命名灵活
4、智能显示等待
from selenium import webdriver
import time
import unittest
import datetime,os
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
class TestBingSo(unittest.TestCase):
@classmethod
def setUpClass(cls):
print("这是初始化开始测试")
parent_abspath = os.path.abspath(os.path.dirname(os.getcwd()))
driver_path =os.path.join(parent_abspath,"driver/chromedriver")
cls.driver = webdriver.Chrome(executable_path=driver_path)
cls.driver.get("https://cn.bing.com/?ensearch=1&FORM=BEHPTB")
cls.driver.maximize_window()
assert 'Bing' in cls.driver.title
@classmethod
def tearDownClass(cls):
print("执行完成环境销毁")
cls.driver.quit()
def test_soso(self):
self.driver.find_element_by_id("sb_form_q").send_keys("python")
self.driver.find_element_by_id("sb_form_go").click()
self.driver.get_screenshot_as_file(str(datetime.datetime.now())+"_soso.png")
wait = WebDriverWait(self.driver,10)
wait.until(EC.title_contains("python"),'页面没加载上')
assert 'python' in self.driver.page_source
if __name__ == '__main__':
unittest.main()
网友评论