def test_outgoing(self):
class Resource(ModelResource):
model_class = Mock()
fields = [
AttributeField(attribute='bar', type=int),
]
field = OneToOneField(attribute='foo', resource_class=Resource)
source_object = Mock()
source_object.foo.bar = 20
target_dict = dict()
field.handle_outgoing(mock_context(), source_object, target_dict)
self.assertEqual(target_dict['foo'], {'bar': 20})
def test_outgoing(self):
class Resource(ModelResource):
model_class = Mock()
fields = [
AttributeField(attribute='bar', type=int),
]
field = OneToOneField(attribute='foo', resource_class=Resource)
source_object = Mock()
source_object.foo.bar = 20
target_dict = dict()
field.handle_outgoing(mock_context(), source_object, target_dict)
self.assertEqual(target_dict['foo'], {'bar': 20})
