Se moquer des objets fichiers ou itérables en python

Ce qui est bon de se moquer et de tests de code que que ces dossiers de l'objet renvoyé par open()à l'aide de maquette bibliothèque?

whitelist_data.py:

WHITELIST_FILE = "testdata.txt"

format_str = lambda s: s.rstrip().lstrip('www.')
whitelist = None

with open(WHITELIST_FILE) as whitelist_data:
    whitelist = set(format_str(line) for line in whitelist_data)

if not whitelist:
    raise RuntimeError("Can't read data from %s file" % WHITELIST_FILE)

def is_whitelisted(substr):
    return 1 if format_str(substr) in whitelist else 0

Voici comment j'ai essayer de le tester.

import unittest
import mock 

TEST_DATA = """
domain1.com
domain2.com
domain3.com
"""

class TestCheckerFunctions(unittest.TestCase):

    def test_is_whitelisted_method(self):
        open_mock = mock.MagicMock()
        with mock.patch('__builtin__.open',open_mock):
            manager = open_mock.return_value.__enter__.return_value
            manager.__iter__ = lambda s: iter(TEST_DATA.splitlines())
            from whitelist_data import is_whitelisted
            self.assertTrue(is_whitelisted('domain1.com'))

if __name__ == '__main__':
    unittest.main()

Résultat de python tests.py est:

$ python tests.py

E
======================================================================
ERROR: test_is_whitelisted_method (__main__.TestCheckerFunctions)
----------------------------------------------------------------------
Traceback (most recent call last):
  File "tests.py", line 39, in test_is_whitelisted_method
    from whitelist_data import is_whitelisted
  File "/Users/supa/Devel/python/whitelist/whitelist_data.py", line 20, in <module>
    whitelist = set(format_str(line) for line in whitelist_data)
TypeError: 'Mock' object is not iterable

----------------------------------------------------------------------
Ran 1 test in 0.001s

UPD: Merci à Adam, j'ai réinstallé maquette de la bibliothèque(pip install -e hg+https://code.google.com/p/mock#egg=mock) et mis à jour tests.py. Fonctionne comme un charme.

source d'informationauteur Victor Miroshnikov

  1. 15

    Vous êtes à la recherche d'un MagicMock. Ceci supporte l'itération.

    Dans la maquette de 0.80beta4, patch renvoie une MagicMock. Donc, ce simple exemple de travaux:

    import mock

def foo():
    for line in open('myfile'):
        print line

@mock.patch('__builtin__.open')
def test_foo(open_mock):
    foo()
    assert open_mock.called

    Si vous êtes en cours d'exécution se moquer de 0.7.x (On dirait que vous êtes), je ne pense pas que vous pouvez accomplir cela avec le patch seul. Vous aurez besoin pour créer la maquette séparément, puis le passer en patch:

    import mock

def foo():
    for line in open('myfile'):
        print line

def test_foo():
    open_mock = mock.MagicMock()
    with mock.patch('__builtin__.open', open_mock):
        foo()
        assert open_mock.called

    Remarque - j'ai couru avec py.test, cependant, ces mêmes approches de travail avec unittest.