# coding:utf-8
from selenium import webdriver
import unittest
class BaiduLinks(unittest.TestCase):
 def setUp(self):
  base_url = 'https://www.baidu.com'
  self.driver = webdriver.Chrome()
  self.driver.implicitly_wait(10)
  self.driver.get(base_url)
 def tearDown(self):
  self.driver.close()
  self.driver.quit()
 def test_baidu_news(self):
  u"""百度新聞"""
  driver = self.driver
  driver.find_element_by_link_text('新聞').click()
  self.assertIn(driver.title, u'v百度新聞——全球最大的中文新聞平臺')
 def test_baidu_hao123(self):
  u"""hao123"""
  driver = self.driver
  driver.find_element_by_link_text('hao123').click()
  self.assertEqual(driver.title, u'hao123_上網從這里開始')
 def test_baidu_tieba(self):
  u"""百度貼吧"""
  driver = self.driver
  driver.find_element_by_link_text('貼吧').click()
  # 錯誤的斷言
  self.assertTrue(driver.find_element_by_link_text('全吧搜索+1'))
if __name__ == '__main__':
 unittest.main()
 print('百度鏈接跳轉成功')

# /usr/bin/env python3
# coding:utf-8
"""
Created on 2018-05-30
Project: learning
@Author:wuya
"""
import os, time, unittest
import HTMLTestRunner
report_path = os.getcwd() # 設置保存報告的路徑，這兒設置的是與執行文件在同一個目錄下
now = time.strftime("%Y-%m-%d %H：%M", time.localtime(time.time())) # 獲取當前時間
title = u"百度頭鏈接測試" # 標題
report_abspath = os.path.join(report_path, title + now + ".html") # 設置報告存放和命名
# 導入用例
def all_case():
 case_path = os.getcwd() # 用例路徑，這兒的用例和執行文件在同一目錄下
 discover = unittest.defaultTestLoader.discover(case_path,             pattern="Test*.py") # 添加用例，在case_path的路徑下，所有以Test開頭的文件都當做用例文件執行
 print(discover)
 return discover
if __name__ == "__main__":
 fp = open(report_abspath, "wb") # 保存報告文件
 runner = HTMLTestRunner.HTMLTestRunner(stream=fp,           title=title + '：',)
 runner.run(all_case()) # 執行用例
 fp.close()