def test_relationships(self): rs = DatabaseWrapper(ExampleDatabaseMock()).relationships eq_([field.name for field in rs.keys()], ['reporter_id']) eq_([(r.to_field.name, r.from_field.name) for r in rs.values()], [('reporter_id', 'id')]) eq_([(r.to_field.model.name, r.from_field.model.name) for r in rs.values()], [('Article', 'Reporter')])
def test_add_relationships(self): d = DatabaseWrapper(ExampleDatabaseMock()) relationships = {'all': {}} d._add_relationships(relationships, ['Reporter', 'Article']) key = 'Article', 'reporter_id' r = relationships['all'] eq_([[f.name for f in key] for key in r.keys()], [['reporter_id', 'id']]) eq_([(rs.to_field.name, rs.from_field.name) for rs in r.values()], [('reporter_id', 'id')])