Selenium Python自动化测试学习知识积累-unittest

#encoding=utf-8
import unittest
from selenium import webdriver
import time

class GloryRoad(unittest.TestCase):
    def setUp(self):
        #启动firefox浏览器
        self.driver =webdriver.Firefox(executable_path="F:\\Python\\Webdriver\\geckodriver.exe")

    def testSoGo(self):
        #访问搜狗首页
        self.driver.get("http://www.sogou.com")
        #清空搜索输入框默认内容
        self.driver.find_element_by_id("query").clear()
        #在搜索输入框中输入“Python自动化测试”
        self.driver.find_element_by_id("query").send_keys("webdriver 实战宝典")
        #点击搜索按钮
        self.driver.find_element_by_id("stb").click()
        #等待3秒
        time.sleep(3)
        assert "吴晓华" in self.driver.page_source,"页面不存在要寻找的关键字!"

    def tearDown(self):
        #退出浏览器
        self.driver.quit()

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