Python mock instancja klasy zmiennej

Question

Używam biblioteki Python mock. Wiem jak mock metodę instancji klasy postępując zgodnie z 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' 

jednak próbował szydzić zmienną instancji klasy, ale nie działa (instance.labels w poniższym przykładzie):

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

Zasadniczo chcę instance.labels pod some_function uzyskać żądaną wartość. Jakieś wskazówki?

Answer 1

Ta wersja some_function() wydruków szydzili labels właściwość:

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

My module.py:

class Foo(object): 

    labels = [5, 6, 7] 

    def method(self): 
     return 'some' 

Łatanie jest takie same:

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 

Pełna sesja konsoli:

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

Dla mnie Twój kod to działa.