def test_append_relationships_to_list_single_relationship(self): """ Ensures that a Relationship (not ListRelationship) is properly added """ class MyResource(ResourceBase): pass relationships = [(MyResource(properties=dict(id=1)), 'name', True)] rel_dict = {} BasicJSONAdapter._append_relationships_to_list(rel_dict, relationships) self.assertDictEqual(dict(name=[dict(id=1)]), rel_dict)
def test_append_relationships_to_list_list_relationship(self): """ Tests whether the relationships are appropriately added to the response """ class MyResource(ResourceBase): pass relationship_list = [MyResource(properties=dict(id=1)), MyResource(properties=dict(id=2))] relationships = [(relationship_list, 'name', True)] rel_dict = {} BasicJSONAdapter._append_relationships_to_list(rel_dict, relationships) self.assertDictEqual(dict(name=[dict(id=1), dict(id=2)]), rel_dict)