def test_checkout_exception(self):
"""Test exception from create IAM account is handled properly."""
mocker = AWSMock()
self.user.username = fake_user_name
self.user.save()
self.c.login(username=self.user.username, password='******')
# Test create IAM account exception handled
with patch('botocore.client.BaseClient._make_api_call',
new=mocker.mock_make_api_call):
with patch.object(AmazonAccountUtils,
'_cleanup_iam_user',
return_value=True):
response = self.c.get(reverse('library:checkout',
args=['amazondemoaccount']),
follow=True)
# Confirm error message displayed
message = list(response.context['messages'])[0].message
self.assertEqual(message,
"An error occurred (409) when calling the "
"CreateUser operation: User fakeuser exists")
# Confirm lendable not created
self.assertEqual(Lendable.all_types.count(), 0)
def test_destroy_account_not_exist(self):
mocker = AWSMock()
with patch('botocore.client.BaseClient._make_api_call',
new=mocker.mock_make_api_call):
amazon_account_utils = AmazonAccountUtils('43543253245',
'6543654rfdfds')
result = amazon_account_utils.destroy_iam_account('john')
self.assertFalse(result)
def test_checkout_group_exception(self):
mocker = AWSMock()
self.c.login(username=self.user.username, password='******')
# Test create IAM account group not found exception handled
with patch('botocore.client.BaseClient._make_api_call',
new=mocker.mock_make_api_call):
with self.settings(AWS_IAM_GROUPS=['Admins'],
AWS_ACCOUNT_ID_ALIAS=None):
response = self.c.get(reverse('library:checkout',
args=['amazondemoaccount']),
follow=True)
# Confirm error message displayed
message = list(response.context['messages'])[0].message
self.assertEqual(message,
"An error occurred (Not Found) when calling the "
"AddUserToGroup operation: Group Admins not found")
# Confirm lendable not created
self.assertEqual(Lendable.all_types.count(), 0)
def test_aws_checkout(self):
"""Test AWS IAM checkout and checkin.
Mock cleanup method with a return value of True.
"""
self.c.login(username=self.user.username, password='******')
mocker = AWSMock()
mocker.create_group({'GroupName': 'Admins'})
# Checkout AWS account
with self.settings(AWS_ACCOUNT_ID_ALIAS='John.Wayne',
AWS_IAM_GROUPS=['Admins']):
with patch('botocore.client.BaseClient._make_api_call',
new=mocker.mock_make_api_call):
# Test check out AWS lendable
response = self.c.get(reverse('library:checkout',
args=['amazondemoaccount']),
follow=True)
# Get lendable and parse due date string
aws_lendable = self.user.lendable_set.first()
date_str = aws_lendable.due_on.strftime("%-d %b %Y").lower()
# Confirm sucess message displayed
message = list(response.context['messages'])[0].message
self.assertEqual("'Amazon Web Services - Demo Account' "
"is checked out to you until %s." % date_str,
message)
# Confirm credential values displayed in modal
self.assertContains(response, '<code>John</code>')
self.assertContains(response,
'https://John.Wayne.signin.aws.amazon.com/console')
# Confirm lendable created
self.assertEqual(Lendable.all_types.count(), 1)
# Test user cannot checkout AWS account twice
response = self.c.get(reverse('library:checkout',
args=['amazondemoaccount']),
follow=True)
# Confirm error message displayed
message = list(response.context['messages'])[0].message
self.assertEqual(message,
"Amazon Web Services - Demo Account "
"unavailable for checkout.")
# Confirm lendable not created
self.assertEqual(Lendable.all_types.count(), 1)
# Get lendable primary key
pk = self.user.lendable_set.first().id
# Test checkin handles generic exception
with patch.object(AmazonDemoAccount,
'checkin',
side_effect=Exception('Checkin Failed!')):
response = self.c.get(reverse('library:checkin',
args=[pk]),
follow=True)
# Confirm error message displayed
message = list(response.context['messages'])[0].message
self.assertEqual(message, 'Checkin Failed!')
# Checkin AWS account
with patch('botocore.client.BaseClient._make_api_call',
new=mocker.mock_make_api_call):
# Test check in AWS lendable
response = self.c.get(reverse('library:checkin',
args=[pk]),
follow=True)
# Confirm success message displayed
message = list(response.context['messages'])[0].message
self.assertEqual("'Amazon Web Services - Demo Account' returned.",
message)
# Confirm lendable checked in and now has a check in date
self.assertEqual(Lendable.all_types.count(), 0)
self.assertIsNotNone(Lendable.all_lendables.first().checked_in_on)
mocker.delete_group({'GroupName': 'Admins'})
def setUp(self):
"""Setup test case."""
self.mocker = AWSMock()
self.kwarg = {username: '******', group_name: 'Admins'}
