擅长:python、mysql、java
<p>为此,我发现当需要设置环境变量时,<a href="https://docs.pytest.org/en/latest/monkeypatch.html" rel="noreferrer">pytest's monkeypatch fixture</a>会产生更好的代码:</p>
<pre><code>def test_conn(monkeypatch):
monkeypatch.setenv('DATABASE_URL', '<URL WITH CREDENTIAL PARAMETERS>')
with patch(app.mysql.connector) as mock_mysql:
conn()
mock_mysql.connect.assert_callled_with(<CREDENTIAL PARAMETERS>)
</code></pre>