您可以尝试此解决方案。只需调用conn一个dummy参数即可：
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:
return "Error"
def func():
print os.environ["mytemp"]
def test_func():
k = mock.patch.dict(os.environ, {"mytemp": "mytemp"})
k.start()
func()
k.stop()
test_func()
602392714
清零编程群