1.编写测试脚本:
1. Test Fixture
def setUp(self):
self.driver = webdriver.Firefox()
self.url = "https://www.baidu.com/"
self.driver.maximize_window()
time.sleep(3)
def tearDown(self):
self.driver.quit()
2.Test Case
class Baidu1(unittest.TestCase):
def setUp(self):
self.driver = webdriver.Firefox()
self.url = "https://www.baidu.com/"
self.driver.maximize_window()
time.sleep(3)
def tearDown(self):
self.driver.quit()
def test_hao(self):
driver = self.driver
url = self.url
driver.get(url)
driver.find_element_by_link_text("hao123").click()
time.sleep(6)
def test_baidu(self):
driver = self.driver
url = self.url
driver.get(url)
driver.find_element_by_id("kw").send_keys("英语")
driver.find_element_by_id("su").submit()
time.sleep(5)
print(driver.title)
try:
self.assertEqual(driver.title, "百度一下_百度搜索", msg="不相等")
except:
self.saveScreenAsPhoto(driver, 'baidu.png')
time.sleep(6)
def saveScreenAsPhoto(self, driver, file_name):
if not os.path.exists("../image"):
os.makedirs("../image")
now = time.strftime("%Y%m%d-%H%M%S", time.localtime(time.time()))
driver.get_screenshot_as_file("./image/" + now + "-" + file_name)
time.sleep(3)
if __name__ == '__main__':
unittest.main()
3.unittest断言
self.assertEqual(driver.title, "百度一下_百度搜索"