Variabile istanza classe mock Python

Question

Sto usando la libreria Python mock. So come deridere un metodo istanza classe seguendo il document:

>>> def some_function(): 
...  instance = module.Foo() 
...  return instance.method() 
... 
>>> with patch('module.Foo') as mock: 
...  instance = mock.return_value 
...  instance.method.return_value = 'the result' 
...  result = some_function() 
...  assert result == 'the result' 

Tuttavia, ha cercato di deridere una variabile istanza classe ma non funziona (instance.labels nell'esempio seguente):

>>> with patch('module.Foo') as mock: 
...  instance = mock.return_value 
...  instance.method.return_value = 'the result' 
...  instance.labels = [1, 1, 2, 2] 
...  result = some_function() 
...  assert result == 'the result' 

Fondamentalmente voglio instance.labels sotto some_function ottenere il valore che voglio. Qualche suggerimento?

Answer 1

Questa versione di some_function() stampe deriso labels proprietà:

def some_function(): 
    instance = module.Foo() 
    print instance.labels 
    return instance.method() 

mio module.py:

class Foo(object): 

    labels = [5, 6, 7] 

    def method(self): 
     return 'some' 

Patching è la stessa come la tua:

with patch('module.Foo') as mock: 
    instance = mock.return_value 
    instance.method.return_value = 'the result' 
    instance.labels = [1,2,3,4,5] 
    result = some_function() 
    assert result == 'the result 

sessione della console completa:

>>> from mock import patch 
>>> import module 
>>> 
>>> def some_function(): 
...  instance = module.Foo() 
...  print instance.labels 
...  return instance.method() 
... 
>>> some_function() 
[5, 6, 7] 
'some' 
>>> 
>>> with patch('module.Foo') as mock: 
...  instance = mock.return_value 
...  instance.method.return_value = 'the result' 
...  instance.labels = [1,2,3,4,5] 
...  result = some_function() 
...  assert result == 'the result' 
...  
... 
[1, 2, 3, 4, 5] 
>>> 

Per me il codice è funzionante.