Question

我的项目有一个名为config.py的文件，除其他外，它具有以下代码：

class Secret(Enum):
    DATABASE_A = 'name_of_secret_database_A'
    DATABASE_A = 'name_of_secret_database_A'

    def secret(self):
        if self.value:
            return get_secret(self.value)
        return {}

def get_secret(secret_name):
    session = Session()
    client = session.client(
        service_name='secretsmanager',
        region_name='us-east-1',
    )
    secret_value = client.get_secret_value(SecretId=secret_name)
    return loads(secret_value.get('SecretString', "{}"))

我需要使用pytest在测试中针对所有枚举调用模拟 get_secret ，例如 Secret.DATABASE_A.secret（）

Answer 1

您可以使用monkeypatch来覆盖get_secret()的行为。我已经将get_secret()方法设置为Secret类的静态方法，但是您可以使其成为所需任何模块的一部分，也可以将其导入。只要确保您也更改了monkeypatch.setattr()调用即可。

import pytest
from enum import Enum


class Secret(Enum):
    DATABASE_A = 'name_of_secret_database_A'
    DATABASE_B = 'name_of_secret_database_B'

    def secret(self):
        if self.value:
            return Secret.get_secret(self.value)
        return {}
        
    @staticmethod
    def get_secret(secret_name):
        session = Session()
        client = session.client(
            service_name='secretsmanager',
            region_name='us-east-1',
        )
        secret_value = client.get_secret_value(SecretId=secret_name)
        return loads(secret_value.get('SecretString', "{}"))

def test_secret_method(monkeypatch):
    def get_secret(secret_name):
        return "supersecret"
    
    monkeypatch.setattr(Secret, "get_secret", get_secret)

    s = Secret.DATABASE_A
    assert s.secret() == "supersecret"

这将返回1个通过的测试。

这里发生的事情是，我也在test_secret_method中创建了一个函数get_secret()，然后用该新方法覆盖了Secret.get_secret()。现在，您可以在test_method中使用Secret类，并确保在不实际运行原始代码的情况下，“ get_secret（）”方法将返回什么。

Pytest模拟AWS安全管理器

1 个答案: