def test_fetch_over_provisioned(self): s = get_storage_table() storage = Storage(s, SinkMetrics()) storage.table.connection = Mock() def raise_error(*args, **kwargs): raise ProvisionedThroughputExceededException(None, None) storage.table.connection.put_item.side_effect = raise_error with self.assertRaises(ProvisionedThroughputExceededException): storage.save_notification("asdf", "asdf", 12)
def test_fetch_over_provisioned(self): s = get_storage_table() storage = Storage(s, SinkMetrics()) storage.table.connection = Mock() def raise_error(*args, **kwargs): raise ProvisionedThroughputExceededException(None, None) storage.table.connection.put_item.side_effect = raise_error with self.assertRaises(ProvisionedThroughputExceededException): storage.save_notification(dummy_uaid, dummy_chid, 12)
def test_fetch_boto_err(self): s = get_storage_table() storage = Storage(s, SinkMetrics()) storage.table.connection = Mock() def raise_error(*args, **kwargs): raise BotoServerError(None, None) storage.table.connection.put_item.side_effect = raise_error with assert_raises(BotoServerError): storage.save_notification(dummy_uaid, dummy_chid, 12)
def test_dont_save_older(self): s = get_storage_table() storage = Storage(s, SinkMetrics()) # Unfortunately moto can't run condition expressions, so # we gotta fake it storage.table.connection = Mock() def raise_error(*args, **kwargs): raise ConditionalCheckFailedException(None, None) storage.table.connection.put_item.side_effect = raise_error result = storage.save_notification("fdas", "asdf", 8) eq_(result, False)
def test_dont_save_older(self): s = get_storage_table() storage = Storage(s, SinkMetrics()) # Unfortunately moto can't run condition expressions, so # we gotta fake it storage.table.connection = Mock() def raise_error(*args, **kwargs): raise ConditionalCheckFailedException(None, None) storage.table.connection.put_item.side_effect = raise_error result = storage.save_notification(dummy_uaid, dummy_chid, 8) eq_(result, False)