def test_push_to_queue(self):
with self.instance.test_request_context() as request:
key = 'movie:tt12345678:imdb'
result = Movie.push(key=key)
self.assertTrue(result)
self.assertEqual(len(self.redis.keys()), 1)
self.assertTrue(self.redis.exists(Movie.not_added_key))
self.assertEqual(self.redis.llen(Movie.not_added_key), 1)
self.assertEqual(self.redis.lpop(Movie.not_added_key), key)
key = 'movie:tt12345678'
result = Movie.push(key=key)
self.assertFalse(result)
self.assertFalse(self.redis.exists(Movie.not_added_key))
key = 'person:tt12345678:imdb'
result = Movie.push(key=key)
self.assertFalse(result)
self.assertFalse(self.redis.exists(Movie.not_added_key))
key = 'person:tt12345678:qwerty'
result = Movie.push(key=key)
self.assertFalse(result)
self.assertFalse(self.redis.exists(Movie.not_added_key))
def test_push_multiple_keys_to_queue(self):
with self.instance.test_request_context() as request:
keys = ['movie:tt1234567890:imdb', 'movie:tt1234567891:imdb',
'movie:tt1234567892:imdb', 'movie:tt1234567893:imdb']
result = Movie.push(keys=keys)
self.assertTrue(result)
self.assertTrue(self.redis.exists(Movie.not_added_key))
self.assertEqual(self.redis.llen(Movie.not_added_key), len(keys))
