patch Mock da/istruzione import in Python

Question

Io sto cercando di ottenere mock.patch a lavorare il seguente pezzo di codice di esempio:

from mock import patch 
from collections import defaultdict 

with patch('collections.defaultdict'): 
    d = defaultdict() 
    print 'd:', d 

Emette il seguente:

d: defaultdict(None, {}) 

Il che significa quel defaultdict non è stato patchato.

Se si sostituisce la dichiarazione/importazione con un istruzione import dritto funziona:

from mock import patch 
import collections 

with patch('collections.defaultdict'): 
d = collections.defaultdict() 
print 'd:', d 

uscita è:

d: <MagicMock name='defaultdict()' id='139953944084176'> 

C'è un modo per rattoppare una chiamata utilizzando da/importazione?

Grazie

Answer 1

Se stai patch qualcosa nello stesso modulo, è possibile utilizzare __main__:

from mock import patch 
from collections import defaultdict 

with patch('__main__.defaultdict'): 
    d = defaultdict() 
    print 'd:', d 

Se stai beffardo qualcosa per un modulo importato, tuttavia, vorrete di utilizzare il nome di quel modulo in modo è patchato il riferimento corretto (o il nome):

# foo.py 

from collections import defaultdict 

def bar(): 
    return defaultdict() 


# foo_test.py  

from mock import patch 
from foo import bar 

with patch('foo.defaultdict'): 
    print bar() 

il punto qui è che cerotto vuole che il percorso completo alla cosa è patch. Questo sembra un po 'strano quando si aggiusta qualcosa nel modulo corrente, dal momento che la gente non usa spesso __main__ (o deve fare riferimento al modulo corrente, per quello).

Answer 2

patch funziona con patch nomi. Non è possibile ottenere nulla rattoppando il nome collections.defaultdict se si utilizza il nome defaultdict (nello spazio dei nomi locale) per accedere all'oggetto. Vedere la documentazione allo http://www.voidspace.org.uk/python/mock/patch.html#id1.