def _setup_random_status(self): """Set up a randomized status for testing. This is being done in a separate function instead of in setUp() to facilitate testing the case where there is no status in the DB. """ self.test_status = { 'provider_id': self.provider_id, 'status': random.choice(list(ProviderStatusCode)), 'last_message': self.FAKE.word(), 'retries': random.randint(0, 10), } with ProviderStatusAccessor(self.aws_test_provider_uuid) as accessor: status = accessor.add(**self.test_status) status.save() self.time_stamp = status.timestamp
def _setup_random_status(self): """Set up a randomized status for testing. This is being done in a separate function instead of in setUp() to facilitate testing the case where there is no status in the DB. """ self.test_status = { "provider_id": self.provider_uuid, "status": random.choice(list(ProviderStatusCode)), "last_message": self.FAKE.word(), "retries": random.randint(0, 10), "timestamp": self.date_accessor.today_with_timezone("UTC"), } with ProviderStatusAccessor(self.aws_provider_uuid) as accessor: status = accessor.add(**self.test_status) status.save() self.time_stamp = status.timestamp
def test_get_provider_uuid(self): """Test get_provider_uuid().""" self._setup_random_status() with ProviderStatusAccessor(self.aws_test_provider_uuid) as accessor: output = accessor.get_provider_uuid() self.assertEqual(output, self.aws_test_provider_uuid)
def test_init_wo_provider(self): """Test __init__() when a provider is not in the DB.""" with self.assertRaises(MasuProviderError): ProviderStatusAccessor(str(uuid.uuid4()))
def test_init(self): """Test __init__() when a status is in the DB.""" self._setup_random_status() with ProviderStatusAccessor(self.aws_test_provider_uuid) as accessor: self.assertIsNotNone(accessor._table) self.assertIsNotNone(accessor._obj)
def test_get_last_message(self): """Test get_last_message().""" self._setup_random_status() with ProviderStatusAccessor(self.aws_provider_uuid) as accessor: output = accessor.get_last_message() self.assertEqual(output, self.test_status.get('last_message'))