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(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)
