def test_get_infrastructure_type_exception(self):
"""Get infrastructure type with exception."""
provider = OCPProvider()
interface = ProviderAccessor(provider.name())
with self.assertRaises(ProviderAccessorError):
with patch.object(OCPProvider,
'infra_type_implementation',
side_effect=Exception('test')):
interface.infrastructure_type(None, None)
def test_cost_usage_source_is_reachable_bucket_provided(self):
"""Verify that the cost usage source is authenticated and created."""
cluster_id = 'my-ocp-cluster-1'
report_source = 'report_location'
provider_interface = OCPProvider()
with self.assertRaises(ValidationError):
provider_interface.cost_usage_source_is_reachable(cluster_id, report_source)
def test_cost_usage_source_no_cluster_id(self):
"""Verify that the cost usage source raises error with no cluster_id."""
cluster_id = None
report_source = None
provider_interface = OCPProvider()
with self.assertRaises(ValidationError):
provider_interface.cost_usage_source_is_reachable(
cluster_id, report_source)
def test_cost_usage_source_is_reachable_no_bucket_provided(self):
"""Verify that the cost usage source is not authenticated and created with no bucket provided."""
cluster_id = 'my-ocp-cluster-1'
report_source = None
provider_interface = OCPProvider()
try:
provider_interface.cost_usage_source_is_reachable(cluster_id, report_source)
except Exception:
self.fail('Unexpected error ')
def _create_service(self, service_name):
"""
Create the provider service object.
This will establish what service (AWS, etc) ProviderAccessor should use
when interacting with Koku core.
Args:
service_name (String): Provider Type
Returns:
(Object) : Some object that is a child of ProviderInterface
"""
if service_name == 'AWS':
return AWSProvider()
elif service_name == 'AWS-local':
return AWSLocalProvider()
elif service_name == 'OCP-local':
return OCPLocalProvider()
elif service_name == 'OCP':
return OCPProvider()
def test_get_name(self):
"""Get name of provider."""
provider = OCPProvider()
self.assertEqual(provider.name(), 'OCP')
def test_get_name_ocp(self):
"""Get name of ocp service provider."""
provider = OCPProvider()
interface = ProviderAccessor(provider.name())
self.assertEqual(provider.name(), interface.service_name())