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_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")