unittest-单元测试

2019-03-27

selenium基础使用

from selenium import webdriver

driver = webdriver.Firefox()
driver.implicitly_wait(30)
driver.maximize_window()

driver.get("https://www.baidu.com/")

search_field = driver.find_element_by_id('kw')
search_field.clear()

search_field.send_keys('wust')
search_field.submit()
# search_field.send_keys(Keys.RETURN)

products = driver.find_elements_by_xpath("//h3[@class='t']/a")

print('Found ' + str(len(products)) + ' products:')

for product in products:
    print(product.text)

driver.quit()

使用unitest编写单元测试

unittest（PyUnit）为任何项目创建全面的测试套件
- Test Fixture（测试夹具）：通过使用测试夹具，可以定义在单个或多个测试执行之前的准备工作和测试执行后的清理工作
- Test Case（测试用例）
- Test Suite（测试套件）
- Test Runner（测试执行器）
- Test Report（测试报告）

# searchtests.py
import unittest
from selenium import webdriver


class SearchTests(unittest.TestCase):
    def setUp(self):
        self.driver = webdriver.Firefox()
        self.driver.implicitly_wait(30)
        self.driver.maximize_window()

        self.driver.get('https://www.baidu.com')

    def test_search_by_category(self):
        self.search_field = self.driver.find_element_by_id('kw')
        self.search_field.clear()

        self.search_field.send_keys('WUST')
        self.search_field.submit()

        products = self.driver.find_elements_by_xpath("//h3[@class='t']/a")
        self.assertEqual(7, len(products))

    def test_search_by_name(self):
        self.search_field = self.driver.find_element_by_id('kw')
        self.search_field.clear()

        self.search_field.send_keys('HUST')
        self.search_field.submit()

        products = self.driver.find_elements_by_xpath("//h3[@class='t']/a")
        self.assertEqual(8, len(products))

    def tearDown(self):
        self.driver.quit()


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

# searchtests_with_class_method.py
import unittest
from selenium import webdriver


class SearchTests(unittest.TestCase):
    @classmethod
    def setUpClass(cls):
        cls.driver = webdriver.Firefox()
        cls.driver.implicitly_wait(30)
        cls.driver.maximize_window()

        cls.driver.get('https://www.baidu.com')
        cls.driver.title

    def test_search_by_category(self):
        self.search_field = self.driver.find_element_by_id('kw')
        self.search_field.clear()

        self.search_field.send_keys('xxx')
        self.search_field.submit()

        products = self.driver.find_elements_by_xpath("//h3[@class='t']/a")
        self.assertEqual(3, len(products))

    @classmethod
    def tearDownClass(cls):
        cls.driver.quit()


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

# homepagetests.py
# Test Suite测试套件，将不同的测试组成一个逻辑组，然后设置统一的测试套件，并通过一个命令来执行测试。

import unittest
from selenium import webdriver
from selenium.common.exceptions import NoSuchElementException
from selenium.webdriver.common.by import By
from builtins import classmethod


class HomePageTest(unittest.TestCase):
    @classmethod
    def setUp(cls):
        cls.driver = webdriver.Firefox()
        cls.driver.implicitly_wait(30)
        cls.driver.maximize_window()

        cls.driver.get('http://demo-store.seleniumacademy.com/')

    def test_search_field(self):
        # check search field exists on Home page
        self.assertTrue(self.is_element_present(By.NAME, 'q'))

    def test_language_option(self):
        # check language options dropdown on Home page
        self.assertTrue(self.is_element_present(By.ID, 'select-language'))

    def test_shopping_cart_empty_message(self):
        # check content of My Shopping Cart block on Home page
        shopping_cart_icon = self.driver.find_element_by_css_selector('div.header-minicart span.icon')
        shopping_cart_icon.click()

        shopping_cart_status = self.driver.find_element_by_css_selector('p.empty').text
        self.assertEqual('You have no items in your shopping cart', shopping_cart_status)

        close_button = self.driver.find_element_by_css_selector('div.minicart-wrapper a.close')
        close_button.click()

    @classmethod
    def tearDown(cls):
        # close the browser window
        cls.driver.quit()

    def is_element_present(self, how, what):
        """
        Utility method to check presence of an element on page
        :params how: By locator type
        :params what: locator value
        """
        try:
            self.driver.find_element(by=how, value=what)
        except NoSuchElementException as e:
            return False
        return True


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

# smoketests.py
# 用TestSuite类来定义和执行测试套件
# 多个测试加入到一个测试套件中去，用TestLoader和TextTestRunner来创建和运行测试套件
import unittest
from searchtests import SearchTests
from homepagetests import HomePageTest

# get all tests from SearchProductTest and HomePageTest class
search_tests = unittest.TestLoader().loadTestsFromTestCase(SearchTests)
home_page_tests = unittest.TestLoader().loadTestsFromTestCase(HomePageTest)

# create a test suite combining search_test and home_page_test
smoke_tests = unittest.TestSuite([home_page_tests, search_tests])

# run the suite
unittest.TextTestRunner(verbosity=2).run(smoke_tests)

# smoketests_with_html_report.py
import unittest
import HTMLTestRunner
import os
from searchtests import SearchTests
from homepagetests import HomePageTest

# get the directory path to output report file
result_dir = os.getcwd()

# get all tests from SearchProductTest and HomePageTest class

search_tests = unittest.TestLoader().loadTestsFromTestCase(SearchTests)
home_page_tests = unittest.TestLoader().loadTestsFromTestCase(HomePageTest)

# create a test suite combining search_test and home_page_test
smoke_tests = unittest.TestSuite([home_page_tests, search_tests])

# open the report file
outfile = open(result_dir + '\SmokeTestReport.html', 'w')

# configure.HTMLTestRunner options
runner = HTMLTestRunner.HTMLTestRunner(stream=outfile, title='Test Report', dscription='Smoke Testes')

# run the suite using HTMLTestRunner
runner.run(smoke_tests)