如何在Python中使用装饰器跨平台/浏览器运行测试？

Question

我有一个测试套件，如下所示：

import unittest
from selenium import webdriver

SAUCE_USERNAME = 'xxx'
SAUCE_ACCESS_KEY = 'xxx'
sauce = SauceClient(SAUCE_USERNAME, SAUCE_ACCESS_KEY)

browsers = [{"platform": "Mac OS X 10.9",
         "browserName": "chrome",
         "version": "31"},
        {"platform": "Windows 8.1",
         "browserName": "internet explorer",
         "version": "11"}]


def on_platforms(platforms):
    def decorator(base_class):
        module = sys.modules[base_class.__module__].__dict__
        for i, platform in enumerate(platforms):
            d = dict(base_class.__dict__)
            d['desired_capabilities'] = platform
            name = "%s_%s" % (base_class.__name__, i + 1)
            module[name] = new.classobj(name, (base_class,), d)
    return decorator

@on_platforms(browsers)
class MyTestSuite(unittest.TestCase):
    @classmethod
    def setUpClass(cls):
        cls.desired_capabilities['name'] = cls.id()
        sauce_url = "http://%s:%s@ondemand.saucelabs.com:80/wd/hub"
        cls.driver = webdriver.Remote(
        desired_capabilities=cls.desired_capabilities,
        command_executor=sauce_url % (SAUCE_USERNAME,SAUCE_ACCESS_KEY))
        cls.driver.implicitly_wait(30)

    def test_1from_sauce(self):
        pass

    def test_2from_sauce(self):
        pass

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

if __name__ == "__main__":
     unittest.main()

我的目标是为浏览器中的浏览器/平台运行test_1from_sauce和test_2from_sauce，而且我也想在setUpClass上设置的浏览器中连续执行这两个操作。 为了进一步说明，我想打开浏览器并进行两个测试，然后退出该驱动程序并启动另一个驱动程序。

现在，当我运行此代码时，出现以下错误：TypeError：必须使用SeleniumSauce_2实例作为第一个参数调用未绑定方法setUpClass（）。

我知道应该在类和子类声明中进行修改，但是我不知道应该做什么或应该更改什么部分。

编辑：我省略了以下行，它工作正常：

cls.desired_capabilities['name'] = cls.id()

Answer 1

您在问题中显示的代码缺少sys和new的导入。 其次，添加正确的导入后运行代码时出现的错误不是您在问题中报告的内容，而是：

EE
======================================================================
ERROR: setUpClass (__main__.MyTestSuite_1)
----------------------------------------------------------------------
Traceback (most recent call last):
  File "tests.py", line 33, in setUpClass
    cls.desired_capabilities['name'] = cls.id()
TypeError: unbound method id() must be called with MyTestSuite_1 instance as first argument (got nothing instead)

我省略了第二个ERROR ，它与上面的相同，只是它是针对MyTestSuite_2而不是MyTestSuite_1 。 问题很明显。 您调用了cls的id()成员，但是id()是一个实例方法而不是类方法，因此它需要类的实例，而不是类本身。 我不确定最终想要什么，但是如果将其用于setUpClass ，则不再发生错误，并尝试进行连接：

@classmethod
def setUpClass(cls):
    cls.desired_capabilities['name'] = cls.__name__
    sauce_url = "http://%s:%s@ondemand.saucelabs.com:80/wd/hub"
    cls.driver = webdriver.Remote(
        desired_capabilities=cls.desired_capabilities,
        command_executor=sauce_url % (SAUCE_USERNAME, SAUCE_ACCESS_KEY))
    cls.driver.implicitly_wait(30)

唯一的修改是方法的第一行。

顺便说一句，我不明白为什么要使用new.classobj 。 一方面，不建议使用new模块。 其次，使用内置type而不是new.classobj 。