def test_upload_timeout_error(self, mock_redis):
mock_queue = Mock(spec=HashQueue)
store = RedisStore(mock_redis, TIER_0, mock_queue, True, None)
store.queue.enqueue.side_effect = TimeoutError
with self.assertRaises(TimeoutError) as ar:
store.upload(self.user_id, self.uuid, self.data)
self.assertIsInstance(ar.exception, TimeoutError)
def test_download_key_error(self, mock_redis):
store = RedisStore(mock_redis, TIER_0, self.queue_cls, True, None)
k = data = None
with self.assertRaises(ValueError) as ar:
k, data = store.download(None, self.uuid)
self.assertIsNone(k)
self.assertIsNone(data)
self.assertIsInstance(ar.exception, ValueError)
def test_delete_key_error(self, mock_redis):
store = RedisStore(mock_redis, TIER_0, self.queue_cls, True, None)
len = total_len = -1
with self.assertRaises(ValueError) as ar:
len, total_len = store.delete(None, self.uuid)
self.assertEqual(-1, len)
self.assertEqual(-1, total_len)
self.assertIsInstance(ar.exception, ValueError)
def test_delete_timeout_error(self, mock_redis):
store = RedisStore(mock_redis, TIER_0, self.queue_cls, True, None)
store.queue.release.side_effect = TimeoutError
len = total_len = -1
with self.assertRaises(TimeoutError) as ar:
len, total_len = store.delete(self.user_id, self.uuid)
self.assertEqual(-1, len)
self.assertEqual(-1, total_len)
self.assertIsInstance(ar.exception, TimeoutError)
def test_upload(self, mock_redis):
store = RedisStore(mock_redis,
TIER_0,
self.queue_cls,
True,
None)
store.queue.enqueue.return_value = None
r = store.upload(self.user_id, self.uuid, self.data)
self.assertIsNone(r)
def test_download_timeout_error(self, mock_redis):
store = RedisStore(mock_redis, TIER_0, self.queue_cls, True, None)
store.queue.dequeue.side_effect = TimeoutError
k = data = None
with self.assertRaises(TimeoutError) as ar:
k, data = store.download(self.user_id, self.uuid)
self.assertIsNone(k)
self.assertIsNone(data)
self.assertIsInstance(ar.exception, TimeoutError)
def test_delete(self, mock_redis):
store = RedisStore(mock_redis,
TIER_0,
self.queue_cls,
True,
None)
store.queue.release.return_value = 10, 20
user_len, total_len = store.delete(self.user_id, self.uuid)
self.assertEqual(10, user_len)
self.assertEqual(20, total_len)
def test_download(self, mock_redis):
store = RedisStore(mock_redis,
TIER_0,
self.queue_cls,
True,
None)
store.queue.dequeue.return_value = self.uuid, self.data
k, data = store.download(self.user_id, self.uuid)
self.assertEqual(data, self.data)
self.assertEqual(k, self.uuid)
def test_upload_key_error(self, mock_redis):
store = RedisStore(mock_redis,
TIER_0,
self.queue_cls,
True,
None)
store.queue.enqueue.return_value= ValueError
with self.assertRaises(ValueError) as ar:
store.upload(None, self.uuid, self.data)
self.assertIsInstance(ar.exception, ValueError)
def test_upload_timeout_error(self, mock_redis):
mock_queue = Mock(spec=HashQueue)
store = RedisStore(mock_redis,
TIER_0,
mock_queue,
True,
None)
store.queue.enqueue.side_effect = TimeoutError
with self.assertRaises(TimeoutError) as ar:
store.upload(self.user_id, self.uuid, self.data)
self.assertIsInstance(ar.exception, TimeoutError)
def test_download_key_error(self, mock_redis):
store = RedisStore(mock_redis,
TIER_0,
self.queue_cls,
True,
None)
k = data = None
with self.assertRaises(ValueError) as ar:
k, data = store.download(None, self.uuid)
self.assertIsNone(k)
self.assertIsNone(data)
self.assertIsInstance(ar.exception, ValueError)
def test_delete_key_error(self, mock_redis):
store = RedisStore(mock_redis,
TIER_0,
self.queue_cls,
True,
None)
len = total_len = -1
with self.assertRaises(ValueError) as ar:
len, total_len = store.delete(None, self.uuid)
self.assertEqual(-1, len)
self.assertEqual(-1, total_len)
self.assertIsInstance(ar.exception, ValueError)
def test_delete_timeout_error(self, mock_redis):
store = RedisStore(mock_redis,
TIER_0,
self.queue_cls,
True,
None)
store.queue.release.side_effect = TimeoutError
len = total_len = -1
with self.assertRaises(TimeoutError) as ar:
len, total_len = store.delete(self.user_id, self.uuid)
self.assertEqual(-1, len)
self.assertEqual(-1, total_len)
self.assertIsInstance(ar.exception, TimeoutError)
def test_download_timeout_error(self, mock_redis):
store = RedisStore(mock_redis,
TIER_0,
self.queue_cls,
True,
None)
store.queue.dequeue.side_effect = TimeoutError
k = data = None
with self.assertRaises(TimeoutError) as ar:
k, data = store.download(self.user_id, self.uuid)
self.assertIsNone(k)
self.assertIsNone(data)
self.assertIsInstance(ar.exception, TimeoutError)
def test_delete(self, mock_redis):
store = RedisStore(mock_redis, TIER_0, self.queue_cls, True, None)
store.queue.release.return_value = 10, 20
user_len, total_len = store.delete(self.user_id, self.uuid)
self.assertEqual(10, user_len)
self.assertEqual(20, total_len)
def test_download(self, mock_redis):
store = RedisStore(mock_redis, TIER_0, self.queue_cls, True, None)
store.queue.dequeue.return_value = self.uuid, self.data
k, data = store.download(self.user_id, self.uuid)
self.assertEqual(data, self.data)
self.assertEqual(k, self.uuid)
def test_upload(self, mock_redis):
store = RedisStore(mock_redis, TIER_0, self.queue_cls, True, None)
store.queue.enqueue.return_value = None
r = store.upload(self.user_id, self.uuid, self.data)
self.assertIsNone(r)
def test_upload_key_error(self, mock_redis):
store = RedisStore(mock_redis, TIER_0, self.queue_cls, True, None)
store.queue.enqueue.return_value = ValueError
with self.assertRaises(ValueError) as ar:
store.upload(None, self.uuid, self.data)
self.assertIsInstance(ar.exception, ValueError)