def test_extend_master(self):
ExtendCommand.add_to_master(PersonSourceMockOne)
ExtendCommand.extend_master(PersonSourceMockTwo)
self.assertEqual(Person.objects.annotate(num_sources=Count("sources")).filter(num_sources__gt=1).count(), 1)
mp = Person.objects.annotate(num_sources=Count("sources")).get(num_sources__gt=1)
self.assertEqual(sorted(mp.props.keys()), sorted(['address', 'occupation', 'sex', 'single']))
def test_add_to_master(self):
ExtendCommand.add_to_master(PersonSourceMockOne)
self.assertEqual(Person.objects.count(), 2)
mp = Person.objects.last()
self.assertEqual(mp.sources.count(), 1)
self.skipTest("Test adding a city")