def test_is_valid_disabled(self):
"""Test when is_valid() should be False when status is DISABLED."""
status = {
'provider_id': self.provider_uuid,
'status': ProviderStatusCode.DISABLED_ERROR,
'last_message': self.FAKE.word(),
'timestamp': DateAccessor().today(),
'retries': 3,
}
with ProviderStatus(self.aws_provider_uuid) as accessor:
accessor.add(**status)
accessor = ProviderStatus(self.aws_provider_uuid)
self.assertFalse(accessor.is_valid())
accessor.close_session()
def test_is_valid_warn(self):
"""Test is_valid() should be True when status is WARNING."""
status = {
'provider_id': self.provider_uuid,
'status': ProviderStatusCode.WARNING,
'last_message': self.FAKE.word(),
'timestamp': DateAccessor().today(),
'retries': 3,
}
with ProviderStatus(self.aws_provider_uuid) as accessor:
accessor.add(**status)
accessor = ProviderStatus(self.aws_provider_uuid)
self.assertTrue(accessor.is_valid())
accessor.close_session()
def test_set_error(self):
"""Test set_error()."""
# set status to READY state.
self._setup_ready_status()
# log an error
accessor = ProviderStatus(self.aws_provider_uuid)
err = Exception(self.FAKE.word())
accessor.set_error(error=err)
accessor.close_session()
# test that state moved from READY to WARNING
with ProviderStatus(self.aws_provider_uuid) as new_accessor:
self.assertEqual(new_accessor.get_status(),
ProviderStatusCode.WARNING)
self.assertEqual(new_accessor.get_last_message(), str(err))
self.assertEqual(new_accessor.get_retries(), 1)
