gpt4 book ai didi

python - 如何在pytest中设置环境变量

转载 作者:行者123 更新时间:2023-12-04 13:24:12 29 4
gpt4 key购买 nike

我有一个使用环境变量的 Lamba 处理程序。如何使用 pytest 设置该值。我收到错误

tests/test_kinesis.py:3: in <module>
from runner import kinesis
runner/kinesis.py:6: in <module>
DATA_ENGINEERING_BUCKET = os.environ["BUCKET"]
../../../../../.pyenv/versions/3.8.8/lib/python3.8/os.py:675: in __getitem__
raise KeyError(key) from None
E KeyError: 'BUCKET'
7:03
我尝试在测试中像这样设置
class TestHandler(unittest.TestCase):
@mock_s3
@mock_lambda
def test_handler(monkeypatch):
monkeypatch.setenv("BUCKET", "test-bucket")
actual = kinesis.handler(kinesis_stream_event, "")
expected = {"statusCode": 200, "body": "OK"}
assert actual == expected
DATA_ENGINEERING_BUCKET = os.environ["BUCKET"]


def handler(event, context):
...

最佳答案

在您的monkeypatch 能够运行之前,您就遇到了故障。首次导入 runner 模块时会加载环境变量。
如果这是您拥有的模块,如果未设置 DATA_ENGINEERING_BUCKET,我建议修改代码以使用默认值。然后您可以在运行时通过调用 module.DATA_ENGINEERING_BUCKET = "my_bucket" 将其值修改为您想要的任何值。 .

DATA_ENGINEERING_BUCKET = os.environ.get("BUCKET", default="default_bucket")
如果您无法修改该文件,那么事情就会变得更加复杂。
我研究了创建一个全局装置,在任何测试加载之前对环境进行猴子补丁并加载一次模块,并收到有关在 session 级装置中使用函数级装置的 pytest 错误。这很有意义,monkeypatch 真的不是为了长期伪造东西。您可以在monkeypatch 之后将模块负载粘贴到您的测试中,但这会生成大量样板文件。
最终创建了一个 fixture 来提供类而不是导入它。 fixture ;套 os.environ到所需的值,加载模块,将 os.environ 重置为其原始值,然后生成模块。任何需要此模块的测试都可以请求 fixture 在其范围内访问它。需要注意的是,因为测试文件是在运行 fixture 之前导入的,任何不使用 fixture 并正常导入模块的测试文件都会引发 KeyError 并导致 pytest 在运行任何测试之前崩溃。
conftest.py
import os, pytest

@pytest.fixture(scope='session')
def kinesis():
old_environ = os.environ
os.environ = {'BUCKET': 'test-bucket'}
import kinesis
os.environ = old_environ
yield kinesis
测试文件
# Do NOT import kinesis in any test file. Rely on the fixture.
class TestHandler(unittest.TestCase):
@mock_s3
@mock_lambda
def test_handler(kinesis):
actual = kinesis.handler(kinesis_stream_event, "")
expected = {"statusCode": 200, "body": "OK"}
assert actual == expected

一种可能更简单的方法 os.environ是在 os 首次加载时创建的环境变量字典。如果您希望每个测试都有一个值,那么您只需要在加载任何测试模块之前添加您想要的值。如果你把 os.environ['BUCKET'] = 'test-bucket'在 conftest.py 的顶部,您将为测试 session 的其余部分设置环境变量。那么只要模块的第一次导入发生在之后,您就不会出现关键错误。这种方法的最大缺点是,除非您知道查看 conftest.py 或 grep 代码,否则很难确定在故障排除时设置环境变量的位置。

关于python - 如何在pytest中设置环境变量,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/69623784/

29 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com