擅长:python、mysql、java
<pre><code>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
</code></pre>
<p>你可以这样做。只要用<code>dummy</code>参数调用<code>conn</code>。</p>
<p>或者</p>
<p>如果不想修改原始函数,请尝试以下操作:</p>
<pre><code>def func():
print os.environ["mytemp"]
def test_func():
k=mock.patch.dict(os.environ,{'mytemp':'mytemp'})
k.start()
func()
k.stop()
test_func()
</code></pre>