Python se moquer de Patch os.environ et valeur de retour

Tests unitaires conn() à l'aide de maquette:

app.py

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.py

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")

Erreur: Affirmation mock_mysql.connect.assert_called_with n'est pas appelé.

qui, je crois, c'est parce que "Database_url" n'est pas dans mon corrigé os.environ et en raison de cet appel d'essai n'est pas fait pour mysql_mock.connectez.

Questions:

1 quels sont les changements que je dois faire pour faire ce test de code de travail?

2.Dois-je également patch 'urlparse'?

OriginalL'auteur immrsteel | 2015-07-23

  1. 32
    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

    Vous pouvez essayer de cette façon.Appelez simplement conn avec un dummy argument.

    Ou

    Si vous ne voulez pas modifier ur fonction d'origine, essayez ceci:

    def func():
    print os.environ["mytemp"]

def test_func():
    k=mock.patch.dict(os.environ,{'mytemp':'mytemp'})
    k.start()
    func()
    k.stop()

test_func()
    Merci man,je l'ai réalisé plus tard, et bientôt après avoir supprimé mon commentaire.Merci beaucoup c'est un travail, mais je suis confus que ci-dessus en utilisant la méthode de la façon dont {'mytemp':'mytemp'} passé en os.environ
    Merci beaucoup,j'ai accepté la réponse et upvote la réponse quand je l'aurais 15 réputation.
    ur le bienvenue 🙂

    OriginalL'auteur vks

  2. 2

    Vous pouvez également utiliser quelque chose comme la modified_environ gestionnaire de contexte de décrire dans ce question pour définir/restaurer les variables d'environnement.

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

    OriginalL'auteur Laurent LAPORTE

  3. 2

    Pour cela, je trouve que pytest de monkeypatch luminaire conduit à un code de meilleure qualité lorsque vous avez besoin de définir des variables d'environnement:

    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>)

    OriginalL'auteur Toote