Python unittest 将参数传递给父测试类

2022-01-14 00:00:00 python python-3.x python-unittest selenium-webdriver automated-tests

问题描述

我有一个名为 basetestcase() 的父测试类这是由所有测试类继承的

I have a parent test class named as basetestcase() This is inherited by all the test classes

class BaseTestCase(unittest.TestCase):

    driver = None
    browser = read from command line
    operatingSystem = read from command line
    url = read from command line

    @classmethod
    def setUpClass(cls):
        """
        SetUp to initialize webdriver session, pages and other needed objects

        Returns:
        None
        """
        # Get webdriver instance
        # Browser should be read from the arguments
        if browser == "iexplorer":
            cls.driver = webdriver.Ie()
        elif browser == "firefox":
            cls.driver = webdriver.Firefox()
        elif browser == "chrome":
            cls.driver = webdriver.Chrome()
        else:
            cls.driver = webdriver.PhantomJS()

        # Similarly I want to get operating system and url also from command line
        driver.get(url)
        print("Tests are running on: " + operatingSystem)

然后我有两个单独的测试类:

Then I have two separate test classes:

class TestClass1(BaseTestCase):

    @classmethod
    def setUpClass(cls):
        super(TestClass1, cls).setUpClass()
        # Create object of another class to use in the test class
        # cls.abc = ABC()

    def test_methodA(self):
        # self.abc.methodFromABC() # This does not work
        # Not sure if I can use self.driver as it was defined as cls.driver in the setUpClass()
        self.driver.find_element(By.ID, "test_id").click()

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

这是第二类，两个类都在单独的 .py 文件中

This is the 2nd class, both the classes are in separate .py files

class TestClass2(GUIBaseTestCase):

    @classmethod
    def setUpClass(self):
        super(TestClass2, self).setUpClass()

    def test_methodA(self):
        self.driver.find_element(By.ID, "test_id").click()

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

然后我有一个测试套件脚本，一个单独的 .py 文件，它将它们组合在一起在一个套件中运行

Then I have a test suite script, a separate .py file which clubs them together to run in a suite

import unittest
from tests.TestClass1 import TestClass1
from tests.TestClass2 import TestClass2

# Get all tests from TestClass1 and TestClass2
tc1 = unittest.TestLoader().loadTestsFromTestCase(TestClass1)
tc2 = unittest.TestLoader().loadTestsFromTestCase(TestClass2)

# Create a test suite combining TestClass1 and TestClass2
smokeTest = unittest.TestSuite([tc1, tc2])

unittest.TextTestRunner(verbosity=2).run(smokeTest)

我想运行测试套件，并想从命令行向 basetestcase 提供浏览器、操作系统和 url，这些参数直接由 basetestcase.py 使用.实际测试类继承 basetestcase.

I want to run the test suite and want to provide browser, operating system and url to the basetestcase from the command line and these arguments are directly used by basetestcase.py. Actual test classes inherit the basetestcase.

您能否帮助我以最佳方式从命令行获取这些值并提供给基本测试用例?

Could you please help me with how to get these values from the command line in the best way and provide to the basetestcase?

解决方案

我也很难在多个浏览器上运行相同的测试用例.经过多次迭代、反复试验以及朋友们的意见，我为我的项目实施了以下解决方案:

I also struggled to run the same test cases on multiple browsers. After a lot of iterations, trial and error and input from friends I implemented the following solutions to my projects:

这里的 TestCase 是包含所有测试的类，浏览器驱动程序为 None.SafariTestCase 继承 TestCase 并覆盖 setUpClass 并将浏览器驱动程序设置为 safari 驱动程序，与 ChromeTestCase 相同，您可以为其他浏览器添加更多类.可以在 TestSuite 文件中获取命令行输入，并根据参数有条件地加载测试:

Here TestCase is the class that has all the tests and the browser driver is None. SafariTestCase inherits the TestCase and overrides setUpClass and sets the browser driver to be safari driver and same with the ChromeTestCase and you can add more class for other browsers. Command Line input can be taken in the TestSuite file and conditionally load tests based on the arguments:

class TestCase(unittest.TestCase):
  @classmethod
  def setUpClass(cls):
    cls.browser = None

  def test_1(self):
    self.assert(self.browser.find_element_by_id('test1')

  def test_2(self):
    self.assert(self.browser.find_element_by_id('test2')

  def test_3(self):
    self.assert(self.browser.find_element_by_id('test3')

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


class SafariTestCase(TestCase):
  @classmethod:
  def setUpClass(cls):
    cls.browser = webdriver.Safari(executable_path='/usr/bin/safaridriver')

class ChromeTestCase(TestCase):
  @classmethod:
  def setUpClass(cls):
    cls.browser = webdriver.Chrome(executable_path='/usr/local/bin/chromedriver')

在运行器文件TestSuite.py中:

In the runner file TestSuite.py:

import TestCase as tc

if len(sys.argv) == 1:
  print("Missing arguments, tell a couple of browsers to test against.")
  sys.exit(1)

if sys.argv[1] == 'safari':
  test = unittest.TestLoader().loadTestsFromTestCase(tc.SafariTestCase)

if sys.argv[1] == 'chrome':
  test = unittest.TestLoader().loadTestsFromTestCase(lt.ChromeTestCase)

unittest.TextTestRunner().run(test)

相关文章