Pytest 模拟隐藏预期密码

Question

我正在测试一个从命令行获取登录名和密码的功能。下面的代码按预期工作：

# ---------- Function part ----------
import getpass

def my_function():
    login = input('Login: ')
    password = getpass.getpass()

    # example to simulate connection to a identification server
    if login+password == 'foobar':
        return True
    else:
        return False

# ---------- Test part ----------
from unittest.mock import patch

@patch("getpass.getpass")
@patch("builtins.input")
def test_my_function(input, getpass):
    input.return_value = 'foo'
    getpass.return_value = 'bar'
    assert my_function()

现在我想隐藏预期的密码 (bar)，以便将测试发布到 public 存储库。

我考虑过对 getpass 输入进行编码，但这会导致连接失败。以及我能想到的所有其他解决方案或多或少地向任何可以阅读代码的人公开密码。

是否有存储密码的解决方案，以便连接函数可以使用它，并且任何阅读源代码的人都无法读取它？

Answer 1

如何让它成为一个 shell 环境？

当你想运行你的测试时，先使用export PASSWORD=bar。

在您的代码中，从 os.environ['PASSWORD'] 读取密码。

Pytest 模拟隐藏预期密码

Pytest mock hide expected password

python

mocking

pytest

getpass