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)
