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

Question

我有一个名为basetestcase（）的父测试类，它由所有测试类继承

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)

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

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文件中

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文件，它们将它们组合在一起以在套件中运行

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。

能否帮助我如何以最佳方式从命令行获取这些值并提供给basetestcase？

Answer 1

我也很难在多个浏览器上运行相同的测试用例。经过大量的迭代，试验和错误以及朋友的输入后，我为我的项目实施了以下解决方案：

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

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中：

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)