pytest 根据环境跳过测试条件

Question

在使用 pytest 框架的脚本和 conftest.py 中，我有一个解析器从命令行获取环境值，以便测试知道它需要在哪个环境中运行。

def pytest_addoption(parser): # obtiene el valor de CLI    
parser.addoption("--env")

--

@pytest.fixture()
def environment(env):
if env == 'DEV':
    entorno = 'DEV'
elif env == 'TE':
    entorno = 'TE'
elif env == 'PROD':
    entorno = 'PROD'
else:
    entorno = 'UAT'
return entorno

我有一些永远不应该在 PROD 中运行的测试，有没有办法使用 pytest 进行标记或跳过条件来检查 env 变量是否为“PROD”不应运行？我想知道是否可以在测试开始运行之前获取环境值，以便我可以在测试顶部添加 @pytest.mark.skipif(env=="PROD") 。我发现的唯一方法是在测试中添加 if 条件，如下所示，但我不认为这是最好的方法：

@pytest.mark.parametrize('policy_number', policy_list)
def test_fnol_vida_fast(self, setUp, environment, request, policy_number):
    try:
        if self.entorno != 'PROD':
            ###################
            Script Script Script
            ###################
        else:
            pytest.skip('Scripts FNOL don't run in PROD')
    except Exception as e:
        self.logger.error("******************* test_fnol_vida_fast (" + str(policy_number) + ") FAILED *******************")
        self.commons.take_screenshot(request.node.name)
        raise e
    finally:
        self.driver.close()
        self.driver.quit()

Answer 1

尝试这样的事情：

import os
import pytest


def requires_env(*envs):
    env = os.environ.get(
        'ENVIRONMENT', 'test'
    )

    return pytest.mark.skipif(
        env not in list(envs),
        reason=f"Not suitable envrionment {env} for current test"
    )

并像这样使用：

@requires_env("test", "stag")
def test_1():
   assert 1 == 1

非常适合我

顺便说一句，您可以删除将字符串转换为列表并比较环境（例如 env == env）的步骤