def update_charge_info(schema_name,
provider_uuid,
start_date=None,
end_date=None):
"""Update usage charge information.
Args:
schema_name (str) The DB schema name.
provider_uuid (str) The provider uuid.
start_date (str, Optional) - Start date of range to update derived cost.
end_date (str, Optional) - End date of range to update derived cost.
Returns
None
"""
worker_stats.CHARGE_UPDATE_ATTEMPTS_COUNTER.inc()
stmt = (f"update_charge_info called with args:\n"
f" schema_name: {schema_name},\n"
f" provider_uuid: {provider_uuid}")
LOG.info(stmt)
updater = ReportChargeUpdater(schema_name, provider_uuid)
updater.update_charge_info(start_date, end_date)
def test_ocp_local_route(self, mock_update):
"""Test that OCP-local charge updating works as expected."""
provider_ocp_uuid = '3c6e687e-1a09-4a05-970c-2ccf44b0952e'
updater = ReportChargeUpdater(self.schema, provider_ocp_uuid)
self.assertIsInstance(updater._updater, OCPReportChargeUpdater)
updater.update_charge_info()
mock_update.assert_called()
def test_aws_route(self, mock_update):
"""Test that AWS charge updating works as expected."""
provider_aws_uuid = '6e212746-484a-40cd-bba0-09a19d132d64'
updater = ReportChargeUpdater(self.schema, provider_aws_uuid)
self.assertIsInstance(updater._updater, AWSReportChargeUpdater)
updater.update_charge_info()
mock_update.assert_called()
def test_azure_route(self, mock_update):
"""Test that Azure charge updating works as expected."""
updater = ReportChargeUpdater(self.schema,
self.azure_test_provider_uuid)
self.assertIsInstance(updater._updater, AzureReportChargeUpdater)
updater.update_charge_info()
mock_update.assert_called()
def test_init_fail(self, mock_updater):
"""Test that an unimplemented provider throws an error."""
mock_updater.side_effect = Exception('general error')
provider_ocp_uuid = '3c6e687e-1a09-4a05-970c-2ccf44b0952e'
with self.assertRaises(ReportChargeUpdaterError):
ReportChargeUpdater(self.schema, provider_ocp_uuid)
def update_charge_info(schema_name, provider_uuid):
"""Update usage charge information.
Args:
schema_name (str) The DB schema name.
provider_uuid (str) The provider uuid.
Returns
None
"""
worker_stats.CHARGE_UPDATE_ATTEMPTS_COUNTER.inc()
stmt = ('update_charge_info called with args:\n'
' schema_name: {},\n'
' provider_uuid: {}')
stmt = stmt.format(schema_name, provider_uuid)
LOG.info(stmt)
updater = ReportChargeUpdater(schema_name, provider_uuid)
updater.update_charge_info()
def test_aws_route(self):
"""Test that AWS charge updating works as expected."""
provider_aws_uuid = '6e212746-484a-40cd-bba0-09a19d132d64'
updater = ReportChargeUpdater(self.schema, provider_aws_uuid)
self.assertIsNone(updater._updater)
def test_init_fail(self, mock_updater):
"""Test that an unimplemented provider throws an error."""
mock_updater.side_effect = Exception('general error')
with self.assertRaises(ReportChargeUpdaterError):
ReportChargeUpdater(self.schema, self.ocp_test_provider_uuid)
