def update_billing_source(self, source_id, billing_source):
"""Store billing source update."""
instance = storage.get_source(source_id, "Unable to PATCH", LOG.error)
instance.billing_source = billing_source
if instance.source_uuid:
instance.pending_update = True
instance.save()
return True
def update_authentication(self, source_id, authentication):
"""Store authentication update."""
instance = storage.get_source(source_id, "Unable to PATCH", LOG.error)
instance.authentication = authentication
if instance.source_uuid:
instance.pending_update = True
instance.save()
return True
def test_get_source_db_down(self, mock_objects):
"""Tests creating a source db record with invalid auth_header."""
mock_objects.get.side_effect = InterfaceError("test_exception")
test_source_id = 2
with self.assertRaises(InterfaceError):
storage.get_source(test_source_id, "error", Mock)
