class WebUsersSyncTest(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() self.api_object.create_or_edit_roles() for user in WebUser.by_domain(TEST_DOMAIN): user.delete() def test_create_webuser(self): with open(os.path.join(self.datapath, 'sample_webusers.json')) as f: webuser = EWSUser(json.loads(f.read())[0]) self.assertEqual(0, len(WebUser.by_domain(TEST_DOMAIN))) ewsghana_webuser = self.api_object.web_user_sync(webuser) self.assertEqual(webuser.email, ewsghana_webuser.username) self.assertEqual(webuser.password, ewsghana_webuser.password) self.assertEqual(webuser.first_name, ewsghana_webuser.first_name) self.assertEqual(webuser.last_name, ewsghana_webuser.last_name) self.assertEqual(webuser.is_active, ewsghana_webuser.is_active) self.assertEqual(False, ewsghana_webuser.is_superuser) self.assertEqual(False, ewsghana_webuser.is_staff) domain_name = ewsghana_webuser.get_domains()[0] self.assertEqual(TEST_DOMAIN, domain_name) self.assertEqual(UserRole.get_read_only_role_by_domain(TEST_DOMAIN)._id, ewsghana_webuser.get_domain_membership(TEST_DOMAIN).role_id) def test_create_facility_manager(self): with open(os.path.join(self.datapath, 'sample_webusers.json')) as f: webuser = EWSUser(json.loads(f.read())[1]) self.assertEqual(0, len(WebUser.by_domain(TEST_DOMAIN))) self.api_object.web_user_sync(webuser) web_users = list(WebUser.by_domain(TEST_DOMAIN)) self.assertEqual(1, len(web_users)) self.assertEqual(0, len(CommCareUser.by_domain(TEST_DOMAIN))) facility_manager_role = UserRole.by_domain_and_name(TEST_DOMAIN, 'Facility manager')[0] dm = web_users[0].get_domain_membership(TEST_DOMAIN) self.assertEqual(facility_manager_role.get_id, dm.role_id)
class SMSUsersSyncTest(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() self.api_object.create_or_edit_roles() for user in CommCareUser.by_domain(TEST_DOMAIN): user.delete() for verified_number in VerifiedNumber.by_domain(TEST_DOMAIN): verified_number.delete() def test_create_smsuser(self): with open(os.path.join(self.datapath, 'sample_smsusers.json')) as f: smsuser = SMSUser(json.loads(f.read())[0]) self.assertEqual(0, len(CommCareUser.by_domain(TEST_DOMAIN))) ewsghana_smsuser = self.api_object.sms_user_sync(smsuser) self.assertIsNotNone(ewsghana_smsuser.get_id) username_part = "%s%d" % (ewsghana_smsuser.name.strip().replace(' ', '.').lower(), smsuser.id) username = "******" % (username_part, TEST_DOMAIN) self.assertEqual(username, ewsghana_smsuser.username) self.assertEqual(smsuser.is_active, str(ewsghana_smsuser.is_active)) self.assertEqual(False, ewsghana_smsuser.is_superuser) self.assertEqual(False, ewsghana_smsuser.is_staff) verified_number = ewsghana_smsuser.get_verified_number() self.assertIsNotNone(verified_number) self.assertEqual(verified_number.phone_number, '2222222222') domain_name = ewsghana_smsuser.get_domains()[0] self.assertEqual(TEST_DOMAIN, domain_name) self.assertIsInstance(ewsghana_smsuser.user_data['role'], list) self.assertEqual(len(ewsghana_smsuser.user_data['role']), 1) self.assertEqual(ewsghana_smsuser.user_data['role'][0], 'Other') def test_edit_smsuser(self): with open(os.path.join(self.datapath, 'sample_smsusers.json')) as f: smsuser = SMSUser(json.loads(f.read())[0]) self.assertEqual(0, len(CommCareUser.by_domain(TEST_DOMAIN))) ewsghana_smsuser = self.api_object.sms_user_sync(smsuser) verified_number = ewsghana_smsuser.get_verified_number() self.assertIsNotNone(verified_number) user_id = ewsghana_smsuser.get_id self.assertIsNotNone(user_id) smsuser.phone_numbers = ['111111111'] ewsghana_smsuser = self.api_object.sms_user_sync(smsuser) self.assertEqual(user_id, ewsghana_smsuser.get_id) self.assertEqual(ewsghana_smsuser.default_phone_number, '111111111') self.assertListEqual(ewsghana_smsuser.phone_numbers, ['111111111']) verified_number = ewsghana_smsuser.get_verified_number() self.assertIsNotNone(verified_number) self.assertEqual(verified_number.phone_number, '111111111') def test_edit_phone_number1(self): """ When phone number is deleted on EWS side it also should be deleted in HQ :return: """ with open(os.path.join(self.datapath, 'sample_smsusers.json')) as f: smsuser = SMSUser(json.loads(f.read())[0]) self.assertEqual(0, len(CommCareUser.by_domain(TEST_DOMAIN))) ewsghana_smsuser = self.api_object.sms_user_sync(smsuser) verified_number = ewsghana_smsuser.get_verified_number() self.assertIsNotNone(verified_number) smsuser.phone_numbers = [] ewsghana_smsuser = self.api_object.sms_user_sync(smsuser) self.assertIsNone(ewsghana_smsuser.default_phone_number) self.assertListEqual(ewsghana_smsuser.phone_numbers, []) verified_number = ewsghana_smsuser.get_verified_number() self.assertIsNone(verified_number) def test_edit_phone_number2(self): """ When phone number is added on EWS side it also should be added in HQ """ with open(os.path.join(self.datapath, 'sample_smsusers.json')) as f: smsuser = SMSUser(json.loads(f.read())[0]) smsuser.phone_numbers = [] self.assertEqual(0, len(CommCareUser.by_domain(TEST_DOMAIN))) ewsghana_smsuser = self.api_object.sms_user_sync(smsuser) verified_number = ewsghana_smsuser.get_verified_number() self.assertIsNone(verified_number) smsuser.phone_numbers = ['111111111'] ewsghana_smsuser = self.api_object.sms_user_sync(smsuser) self.assertIsNotNone(ewsghana_smsuser.default_phone_number) self.assertListEqual(ewsghana_smsuser.phone_numbers, ['111111111']) verified_number = ewsghana_smsuser.get_verified_number() self.assertIsNotNone(verified_number) self.assertEqual(verified_number.phone_number, '111111111') def test_edit_phone_number3(self): """ When phone number is edited on EWS side it also should be edited in HQ """ with open(os.path.join(self.datapath, 'sample_smsusers.json')) as f: smsuser = SMSUser(json.loads(f.read())[0]) self.assertEqual(0, len(CommCareUser.by_domain(TEST_DOMAIN))) ewsghana_smsuser = self.api_object.sms_user_sync(smsuser) verified_number = ewsghana_smsuser.get_verified_number() self.assertIsNotNone(verified_number) self.assertEqual(verified_number.phone_number, '2222222222') smsuser.phone_numbers = ['111111111'] ewsghana_smsuser = self.api_object.sms_user_sync(smsuser) self.assertIsNotNone(ewsghana_smsuser.default_phone_number) self.assertListEqual(ewsghana_smsuser.phone_numbers, ['111111111']) verified_number = ewsghana_smsuser.get_verified_number() self.assertIsNotNone(verified_number) self.assertEqual(verified_number.phone_number, '111111111') def test_edit_phone_number4(self): """ Number shouldn't be changed when is not edited on EWS side. """ with open(os.path.join(self.datapath, 'sample_smsusers.json')) as f: smsuser = SMSUser(json.loads(f.read())[0]) self.assertEqual(0, len(CommCareUser.by_domain(TEST_DOMAIN))) ewsghana_smsuser = self.api_object.sms_user_sync(smsuser) verified_number = ewsghana_smsuser.get_verified_number() self.assertIsNotNone(verified_number) self.assertEqual(verified_number.phone_number, '2222222222') ewsghana_smsuser = self.api_object.sms_user_sync(smsuser) verified_number = ewsghana_smsuser.get_verified_number() self.assertIsNotNone(verified_number) self.assertEqual(verified_number.phone_number, '2222222222')
class SMSUsersSyncTest(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() self.api_object.create_or_edit_roles() for user in CommCareUser.by_domain(TEST_DOMAIN): user.delete() for verified_number in VerifiedNumber.by_domain(TEST_DOMAIN): verified_number.delete() def test_create_smsuser(self): with open(os.path.join(self.datapath, 'sample_smsusers.json')) as f: smsuser = SMSUser(json.loads(f.read())[0]) self.assertEqual(0, len(CommCareUser.by_domain(TEST_DOMAIN))) ewsghana_smsuser = self.api_object.sms_user_sync(smsuser) self.assertIsNotNone(ewsghana_smsuser.get_id) username_part = "%s%d" % (ewsghana_smsuser.name.strip().replace( ' ', '.').lower(), smsuser.id) username = "******" % (username_part, TEST_DOMAIN) self.assertEqual(username, ewsghana_smsuser.username) self.assertEqual(smsuser.is_active, str(ewsghana_smsuser.is_active)) self.assertEqual(False, ewsghana_smsuser.is_superuser) self.assertEqual(False, ewsghana_smsuser.is_staff) verified_number = ewsghana_smsuser.get_verified_number() self.assertIsNotNone(verified_number) self.assertEqual(verified_number.phone_number, '2222222222') domain_name = ewsghana_smsuser.get_domains()[0] self.assertEqual(TEST_DOMAIN, domain_name) self.assertIsInstance(ewsghana_smsuser.user_data['role'], list) self.assertEqual(len(ewsghana_smsuser.user_data['role']), 1) self.assertEqual(ewsghana_smsuser.user_data['role'][0], 'Other') def test_edit_smsuser(self): with open(os.path.join(self.datapath, 'sample_smsusers.json')) as f: smsuser = SMSUser(json.loads(f.read())[0]) self.assertEqual(0, len(CommCareUser.by_domain(TEST_DOMAIN))) ewsghana_smsuser = self.api_object.sms_user_sync(smsuser) verified_number = ewsghana_smsuser.get_verified_number() self.assertIsNotNone(verified_number) user_id = ewsghana_smsuser.get_id self.assertIsNotNone(user_id) smsuser.phone_numbers = ['111111111'] ewsghana_smsuser = self.api_object.sms_user_sync(smsuser) self.assertEqual(user_id, ewsghana_smsuser.get_id) self.assertEqual(ewsghana_smsuser.default_phone_number, '111111111') self.assertListEqual(ewsghana_smsuser.phone_numbers, ['111111111']) verified_number = ewsghana_smsuser.get_verified_number() self.assertIsNotNone(verified_number) self.assertEqual(verified_number.phone_number, '111111111') def test_edit_phone_number1(self): """ When phone number is deleted on EWS side it also should be deleted in HQ :return: """ with open(os.path.join(self.datapath, 'sample_smsusers.json')) as f: smsuser = SMSUser(json.loads(f.read())[0]) self.assertEqual(0, len(CommCareUser.by_domain(TEST_DOMAIN))) ewsghana_smsuser = self.api_object.sms_user_sync(smsuser) verified_number = ewsghana_smsuser.get_verified_number() self.assertIsNotNone(verified_number) smsuser.phone_numbers = [] ewsghana_smsuser = self.api_object.sms_user_sync(smsuser) self.assertIsNone(ewsghana_smsuser.default_phone_number) self.assertListEqual(ewsghana_smsuser.phone_numbers, []) verified_number = ewsghana_smsuser.get_verified_number() self.assertIsNone(verified_number) def test_edit_phone_number2(self): """ When phone number is added on EWS side it also should be added in HQ """ with open(os.path.join(self.datapath, 'sample_smsusers.json')) as f: smsuser = SMSUser(json.loads(f.read())[0]) smsuser.phone_numbers = [] self.assertEqual(0, len(CommCareUser.by_domain(TEST_DOMAIN))) ewsghana_smsuser = self.api_object.sms_user_sync(smsuser) verified_number = ewsghana_smsuser.get_verified_number() self.assertIsNone(verified_number) smsuser.phone_numbers = ['111111111'] ewsghana_smsuser = self.api_object.sms_user_sync(smsuser) self.assertIsNotNone(ewsghana_smsuser.default_phone_number) self.assertListEqual(ewsghana_smsuser.phone_numbers, ['111111111']) verified_number = ewsghana_smsuser.get_verified_number() self.assertIsNotNone(verified_number) self.assertEqual(verified_number.phone_number, '111111111') def test_edit_phone_number3(self): """ When phone number is edited on EWS side it also should be edited in HQ """ with open(os.path.join(self.datapath, 'sample_smsusers.json')) as f: smsuser = SMSUser(json.loads(f.read())[0]) self.assertEqual(0, len(CommCareUser.by_domain(TEST_DOMAIN))) ewsghana_smsuser = self.api_object.sms_user_sync(smsuser) verified_number = ewsghana_smsuser.get_verified_number() self.assertIsNotNone(verified_number) self.assertEqual(verified_number.phone_number, '2222222222') smsuser.phone_numbers = ['111111111'] ewsghana_smsuser = self.api_object.sms_user_sync(smsuser) self.assertIsNotNone(ewsghana_smsuser.default_phone_number) self.assertListEqual(ewsghana_smsuser.phone_numbers, ['111111111']) verified_number = ewsghana_smsuser.get_verified_number() self.assertIsNotNone(verified_number) self.assertEqual(verified_number.phone_number, '111111111') def test_edit_phone_number4(self): """ Number shouldn't be changed when is not edited on EWS side. """ with open(os.path.join(self.datapath, 'sample_smsusers.json')) as f: smsuser = SMSUser(json.loads(f.read())[0]) self.assertEqual(0, len(CommCareUser.by_domain(TEST_DOMAIN))) ewsghana_smsuser = self.api_object.sms_user_sync(smsuser) verified_number = ewsghana_smsuser.get_verified_number() self.assertIsNotNone(verified_number) self.assertEqual(verified_number.phone_number, '2222222222') ewsghana_smsuser = self.api_object.sms_user_sync(smsuser) verified_number = ewsghana_smsuser.get_verified_number() self.assertIsNotNone(verified_number) self.assertEqual(verified_number.phone_number, '2222222222')
class TestStockDataSync(TestCase): def setUp(self): self.endpoint = MockEndpoint('http://test-api.com/', 'dummy', 'dummy') self.stock_api_object = MockEWSStockDataSynchronization(TEST_DOMAIN, self.endpoint) self.datapath = os.path.join(os.path.dirname(__file__), 'data') initial_bootstrap(TEST_DOMAIN) self.api_object = EWSApi(TEST_DOMAIN, self.endpoint) self.api_object.prepare_commtrack_config() config = EWSGhanaConfig() config.domain = TEST_DOMAIN config.enabled = True config.all_stock_data = True config.password = '******' config.username = '******' config.url = 'http://test-api.com/' config.save() l1 = Location( name='Test location 1', external_id='3445', location_type='Hospital', domain=TEST_DOMAIN ) l2 = Location( name='Test location 2', external_id='4407', location_type='Hospital', domain=TEST_DOMAIN ) l1.save() l2.save() SupplyPointCase.create_from_location(TEST_DOMAIN, l1) SupplyPointCase.create_from_location(TEST_DOMAIN, l2) l1.save() l2.save() with open(os.path.join(self.datapath, 'sample_products.json')) as f: for product_json in json.loads(f.read()): self.api_object.product_sync(Product(product_json)) def test_stock_data_migration(self): stock_data_task(self.stock_api_object) self.assertEqual(StockTransaction.objects.filter(report__domain=TEST_DOMAIN).count(), 11) self.assertEqual(StockState.objects.all().count(), 8) self.assertEqual(StockTransaction.objects.filter(subtype='inferred').count(), 1) self.assertEqual(StockReport.objects.all().count(), 1) def test_stock_data_migration2(self): StockDataCheckpoint.objects.create( domain=TEST_DOMAIN, api='', limit=1000, offset=0, date=datetime.datetime(2015, 7, 1) ) stock_data_task(self.stock_api_object) self.assertEqual(StockTransaction.objects.filter(report__domain=TEST_DOMAIN).count(), 11) self.assertEqual(StockState.objects.all().count(), 8) self.assertEqual(StockTransaction.objects.filter(subtype='inferred').count(), 0) self.assertEqual(StockReport.objects.all().count(), 1) def tearDown(self): Domain.get_by_name(TEST_DOMAIN).delete()
class TestStockDataSync(TestCase): def setUp(self): self.endpoint = MockEndpoint('http://test-api.com/', 'dummy', 'dummy') self.stock_api_object = MockEWSStockDataSynchronization( TEST_DOMAIN, self.endpoint) self.datapath = os.path.join(os.path.dirname(__file__), 'data') initial_bootstrap(TEST_DOMAIN) self.api_object = EWSApi(TEST_DOMAIN, self.endpoint) self.api_object.prepare_commtrack_config() config = EWSGhanaConfig() config.domain = TEST_DOMAIN config.enabled = True config.all_stock_data = True config.password = '******' config.username = '******' config.url = 'http://test-api.com/' config.save() l1 = Location(name='Test location 1', external_id='3445', location_type='Hospital', domain=TEST_DOMAIN) l2 = Location(name='Test location 2', external_id='4407', location_type='Hospital', domain=TEST_DOMAIN) l1.save() l2.save() SupplyPointCase.create_from_location(TEST_DOMAIN, l1) SupplyPointCase.create_from_location(TEST_DOMAIN, l2) l1.save() l2.save() with open(os.path.join(self.datapath, 'sample_products.json')) as f: for product_json in json.loads(f.read()): self.api_object.product_sync(Product(product_json)) def test_stock_data_migration(self): stock_data_task(self.stock_api_object) self.assertEqual( StockTransaction.objects.filter( report__domain=TEST_DOMAIN).count(), 11) self.assertEqual(StockState.objects.all().count(), 8) self.assertEqual( StockTransaction.objects.filter(subtype='inferred').count(), 1) self.assertEqual(StockReport.objects.all().count(), 1) def test_stock_data_migration2(self): StockDataCheckpoint.objects.create(domain=TEST_DOMAIN, api='', limit=1000, offset=0, date=datetime.datetime(2015, 7, 1)) stock_data_task(self.stock_api_object) self.assertEqual( StockTransaction.objects.filter( report__domain=TEST_DOMAIN).count(), 11) self.assertEqual(StockState.objects.all().count(), 8) self.assertEqual( StockTransaction.objects.filter(subtype='inferred').count(), 0) self.assertEqual(StockReport.objects.all().count(), 1) def tearDown(self): Domain.get_by_name(TEST_DOMAIN).delete()
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)