Patcher une fonction avec Mock uniquement pour un module?

Question

Je dois corriger os.listdir et d'autres fonctions os pour tester ma fonction Python. Mais quand ils sont corrigés, l'instruction import échoue. Est-il possible de patcher cette fonction uniquement à l'intérieur de ce seul module, le vrai, et laisser tests.py fonctionner normalement?

Voici un exemple qui casse import:

import os 
from mock import patch 

# when both isdir and isfile are patched 
# the function crashes 
@patch('os.path.isdir', return_value=False) 
@patch('os.path.isfile', return_value=False) 
def test(*args): 
    import ipdb; ipdb.set_trace() 
    real_function(some_arguments) 
    pass 

test() 

Je veux real_function de voir une version patchée os.path, et des tests pour voir les fonctions normales.

here's the traceback

Answer 1

vous pouvez utiliser patch en tant que gestionnaire de contexte, il ne s'appliquera au code dans l'instruction with:

import os 
from mock import patch 

def test(*args): 
    import ipdb; ipdb.set_trace() 
    with patch('os.path.isdir', return_value=False): 
     with patch('os.path.isfile', return_value=False): 
      real_function(some_arguments) 
    pass 

test() 

Answer 2

Les travaux ci-dessous pour ce que vous avez besoin. Notez que j'imprime os.listdir avant et après le test, juste pour montrer que tout est retourné dans son état correct.

import unittest, os 
from unittest import TestLoader, TextTestRunner 

def tested_func(path): 
    return os.listdir(path * 2) 

class Example(unittest.TestCase): 

    def setUp(self): 
     self._listdir = os.listdir 
     os.listdir = lambda p: ['a', 'b', 'a', 'b'] 

    def test_tested_func(self): 
     self.assertEqual(tested_func('some_path'), ['a', 'b', 'a', 'b']) 

    def tearDown(self): 
     os.listdir = self._listdir 

print os.listdir 
TextTestRunner().run(TestLoader().loadTestsFromTestCase(Example)) 
print os.listdir 

Answer 3

Vous pouvez utiliser un correcteur comme décorateur sur les méthodes de test ou les classes de test. Cela rend le code très propre. Notez que l'objet corrigé est transmis à la méthode de test. Vous pouvez faire la même chose au niveau de la classe de test et l'objet fantaisie sera passé dans chaque méthode de test. Il n'est pas nécessaire d'utiliser setUp ou tearDown dans ce cas car tout est géré automatiquement. Notez également que vous vous moquez de la fonction que vous voulez patcher dans le module que vous voulez réellement tester.

Answer 4

Vous non seulement « peut », mais vous « devez » faire http://mock.readthedocs.org/en/latest/patch.html#where-to-patch

ps: Si vous voulez faire plus « déclaratif » façon de décrire les side_effects de l'unité que vous voulez patcher, je recommande d'utiliser la bibliothèque http://mockstar.readthedocs.org/