1

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

# ---------- 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) 以便将测试发布到公共存储库。

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

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

4

1 回答 1

2

让它成为一个shell环境怎么样？

当您要运行测试时，export PASSWORD=bar请先使用。

在您的代码中，从os.environ['PASSWORD'].

于 2018-03-02T09:32:36.443 回答