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')])
