def test_outgoing_with_resource_uri(self):
class MockResource(ModelResource):
model_class = mock_orm.Model
resource_path = 'bar'
fields = [
AttributeField(attribute='bar', type=int),
]
field = RelatedManagerField(attribute='foo', resource_class=MockResource)
source_object = mock_orm.Model()
related_manager = mock_orm.Manager()
related_manager.all = Mock(return_value=mock_orm.QuerySet(
mock_orm.Model(pk=4, bar=14)
))
source_object.foo = related_manager
target_dict = {}
field.handle_outgoing(mock_context(), source_object, target_dict)
target = target_dict['foo']
# Not testing the hash of the dictionary that is tested else were
self.assertEqual(
[{'resourceUri': 'uri://bar', 'bar': 14}],
target
)
def test_outgoing(self):
class MockResource(ModelResource):
model_class = mock_orm.Model
fields = [
AttributeField(attribute='bar', type=int),
]
field = RelatedManagerField(attribute='foo',
resource_class=MockResource)
source_object = mock_orm.Model()
related_manager = mock_orm.Manager()
related_manager.all = Mock(
return_value=mock_orm.QuerySet(mock_orm.Model(pk=4, bar=14)))
source_object.foo = related_manager
target_dict = {}
field.handle_outgoing(mock_context(), source_object, target_dict)
self.assertEqual([{'_id': '4', 'bar': 14}], target_dict['foo'])
def test_outgoing(self):
class MockResource(ModelResource):
model_class = mock_orm.Model
fields = [
AttributeField(attribute='bar', type=int),
]
field = RelatedManagerField(attribute='foo', resource_class=MockResource)
source_object = mock_orm.Model()
related_manager = mock_orm.Manager()
related_manager.all = Mock(return_value=mock_orm.QuerySet(
mock_orm.Model(pk=4, bar=14)
))
source_object.foo = related_manager
target_dict = {}
field.handle_outgoing(mock_context(), source_object, target_dict)
self.assertEqual([{'_id': '4', 'bar': 14}], target_dict['foo'])
def test_outgoing_with_resource_uri(self):
class MockResource(ModelResource):
model_class = mock_orm.Model
resource_path = 'bar'
fields = [
AttributeField(attribute='bar', type=int),
]
field = RelatedManagerField(attribute='foo',
resource_class=MockResource)
source_object = mock_orm.Model()
related_manager = mock_orm.Manager()
related_manager.all = Mock(
return_value=mock_orm.QuerySet(mock_orm.Model(pk=4, bar=14)))
source_object.foo = related_manager
target_dict = {}
field.handle_outgoing(mock_context(), source_object, target_dict)
target = target_dict['foo']
# Not testing the hash of the dictionary that is tested else were
self.assertEqual([{'resourceUri': 'uri://bar', 'bar': 14}], target)
