def test_delete_regions_succes(self): logic = customer_logic.CustomerLogic() logic.delete_region('customer_id', 'region_id', 'transaction_is') assert record_mock.delete_region_for_customer.called assert data_manager_mock.commit.called assert customer_logic.RdsProxy.send_customer_dict.called
def test_update_customer_success(self): logic = customer_logic.CustomerLogic() logic.update_customer(customer, 'some_uuid', 'some_trans_id') assert record_mock.delete_by_primary_key.called assert data_manager_mock.commit.called assert not data_manager_mock.rollback.called
def test_add_users_success(self): logic = customer_logic.CustomerLogic() users = [models.User(), models.User()] logic.add_users('some_uuid', 'some_region', users, 'some_transaction') assert data_manager_mock.add_user.call_count == 2 assert data_manager_mock.commit.called
def test_get_customer_success(self): logic = customer_logic.CustomerLogic() get_mock = mock.MagicMock() get_mock.json.return_value = STATUS_JSON customer_logic.requests.get = mock.MagicMock(return_value=get_mock) logic.get_customer('customer_id') self.assertTrue(data_manager_mock.get_cusomer_by_uuid_or_name.called)
def test_add_regions_success(self): regions = [models.Region(), models.Region()] logic = customer_logic.CustomerLogic() logic.add_regions('customer_uuid', regions, 'transaction_id') assert data_manager_mock.commit.called assert customer_logic.RdsProxy.send_customer_dict.called
def test_get_customer_not_found(self): global flow_type flow_type = 1 # customer_logic.requests.get = mock.MagicMock(return_value=None) logic = customer_logic.CustomerLogic() self.assertRaises(ErrorStatus, logic.get_customer, 'id')
def test_create_customer_fail_rollback(self): global mock_returns_error mock_returns_error = True logic = customer_logic.CustomerLogic() self.assertRaises(SystemError, logic.create_customer, customer, 'some_uuid', 'some_trans_id')
def test_delete_regions_error(self): global mock_returns_error mock_returns_error = True logic = customer_logic.CustomerLogic() self.assertRaises(SystemError, logic.delete_region, 'customer_id', 'region_id', 'transaction_is') assert data_manager_mock.rollback.called
def test_replace_regions_fail(self): global mock_returns_error mock_returns_error = True regions = [models.Region(), models.Region()] logic = customer_logic.CustomerLogic() self.assertRaises(SystemError, logic.replace_regions, 'customer_uuid', regions, 'transaction_id') assert data_manager_mock.rollback.called
def test_replace_users_error(self): global mock_returns_error mock_returns_error = True logic = customer_logic.CustomerLogic() users = [models.User()] self.assertRaises(SystemError, logic.replace_users, 'id', 'region', users, 'trans_id')
def test_replace_regions_success(self): regions = [models.Region(), models.Region()] logic = customer_logic.CustomerLogic() logic.replace_regions('customer_uuid', regions, 'transaction_id') assert data_manager_mock.commit.called assert customer_logic.RdsProxy.send_customer_dict.called assert record_mock.delete_all_regions_for_customer.called
def test_enable_error(self): global mock_returns_error mock_returns_error = True logic = customer_logic.CustomerLogic() self.assertRaises(SystemError, logic.enable, 'id', models.Enabled(True), 'trans_id') self.assertTrue(data_manager_mock.rollback.called)
def test_delete_default_users_fail_notfound(self): global rowcount rowcount = 0 logic = customer_logic.CustomerLogic() with self.assertRaises(customer_logic.NotFound): logic.delete_default_users('customer_id', 'user_id', 'transaction_is') rowcount = 1 assert record_mock.delete_user_from_region.called assert data_manager_mock.rollback.called
def test_create_customer_success_with_regions(self): customer.regions = [models.Region(name="a")] logic = customer_logic.CustomerLogic() logic.create_customer(customer, 'some_uuid', 'some_trans_id') # sql_customer, trans_id = customer_logic.RdsProxy.send_customer.call_args_list[0][0] # assert trans_id is 'some_trans_id' # assert type(sql_customer) is sql_models.Customer assert data_manager_mock.commit.called assert not data_manager_mock.rollback.called
def test_replace_default_users_fail(self): global mock_returns_error mock_returns_error = True users = [models.User(), models.User()] logic = customer_logic.CustomerLogic() self.assertRaises(SystemError, logic.replace_default_users, 'id', users, 'trans_id') assert data_manager_mock.rollback.called
def test_replace_users_success(self): logic = customer_logic.CustomerLogic() users = [models.User(), models.User()] logic.replace_users('some_uuid', 'some_region', users, 'some_transaction') assert record_mock.delete_all_users_from_region.called assert data_manager_mock.add_user.called assert data_manager_mock.commit.called
def test_add_default_users_with_regions_success(self): user = models.User() user.role = ['user', 'admin'] users = [user, models.User()] logic = customer_logic.CustomerLogic() logic.add_default_users('customer_uuid', users, 'transaction_id') assert data_manager_mock.commit.called assert data_manager_mock.add_user.call_count == 2 assert data_manager_mock.add_role.call_count == 2
def test_delete_customer_by_uuid_success(self): logic = customer_logic.CustomerLogic() logic.delete_customer_by_uuid('customer_id') # Customer found in CMS DB but not found in RDS customer_logic.RdsProxy.get_status.return_value = RdsStatus( status_code=404) logic.delete_customer_by_uuid('customer_id') global flow_type # Change the flow to "customer not found in CMS DB" flow_type = 1 logic.delete_customer_by_uuid('customer_id')
def test_replace_default_users_no_regions_success(self): user = models.User() user.role = ['user', 'admin'] users = [user, models.User()] logic = customer_logic.CustomerLogic() logic.replace_default_users('customer_uuid', users, 'transaction_id') assert data_manager_mock.commit.called assert record_mock.delete_all_users_from_region.called assert not customer_logic.RdsProxy.send_customer.called assert data_manager_mock.add_user.call_count == 2 assert data_manager_mock.add_role.call_count == 2
def test_add_customer_with_default_users(self): default_quota = models.Quota() customer.defaultQuotas = [default_quota] default_region = models.Region() customer.regions = [default_region] default_user = models.User() default_user.role = ['user', 'admin'] default_region.users = [default_user] logic = customer_logic.CustomerLogic() logic.create_customer(customer, 'some_uuid', 'some_trans_id') assert data_manager_mock.add_user.called
def test_delete_customer_by_uuid_errors(self): global mock_returns_error mock_returns_error = True logic = customer_logic.CustomerLogic() self.assertRaises(SystemError, logic.delete_customer_by_uuid, 'customer_id') # RDS returned an empty json mock_returns_error = False customer_logic.RdsProxy.get_status.return_value = RdsStatus(oy=True) self.assertRaises(customer_logic.ErrorStatus, logic.delete_customer_by_uuid, 'customer_id') # RDS returned 500 customer_logic.RdsProxy.get_status.return_value = RdsStatus( status_code=500) self.assertRaises(customer_logic.ErrorStatus, logic.delete_customer_by_uuid, 'customer_id') # RDS returned Error status customer_logic.RdsProxy.get_status.return_value = RdsStatus( status='Error') self.assertRaises(customer_logic.ErrorStatus, logic.delete_customer_by_uuid, 'customer_id')
def test_delete_default_users_succes(self): logic = customer_logic.CustomerLogic() logic.delete_default_users('customer_id', 'user_id', 'transaction_is') assert record_mock.delete_user_from_region.called assert data_manager_mock.commit.called
def test_create_customer_add_all_default_users(self): logic = customer_logic.CustomerLogic() logic.create_customer(customer, 'some_uuid', 'some_trans_id') assert data_manager_mock.add_user.call_count == 2
def test_add_regions_action(self): regions = [models.Region(), models.Region()] logic = customer_logic.CustomerLogic() logic.add_regions('some_uuid', regions, 'some_trans_id') res = customer_logic.RdsProxy.send_customer.call_args_list
def test_get_customer_list_by_criteria(self): logic = customer_logic.CustomerLogic() result = logic.get_customer_list_by_criteria(None, None, None, None, {"key:value"})
def test_delete_customer_by_uuid_conflict(self): global flow_type flow_type = 2 logic = customer_logic.CustomerLogic() self.assertRaises(customer_logic.ErrorStatus, logic.delete_customer_by_uuid, 'customer_id')
def test_enable_success(self): logic = customer_logic.CustomerLogic() logic.enable('customer_id', models.Enabled(True), 'transaction_is') self.assertTrue(record_mock.read_customer_by_uuid.called) self.assertTrue(customer_logic.RdsProxy.send_customer.called)