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)