def test_aws_errors(self):
"""Test AWS error types."""
test_matrix = [
{
"key": ProviderErrors.AWS_ROLE_ARN_UNREACHABLE,
"internal_message": "internal resource name message string",
"expected_message": ProviderErrors.AWS_ROLE_ARN_UNREACHABLE_MESSAGE,
},
{
"key": ProviderErrors.AWS_BILLING_SOURCE_NOT_FOUND,
"internal_message": "internal billing source message string",
"expected_message": ProviderErrors.AWS_BILLING_SOURCE_NOT_FOUND_MESSAGE,
},
{
"key": ProviderErrors.AWS_COMPRESSION_REPORT_CONFIG,
"internal_message": "internal compression error message",
"expected_message": ProviderErrors.AWS_COMPRESSION_REPORT_CONFIG_MESSAGE,
},
{
"key": ProviderErrors.AWS_BUCKET_MISSING,
"internal_message": ProviderErrors.AWS_BUCKET_MISSING_MESSAGE,
"expected_message": ProviderErrors.AWS_BUCKET_MISSING_MESSAGE,
},
]
for test in test_matrix:
with self.subTest(test=test):
key = test.get("key")
message = test.get("internal_message")
error = ValidationError(error_obj(key, message))
message_obj = SourcesErrorMessage(error)
self.assertEquals(message_obj.display(source_id=1), test.get("expected_message"))
def test_azure_errors(self):
"""Test Azure error types."""
test_matrix = [
{
"key": ProviderErrors.AZURE_CLIENT_ERROR,
"internal_message": ", AdalError: Get Token request returned http error: 401 and server response:",
"expected_message": ProviderErrors.AZURE_INCORRECT_CLIENT_SECRET_MESSAGE,
},
{
"key": ProviderErrors.AZURE_CLIENT_ERROR,
"internal_message": ", AdalError: Get Token request returned http error: 400 and server response:",
"expected_message": ProviderErrors.AZURE_INCORRECT_CLIENT_ID_MESSAGE,
},
{
"key": ProviderErrors.AZURE_CLIENT_ERROR,
"internal_message": (
"Azure Error: ResourceGroupNotFound\nMessage: Resource group" "'RG2' could not be found."
),
"expected_message": ProviderErrors.AZURE_INCORRECT_RESOURCE_GROUP_MESSAGE,
},
{
"key": ProviderErrors.AZURE_CLIENT_ERROR,
"internal_message": (
"Azure Error: ResourceNotFound\nMessage: The "
"Resource 'Microsoft.Storage/storageAccounts/mysa5' under "
"resource group 'RG1' was not found"
),
"expected_message": ProviderErrors.AZURE_INCORRECT_STORAGE_ACCOUNT_MESSAGE,
},
{
"key": ProviderErrors.AZURE_CLIENT_ERROR,
"internal_message": (
"Azure Error: SubscriptionNotFound\nMessage: The "
"subscription '2639de71-ca37-4a17-a104-17665a50e7fd'"
" could not be found."
),
"expected_message": ProviderErrors.AZURE_INCORRECT_SUBSCRIPTION_ID_MESSAGE,
},
{
"key": ProviderErrors.AZURE_CLIENT_ERROR,
"internal_message": "Random azure error",
"expected_message": ProviderErrors.AZURE_GENERAL_CLIENT_ERROR_MESSAGE,
},
]
for test in test_matrix:
key = test.get("key")
message = test.get("internal_message")
error = ValidationError(error_obj(key, message))
message_obj = SourcesErrorMessage(error)
self.assertEquals(message_obj.display(source_id=1), test.get("expected_message"))
def build_source_status(self, error_obj):
"""
Format the availability status for a source.
Connectivity and account validation checks are performed to
ensure that Koku can access a cost usage report from the provider.
This method will return the detailed error message in the event that
the provider fails the service provider checks in a format that
the platform is expecting.
Args:
error_obj (Object): ValidationError or String
Returns:
status (Dict): {'availability_status': 'unavailable/available',
'availability_status_error': 'User facing String'}
"""
if error_obj:
status = "unavailable"
else:
status = "available"
error_obj = ""
user_facing_string = SourcesErrorMessage(error_obj).display(self._source_id)
return {"availability_status": status, "availability_status_error": user_facing_string}
def test_available_source(self):
"""Test an available source message."""
message_obj = SourcesErrorMessage(None).display(source_id=1)
self.assertEquals(message_obj, "")
def test_general_string_error(self):
"""Test general string error fallback."""
random_error_dict = {"rando": "error"}
message_obj = SourcesErrorMessage(random_error_dict)
self.assertEquals(message_obj.display(source_id=1), str(random_error_dict))