# -*- coding: utf-8 -*-

 import unittest

 import time

 from selenium import webdriver

 import xlrd,xlwt

 import HTMLTestRunner

 from openpyxl import load_workbook

 class mylogintest(unittest.TestCase):#定义一个mylogintest类继承unittest.TestCase类

     '''打开百度测试'''

     # @classmethod

     # def tearDownClass(cls):#用例执行后的初始化工作

     #      print('结束测试')

     def setUp(self):#用例执行前的初始化工作

         print("开始测试")

         self.driver = webdriver.Firefox()

     def tearDown(self):#用例执行后的初始化工作

         print("结束测试")

         self.driver.quit()

     def testopenBD(self):

         '''打开百度'''

         book = xlrd.open_workbook("mylogintest.xlsx")  # 新建文件对象

         sheet = book.sheet_by_name("Sheet1")  # 获取文件对象中sheet名为login的sheet对象

         #     # # url1 = sheet.cell(1, 0).value 获取第2行第1列的值

         #     # # url2 = sheet.cell(1, 1).value获取第2行第2列的值

         nrows = sheet.nrows  # 获取总行数

         urllist = []

         for i in range(1, nrows):

             print(i)

             nrowsvalue = sheet.row_values(i, 0)#获取第i+1行 第1列的值

             urllist.append(nrowsvalue)  # 这里添加元素不能用list = list.append(nrowsvalue);否则会出现nonetype的错误；

             print(urllist)

         driver = self.driver

         for i in range(len(urllist)):  # 获取list长度;range表示默认0开始的整数迭代

             print(urllist[i])

             driver.get(''.join(urllist[i]))#将list类型转换成字符串类型

             print(driver.title)

             self.assertEqual(driver.title,'百度一下，你就知道','失败')#添加断言；当driver.title的值为‘百度一下，你就知道’时，断言通过，否则不通过，程序将会终断

             time.sleep(3)

     # @classmethod

     # def setUpClass(cls):##用例执行前的初始化工作

     #     print('开始测试')

 if __name__=='__main__':

     filename = './'+'mylogintestresult.html'#定义文件名与路径

     fp = open(filename,'wb')#生成文件

     ut = unittest.TestSuite()  # 创建测试套件；TestSuite不加括号的结果：addTest() missing 1 required positional argument: 'test'

     ut.addTest(mylogintest('testopenBD'))#将要执行的用例加入测试套件中

     runner = HTMLTestRunner.HTMLTestRunner(stream=fp,title='分别打开百度51网站',description='百度51') #如果引用的是from HTMLTestRunner import HTMLTestRunner,这里就不需要写两个HTMLTestRunner，一个就够了

     runner.run(ut)#运行测试套件

     fp.close()