def test_retrieves_aws_credentials_from_botocore(self, botocore_mock):
credentials_mock = mock.Mock(
token='token',
secret_key='secret_key',
access_key='access_key',
)
botocore_mock.get_session().get_credentials.return_value = credentials_mock
credentials = get_aws_credentials()
botocore_mock.get_session().get_credentials.assert_called_once_with()
self.assertEqual(credentials.token, 'token')
self.assertEqual(credentials.secret_key, 'secret_key')
self.assertEqual(credentials.access_key, 'access_key')
def test_retrieves_aws_credentials_from_botocore(self, fetcher_mock,
provider_mock):
credentials_mock = mock.Mock(
token='token',
secret_key='secret_key',
access_key='access_key',
)
aws_response_mock = mock.Mock()
aws_response_mock.load.return_value = credentials_mock
fetcher_mock.return_value = 'metadata'
provider_mock.return_value = aws_response_mock
credentials = get_aws_credentials()
provider_mock.assert_called_once_with(iam_role_fetcher='metadata')
self.assertEqual(credentials.token, 'token')
self.assertEqual(credentials.secret_key, 'secret_key')
self.assertEqual(credentials.access_key, 'access_key')
def test_retrieves_aws_credentials_from_botocore(self, fetcher_mock,
provider_mock):
credentials_mock = mock.Mock(
token='token',
secret_key='secret_key',
access_key='access_key',
)
aws_response_mock = mock.Mock()
aws_response_mock.load.return_value = credentials_mock
fetcher_mock.return_value = 'metadata'
provider_mock.return_value = aws_response_mock
credentials = get_aws_credentials()
provider_mock.assert_called_once_with(iam_role_fetcher='metadata')
self.assertEqual(credentials.token, 'token')
self.assertEqual(credentials.secret_key, 'secret_key')
self.assertEqual(credentials.access_key, 'access_key')
def test_retrieves_aws_credentials_from_django_config(self):
credentials = get_aws_credentials()
self.assertIsNone(credentials.token)
self.assertEqual(credentials.secret_key, 'local_secret_key')
self.assertEqual(credentials.access_key, 'local_access_key')
def test_retrieves_aws_credentials_from_django_config(self):
credentials = get_aws_credentials()
self.assertIsNone(credentials.token)
self.assertEqual(credentials.secret_key, 'local_secret_key')
self.assertEqual(credentials.access_key, 'local_access_key')