def test_save_over_provisioned(self): s = get_storage_table() storage = Storage(s, SinkMetrics()) storage.table = Mock() def raise_error(*args, **kwargs): raise ProvisionedThroughputExceededException(None, None) storage.table.query_2.side_effect = raise_error with self.assertRaises(ProvisionedThroughputExceededException): storage.fetch_notifications("asdf")
def test_save_over_provisioned(self): s = get_storage_table() storage = Storage(s, SinkMetrics()) storage.table = Mock() def raise_error(*args, **kwargs): raise ProvisionedThroughputExceededException(None, None) storage.table.query_2.side_effect = raise_error with self.assertRaises(ProvisionedThroughputExceededException): storage.fetch_notifications(dummy_uaid)
def test_preflight_check(self): router = Router(get_router_table(), SinkMetrics()) storage = Storage(get_storage_table(), SinkMetrics()) pf_uaid = "deadbeef00000000deadbeef01010101" preflight_check(storage, router, pf_uaid) # now check that the database reports no entries. notifs = storage.fetch_notifications(pf_uaid) eq_(len(notifs), 0) self.assertRaises(ItemNotFound, router.get_uaid, pf_uaid)
def test_preflight_check_wait(self): router = Router(get_router_table(), SinkMetrics()) storage = Storage(get_storage_table(), SinkMetrics()) storage.table.describe = mock_describe = Mock() values = [ dict(Table=dict(TableStatus="PENDING")), dict(Table=dict(TableStatus="ACTIVE")), ] def return_vals(*args, **kwargs): return values.pop(0) mock_describe.side_effect = return_vals pf_uaid = "deadbeef00000000deadbeef01010101" preflight_check(storage, router, pf_uaid) # now check that the database reports no entries. notifs = storage.fetch_notifications(pf_uaid) eq_(len(notifs), 0) assert_raises(ItemNotFound, router.get_uaid, pf_uaid)