使用mock对conn()进行单元测试:
app.朋友
import mysql.connector
import os,urlparse
def conn():
if 'DATABASE_URL' in os.environ:
url=urlparse(os.environ['DATABASE_URL'])
g.db = mysql.connector.connect(user=url.username,password=url.password, host=url.hostname,database=url.path[1:])
else mysql.connector.error.Errors as err:
return "Error
test.朋友
def test_conn(self):
with patch(app.mysql.connector) as mock_mysql:
with patch(app.os.environ) as mock_environ
con()
mock_mysql.connect.assert_callled_with("credentials")
错误:没有调用断言mock_mysql.connect.assert_called_with
。
我认为这是因为'Database_url'不在我修补的os.environ中,因为没有对mysql_mock.connect进行测试调用。
问题:
1我需要做些什么改变才能使这个测试代码工作?
我还要修补'urlparse'吗?
import mysql.connector
import os,urlparse
@mock.patch.dict(os.environ,{'DATABASE_URL':'mytemp'})
def conn(mock_A):
print os.environ["mytemp"]
if 'DATABASE_URL' in os.environ:
url=urlparse(os.environ['DATABASE_URL'])
g.db = mysql.connector.connect(user=url.username,password=url.password, host=url.hostname,database=url.path[1:])
else mysql.connector.error.Errors as err:
return "Error
你可以尝试这种方式。只需用conn
参数调用dummy
。
要么
如果您不想修改原始功能,请尝试以下操作:
def func():
print os.environ["mytemp"]
def test_func():
k=mock.patch.dict(os.environ,{'mytemp':'mytemp'})
k.start()
func()
k.stop()
test_func()
为此,我发现当您需要设置环境变量时,pytest's monkeypatch fixture会产生更好的代码:
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>)
您还可以使用此modified_environ
中描述的question上下文管理器来设置/恢复环境变量。
with modified_environ(DATABASE_URL='mytemp'):
func()