def setUp(self): super(TestRegion, self).setUp() self.org = factories.Org() self.temba_groups = { '1': TembaGroup.create(uuid="1", name="Uganda"), '2': TembaGroup.create(uuid="2", name="Kampala"), '3': TembaGroup.create(uuid="3", name="Entebbe"), '4': TembaGroup.create(uuid="4", name="Makerere"), } self.uganda = factories.Region(org=self.org, uuid='1', name="Uganda") self.kampala = factories.Region(org=self.org, uuid='2', name="Kampala", parent=self.uganda) self.entebbe = factories.Region(org=self.org, uuid='3', name="Entebbe", parent=self.uganda) self.makerere = factories.Region(org=self.org, uuid='4', name="makerere", parent=self.kampala) self.inactive = factories.Region(org=self.org, uuid='5', name="inactive", parent=self.kampala, is_active=False) self.mock_temba_client.get_contacts.return_value = []
def setUp(self): super(TestRegion, self).setUp() self.org = factories.Org() self.temba_groups = { '1': TembaGroup.create(uuid="1", name="Uganda"), '2': TembaGroup.create(uuid="2", name="Kampala"), '3': TembaGroup.create(uuid="3", name="Entebbe"), '4': TembaGroup.create(uuid="4", name="Makerere"), } self.uganda = factories.Region( org=self.org, uuid='1', name="Uganda") self.kampala = factories.Region( org=self.org, uuid='2', name="Kampala", parent=self.uganda) self.entebbe = factories.Region( org=self.org, uuid='3', name="Entebbe", parent=self.uganda) self.makerere = factories.Region( org=self.org, uuid='4', name="makerere", parent=self.kampala) self.inactive = factories.Region( org=self.org, uuid='5', name="inactive", parent=self.kampala, is_active=False) self.mock_temba_client.get_contacts.return_value = []
def test_sync_create_new(self): """Create a new group if the UUID hasn't been seen before.""" self.temba_groups['6'] = TembaGroup.create(uuid='6', name="New") self.mock_temba_client.get_groups.return_value = self.temba_groups.values() uuids = ['1', '2', '3', '4', '6'] models.Region.sync_with_temba(self.org, uuids) self.refresh_regions() new_region = models.Region.objects.get(uuid='6') self.assertEqual(set(models.Region.get_all(self.org)), set([ self.uganda, self.kampala, self.makerere, self.entebbe, new_region, ])) self.assertEqual(new_region.name, "New") self.assertIsNone(new_region.parent, None) self.assertEqual(self.mock_temba_client.get_groups.call_count, 1) self.assertEqual(self.mock_temba_client.get_contacts.call_count, 2)
def test_sync_create_new(self): """Create a new group if the UUID hasn't been seen before.""" self.temba_groups['6'] = TembaGroup.create(uuid='6', name="New") self.mock_temba_client.get_groups.return_value = self.temba_groups.values( ) uuids = ['1', '2', '3', '4', '6'] models.Region.sync_with_temba(self.org, uuids) self.refresh_regions() new_region = models.Region.objects.get(uuid='6') self.assertEqual( set(models.Region.get_all(self.org)), set([ self.uganda, self.kampala, self.makerere, self.entebbe, new_region, ])) self.assertEqual(new_region.name, "New") self.assertIsNone(new_region.parent, None) self.assertEqual(self.mock_temba_client.get_groups.call_count, 1) self.assertEqual(self.mock_temba_client.get_contacts.call_count, 2)