Python: Mock side_effect on object attribute

Question

¿Es posible tener un side_effect en una propiedad? Si miro la documentación de Mock, parece que solo es posible en los métodos de objeto.

estoy tratando de probar lo siguiente:

def get_object(self): 
    try: 
     return self.request.user.shop 
    except Shop.DoesNotExist: 
     return None 

Quiero tienda a elevar una excepción DoesNotExist.

Supongo que no estaba lo suficientemente claro, pero estoy hablando de la biblioteca de simulación de espacio vacío.

http://www.voidspace.org.uk/python/mock/index.html

Answer 1

El autor hizo un post sobre este problema. Fui por primera solución que se ve así:

class UserMock(Mock): 
    @property 
    def shop(self): 
     raise Shop.DoesNotExist() 

http://www.voidspace.org.uk/python/weblog/arch_d7_2010_11_20.shtml#e1196

Answer 2

Sí, se puede utilizar una propiedad para ello:

In [1]: class ShopDoesNotExist(Exception): 
    ...:  pass 
    ...: 

In [2]: class User(object): 
    ...:  @property 
    ...:  def shop(self): 
    ...:   raise ShopDoesNotExist 
    ...: 
    ...: 

In [3]: u = User() 

In [4]: u.shop 
--------------------------------------------------------------------------- 
ShopDoesNotExist       Traceback (most recent call last) 

Answer 3

Vale la pena señalar que ahora existe la clase PropertyMock:

>>> m = MagicMock() 
>>> p = PropertyMock(side_effect=ValueError) 
>>> type(m).foo = p 
>>> m.foo 
Traceback (most recent call last): 
.... 
ValueError 

Ese ejemplo fue taken from the official site.

Answer 4

También puede intentar parchear el campo relacionado usando PropertyMock como argumento new_callable.

Ejemplo:

from unittest import TestCase 
import mock 
from django import models 
from django.core.exceptions import ObjectDoesNotExist 

class Foo(models.Model): 
    # ... 
    @property 
    def has_pending_related(self): 
     try: 
      return self.related_field.is_pending 
     except ObjectDoesNotExist: 
      return False 

class FooTestCase(TestCase): 
    # ... 
    @mock.patch.object(Foo, 'related_field', new_callable=mock.PropertyMock) 
    def test_pending_related(self, related_field): 
     related_field.side_effect = ObjectDoesNotExist 
     foo = Foo() 
     self.assertFalse(foo.has_pending_related)