Python mock Patch os.environ and return value

Question

Unit testing conn() using mock:

app.py

import mysql.connector
import os, urlparse


def conn():
    if "DAT         


        

Answer 1

You can also use something like the modified_environ context manager describe in this question to set/restore the environment variables.

with modified_environ(DATABASE_URL='mytemp'):
    func()