from selenium import webdriver
import HTMLTestRunner,unittest,time,os

class BaiduTest(unittest.TestCase):
    def setUp(self): #测试准备工作
        self.driver = webdriver.Chrome() #加载驱动
        self.driver.implicitly_wait(30) #隐式等待
        self.base_url = "http://www.baidu.com" #url

    def test_baidu(self): #测试用例
        driver = self.driver
        print("========【case_0001】 百度搜索=============")
        driver.get(self.base_url + "/") #get发送url
        driver.find_element_by_id("kw").clear() #清空
        driver.find_element_by_id("kw").send_keys("shuzf") #搜索shuzf
        driver.find_element_by_id("su").click() #点击
        time.sleep(5)#休息5s
        self.assertEqual("shuzf_百度搜索", driver.title)  # 断言判断是否相等
        now = time.strftime("%Y-%m-%d-%H_%M_%S", time.localtime(time.time()))
        if not os.path.exists('result/image/'):  os.makedirs('result/image/') #判断当前路径是否存在，没有则创建文件夹
        pic_path = 'result/image/' + now + '.png'
        driver.save_screenshot(pic_path) #生成图片

    def tearDown(self): #释放资源
        self.driver.quit() #退出

if __name__ == "__main__":
    testunit = unittest.TestSuite() #构造测试套件
    testunit.addTest(BaiduTest("test_baidu")) #添加测试用例

    #HtmlFile = "c:/"+now+"run.html"
    now = time.strftime("%Y-%m-%d-%H_%M_%S", time.localtime(time.time()))
    if not os.path.exists('result/'):  os.makedirs('result/') #判断当前路径是否存在，没有则创建文件夹
    HtmlFile = "result/" + now + ".html"
    fp = open(HtmlFile, "wb") #生成报告
    runner = HTMLTestRunner.HTMLTestRunner(stream=fp, title="百度测试报告", description="用例测试情况")
    runner.run(testunit) #执行套件
    fp.close() #关闭


#注意点1、浏览器驱动的安装 2，文件夹的生成

2，使用命令行执行文件，而不是pycharm

　　>python demo.py

优质内容筛选与推荐>>
1、sqlite and 安装sqlite
2、centos开机无法启动eth0
3、将win7笔记本电脑变身WiFi热点，让手机、ipad共享上网!
4、ViewAnimator 当进行视图切换时显示动画效果
5、jQuery学习之DOM操作