Unittest framework

import unittest
import time
from selenium import webdriver
import os


class test_baidu1(unittest.TestCase):
    def setUp(self):
        #  Get drive path 
        current_path = os.path.dirname(__file__)  #  Get the current path 
        driver_path = os.path.join(current_path, '../../../driver/chromedriver.exe')  #  Put the current path +chromedriver Path connection 

        self.driver = webdriver.Chrome(executable_path=driver_path)
        self.driver.get('https://www.baidu.com')
        self.driver.maximize_window()

    def tearDown(self):
        time.sleep(1)
        self.driver.quit()

    def test_cat(self):
        self.driver.find_element_by_name('wd').send_keys(' Panda ')
        self.driver.find_element_by_id('su').click()
        time.sleep(1)
        self.assertEqual(self.driver.title, " Panda _ Baidu search ")

    def test_dog(self):
        self.driver.find_element_by_name('wd').send_keys(' Dog ')
        self.driver.find_element_by_id('su').click()
        time.sleep(1)
        self.assertEqual(self.driver.title, " Dog _ Baidu search ")

    def test_pig(self):
        self.driver.find_element_by_name('wd').send_keys(' The pig ')
        self.driver.find_element_by_id('su').click()
        time.sleep(1)
        self.assertEqual(self.driver.title, " The pig _ Baidu search ")


def suite():
    suite = unittest.TestSuite()  #  Create a suite object 
    suite.addTest(test_baidu1('test_cat'))  #  Class name （ Method name ）
    suite.addTest(test_baidu1('test_dog'))  #  Class name （ Method name ）
    suite.addTest(test_baidu1('test_pig'))  #  Class name （ Method name ）
    return suite


if __name__ == '__main__':
    unittest.main(defaultTest='suite')    #  Only the methods in the suite will be executed 

import unittest
import time
from selenium import webdriver
import os


class test_baidu2(unittest.TestCase):
    def setUp(self):
        #  Get drive path 
        current_path = os.path.dirname(__file__)  #  Get the current path 
        driver_path = os.path.join(current_path, '../../../driver/chromedriver.exe')  #  Put the current path +chromedriver Path connection 

        self.driver = webdriver.Chrome(executable_path=driver_path)
        self.driver.get('https://www.baidu.com')
        self.driver.maximize_window()

    def tearDown(self):
        time.sleep(1)
        self.driver.quit()

    def test_cat(self):
        self.driver.find_element_by_name('wd').send_keys(' Worm ')
        self.driver.find_element_by_id('su').click()
        time.sleep(1)
        self.assertEqual(self.driver.title, " Worm _ Baidu search ")

    def test_dog(self):
        self.driver.find_element_by_name('wd').send_keys(' bear ')
        self.driver.find_element_by_id('su').click()
        time.sleep(1)
        self.assertEqual(self.driver.title, " bear _ Baidu search ")

    def test_pig(self):
        self.driver.find_element_by_name('wd').send_keys(' Tujia pig ')
        self.driver.find_element_by_id('su').click()
        time.sleep(1)
        self.assertEqual(self.driver.title, " Tujia pig _ Baidu search ")


def suite():
    suite = unittest.TestSuite()  #  Create a suite object 
    suite.addTest(test_baidu2('test_cat'))  #  Class name （ Method name ）
    suite.addTest(test_baidu2('test_dog'))  #  Class name （ Method name ）
    suite.addTest(test_baidu2('test_pig'))  #  Class name （ Method name ）
    return suite


if __name__ == '__main__':
    unittest.main(defaultTest='suite')    #  Only the methods in the suite will be executed 

#  Operation main entrance , perform test_case  Remove all modules 
import os

#  Get all the file names in the directory 
current_path = os.path.dirname(__file__)  #  Get local path 
case_path = os.path.join(current_path, 'unittest_case')  #  Local path into  unittest_case  file 
file_name = os.listdir(case_path)  #  lookup unitest_case  Next .py  The file of  ['dome01.py', 'dome02.py', '__init__.py']
print(file_name)
#  To perform all  .py Final document 
for file in file_name:  #  Traverse  file_name  file 
    f = file.split('.')[-1]  #  cutting   ['dome01', 'py']  ['dome02', 'py'] ['__init__', 'py']

    if f == 'py':
        os.system('python ' + case_path+'/'+file+'  1>>log.txt 2>&1')

import os, unittest, datetime
import HTMLTestRunner

current_path = os.path.dirname(__file__)   #  Get the current address 
case_path = os.path.join(current_path,'unittest_case')

#  Read the test methods in all modules under the directory 【 This method must use test The first method name 】
discover = unittest.defaultTestLoader.discover(start_dir=case_path,
                                               pattern='dome*.py',
                                               top_level_dir=None)
# print(discover)

#  Create a master suite 
main_suite = unittest.TestSuite()
main_suite.addTests(discover)

# now = datetime.now()
# report_file = current_path + "/test_report/time_report_{}.html".format(now)
report_file = current_path+ "/test_report/report.html"
print(report_file)
report_ojb = open(report_file, 'wb')
runner = HTMLTestRunner.HTMLTestRunner(stream=report_ojb,
                                       title=" Baidu search test report ",
                                       description=" Use case execution ")
#  Execute the entire test suite 
runner.run(main_suite)
# if __name__ == '__main__':
#   unittest.main(defaultTest='main_suite')