Python模拟补丁os.environ并返回值

您可以尝试此解决方案。只需调用

conn

dummy

import mysql.connectorimport 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()