py.test conftest - 根据sys.argv修改夹具参数

Question

我有一个用于 Selenium 测试的 pytest 工作环境。我在 conftest.py 中使用了一个参数化的夹具，它允许我测试所有不同的浏览器而无需重写测试。我想将一个命令行参数传递给我的脚本，这样我就可以运行只在特定的浏览器中使用它，而不是所有的浏览器。为此，我需要修改传递到我的夹具中的变量。到目前为止，我还没弄清楚该怎么做。我下面的示例不起作用，可能是因为 pytest 将 conftest.py 与我的启动脚本描述的变量分开解析。

#conftest.py
browsers = { "ff" : webdriver.Firefox, "ie" : webdriver.Ie }
@pytest.yield_fixture(params=browsers.keys())
def browser(request):
    driver = browsers[request.param]()
    yield driver
    driver.quit()

#test_simple.py
def test_simple(browser):
    browser.get("http://whosebug.com")

#main.py
browsers = {}
if __name__ == "__main__":
    if len(sys.argv) > 1:
        for arg in sys.argv[1:]:
            if arg == "-ff":
                browsers = { "ff" : webdriver.Firefox }
            elif arg == "-ie":
                browsers = { "ie" : webdriver.Ie }
            elif arg == "-all":
                browsers = { "ff" : webdriver.Firefox, "ie" : webdriver.Ie }

    pytest.main()

如何将值传递给 conftest.py 中的参数化夹具？

Answer 1

只需使用 ENV 变量：

#conftest.py
browsers = { "ff" : webdriver.Firefox, "ie" : webdriver.Ie }
@pytest.yield_fixture(params=browsers.keys())
def browser(request):
    driver = browsers[request.param]()
    yield driver
    driver.quit()

#test_simple.py
def test_simple(browser):
    browser.get("http://whosebug.com")

#main.py
browsers = {}
browserToRun = os.environ.get('BROWSER_TO_RUN')
if __name__ == "__main__":
            if browserToRun == "ff":
                browsers = { "ff" : webdriver.Firefox }
            elif browserToRun == "ie":
                browsers = { "ie" : webdriver.Ie }
            elif browserToRun == "all":
                browsers = { "ff" : webdriver.Firefox, "ie" : webdriver.Ie }

    pytest.main()

和运行测试如下：

BROWSER_TO_RUN=ff run_tests.py

Answer 2

参见：Pass different values to a test function, depending on command line options。

您实际上无法使用命令行选项更改参数化，因为参数化定义发生在导入期间，但如果用户在命令行中指定一个浏览器，您可以轻松跳过对其他浏览器的测试：

# conftest.py
import pytest

browsers = {"ff": 'FIREFOX', 'ie': 'INTERNETEXPLORER'}

def pytest_addoption(parser):
    parser.addoption("--browser", default='',
        type='choice', choices=sorted(browsers),
        help="runs tests only for given browser")    

@pytest.yield_fixture(params=browsers.keys())
def browser(request):
    selected = request.config.getoption('browser')
    if selected and selected != request.param:
        pytest.skip('browser {} selected in the command line'.format(selected))
    driver = browsers[request.param]
    yield driver

有了这个，当用户运行s pytest 没有向 --browser 传递任何值时，所有测试运行照常：

============================= test session starts =============================
platform win32 -- Python 2.7.6 -- py-1.4.26 -- pytest-2.7.0.dev1 -- X:\temp\sandbox\.env27\Scripts\python.exe
plugins: xdist
collecting ... collected 2 items

test_simple.py::test_simple[ie] PASSED
test_simple.py::test_simple[ff] PASSED

========================== 2 passed in 0.01 seconds ===========================

但是如果用户通过 --browser=ie，则跳过 firefox 测试：

============================= test session starts =============================
platform win32 -- Python 2.7.6 -- py-1.4.26 -- pytest-2.7.0.dev1 -- X:\temp\sandbox\.env27\Scripts\python.exe
plugins: xdist
collecting ... collected 2 items

test_simple.py::test_simple[ie] PASSED
test_simple.py::test_simple[ff] SKIPPED

===================== 1 passed, 1 skipped in 0.01 seconds =====================

Answer 3

 pytest -k test_simple['ie']

上面一行会选择性地运行匹配测试

py.test conftest - 根据sys.argv修改夹具参数

py.test conftest - Modify Fixture Parameter Based on sys.argv

python

selenium

pytest