def test_fetch_layer_uri_fails_with_no_creds(self):
lambda_client_mock = Mock()
lambda_client_mock.get_layer_version.side_effect = NoCredentialsError()
download_layers = LayerDownloader("/", ".", Mock(), lambda_client_mock)
layer = Mock()
layer.layer_arn = "arn"
layer.version = 1
with self.assertRaises(CredentialsRequired):
download_layers._fetch_layer_uri(layer=layer)
def test_fetch_layer_uri_fails_with_no_creds(self):
lambda_client_mock = Mock()
lambda_client_mock.get_layer_version.side_effect = NoCredentialsError()
download_layers = LayerDownloader("/", ".", lambda_client_mock)
layer = Mock()
layer.layer_arn = "arn"
layer.version = 1
with self.assertRaises(CredentialsRequired):
download_layers._fetch_layer_uri(layer=layer)
def test_fetch_layer_uri_re_raises_client_error(self):
lambda_client_mock = Mock()
lambda_client_mock.get_layer_version.side_effect = ClientError(
error_response={'Error': {'Code': 'Unknown'}}, operation_name="lambda")
download_layers = LayerDownloader("/", ".", lambda_client_mock)
layer = Mock()
layer.layer_arn = "arn"
layer.version = 1
with self.assertRaises(ClientError):
download_layers._fetch_layer_uri(layer=layer)
def test_fetch_layer_uri_fails_with_ResourceNotFoundException(self):
lambda_client_mock = Mock()
lambda_client_mock.get_layer_version.side_effect = ClientError(
error_response={'Error': {'Code': 'ResourceNotFoundException'}}, operation_name="lambda")
download_layers = LayerDownloader("/", ".", lambda_client_mock)
layer = Mock()
layer.layer_arn = "arn"
layer.version = 1
with self.assertRaises(ResourceNotFound):
download_layers._fetch_layer_uri(layer=layer)
def test_fetch_layer_uri_re_raises_client_error(self):
lambda_client_mock = Mock()
lambda_client_mock.get_layer_version.side_effect = ClientError(
error_response={'Error': {'Code': 'Unknown'}}, operation_name="lambda")
download_layers = LayerDownloader("/", ".", lambda_client_mock)
layer = Mock()
layer.layer_arn = "arn"
layer.version = 1
with self.assertRaises(ClientError):
download_layers._fetch_layer_uri(layer=layer)
def test_fetch_layer_uri_fails_with_ResourceNotFoundException(self):
lambda_client_mock = Mock()
lambda_client_mock.get_layer_version.side_effect = ClientError(
error_response={'Error': {'Code': 'ResourceNotFoundException'}}, operation_name="lambda")
download_layers = LayerDownloader("/", ".", lambda_client_mock)
layer = Mock()
layer.layer_arn = "arn"
layer.version = 1
with self.assertRaises(ResourceNotFound):
download_layers._fetch_layer_uri(layer=layer)
def test_fetch_layer_uri_fails_with_AccessDeniedException(self):
lambda_client_mock = Mock()
lambda_client_mock.get_layer_version.side_effect = ClientError(
error_response={"Error": {"Code": "AccessDeniedException"}}, operation_name="lambda"
)
download_layers = LayerDownloader("/", ".", Mock(), lambda_client_mock)
layer = Mock()
layer.layer_arn = "arn"
layer.version = 1
with self.assertRaises(CredentialsRequired):
download_layers._fetch_layer_uri(layer=layer)
def test_fetch_layer_uri_is_successful(self):
lambda_client_mock = Mock()
lambda_client_mock.get_layer_version.return_value = {"Content": {"Location": "some/uri"}}
download_layers = LayerDownloader("/", ".", lambda_client_mock)
layer = Mock()
layer.layer_arn = "arn"
layer.version = 1
actual_uri = download_layers._fetch_layer_uri(layer=layer)
self.assertEquals(actual_uri, "some/uri")
def test_fetch_layer_uri_is_successful(self):
lambda_client_mock = Mock()
lambda_client_mock.get_layer_version.return_value = {"Content": {"Location": "some/uri"}}
download_layers = LayerDownloader("/", ".", Mock(), lambda_client_mock)
layer = Mock()
layer.layer_arn = "arn"
layer.version = 1
actual_uri = download_layers._fetch_layer_uri(layer=layer)
self.assertEqual(actual_uri, "some/uri")