class LocationSyncTest(TestCase): def setUp(self): self.endpoint = MockEndpoint('http://test-api.com/', 'dummy', 'dummy') self.api_object = EWSApi(TEST_DOMAIN, self.endpoint) self.datapath = os.path.join(os.path.dirname(__file__), 'data') initial_bootstrap(TEST_DOMAIN) self.api_object.prepare_commtrack_config() for location in CouchLocation.by_domain(TEST_DOMAIN): location.delete() for sql_location in SQLLocation.objects.all(): sql_location.delete() def test_create_non_facility_location(self): with open(os.path.join(self.datapath, 'sample_locations.json')) as f: location = Location(json.loads(f.read())[0]) ewsghana_location = self.api_object.location_sync(location) self.assertEqual(ewsghana_location.name, location.name) self.assertEqual(ewsghana_location.location_type, location.type) self.assertEqual(ewsghana_location.longitude, float(location.longitude)) self.assertEqual(ewsghana_location.latitude, float(location.latitude)) self.assertEqual(ewsghana_location.parent, location.parent_id) self.assertFalse(ewsghana_location.is_archived) sql_location = ewsghana_location.sql_location self.assertEqual(ewsghana_location.get_id, sql_location.location_id) self.assertIsNotNone(sql_location.id) self.assertIsNone(ewsghana_location.linked_supply_point()) self.assertIsNone(sql_location.supply_point_id) def test_create_facility_location(self): with open(os.path.join(self.datapath, 'sample_locations.json')) as f: location = Location(json.loads(f.read())[1]) with open(os.path.join(self.datapath, 'sample_products.json')) as f: for p in json.loads(f.read()): self.api_object.product_sync(Product(p)) self.assertEqual(8, SQLProduct.objects.filter(domain=TEST_DOMAIN).count()) ewsghana_location = self.api_object.location_sync(location) self.assertEqual(ewsghana_location.name, location.supply_points[0].name) self.assertEqual(ewsghana_location.site_code, location.supply_points[0].code) self.assertEqual("Hospital", ewsghana_location.location_type) self.assertEqual(ewsghana_location.longitude, float(location.longitude)) self.assertEqual(ewsghana_location.latitude, float(location.latitude)) self.assertFalse(ewsghana_location.is_archived) sql_location = ewsghana_location.sql_location self.assertEqual(ewsghana_location.get_id, sql_location.location_id) self.assertEqual(int(sql_location.parent.external_id), location.parent_id) self.assertIsNotNone(sql_location.id) self.assertIsNotNone(sql_location.supply_point_id) supply_point = SupplyPointCase.get_by_location_id(TEST_DOMAIN, sql_location.location_id) self.assertIsNotNone(supply_point) self.assertEqual(location.supply_points[0].id, int(supply_point.external_id)) self.assertEqual(location.supply_points[0].name, supply_point.name) self.assertListEqual(location.supply_points[0].products, [product.code for product in ewsghana_location.sql_location.products]) def test_create_region_with_two_supply_points(self): with open(os.path.join(self.datapath, 'sample_locations.json')) as f: location = Location(json.loads(f.read())[4]) ewsghana_location = self.api_object.location_sync(location) self.assertEqual(2, SQLLocation.objects.filter( domain=TEST_DOMAIN, location_type__administrative=False).count() ) self.assertIsNone(ewsghana_location.linked_supply_point()) self.assertIsNone(ewsghana_location.sql_location.supply_point_id) self.assertEqual(location.name, ewsghana_location.sql_location.name) self.assertEqual(location.code, ewsghana_location.sql_location.site_code) self.assertFalse(ewsghana_location.is_archived) def test_facility_without_supply_point(self): with open(os.path.join(self.datapath, 'sample_locations.json')) as f: location = Location(json.loads(f.read())[5]) ewsghana_location = self.api_object.location_sync(location) self.assertEqual(1, SQLLocation.objects.filter( domain=TEST_DOMAIN, location_type__administrative=False).count() ) self.assertIsNotNone(ewsghana_location.linked_supply_point()) self.assertIsNotNone(ewsghana_location.sql_location.supply_point_id) self.assertEqual(ewsghana_location.name, location.name) self.assertEqual(ewsghana_location.site_code, location.code) self.assertTrue(ewsghana_location.is_archived) def test_facility_with_inactive_and_active_supply_point(self): with open(os.path.join(self.datapath, 'sample_locations.json')) as f: location = Location(json.loads(f.read())[6]) ewsghana_location = self.api_object.location_sync(location) self.assertEqual("tsactive", ewsghana_location.site_code) self.assertEqual("Active Test hospital", ewsghana_location.name) self.assertFalse(ewsghana_location.is_archived)