def test_find_redis_notes(self, get_redis_connection):
"""Verifies get_common_words is properly functioning"""
get_redis_connection.return_value = self.redis
redis_notes.add_redis_note(1373500800,
"today: this is a simple test #yolo")
redis_notes.add_redis_note(1450794188, "today: this is note 2")
first_note = [(1373500800, 'today: this is a simple test #yolo')]
second_note = [(1450794188, 'today: this is note 2')]
both_notes = [(1373500800, 'today: this is a simple test #yolo'),
(1450794188, 'today: this is note 2')]
request = get_search_request(['simple'])
self.assertListEqual(first_note, redis_notes.find_redis_notes(request))
request = get_search_request(['today'])
self.assertListEqual(both_notes, redis_notes.find_redis_notes(request))
request = get_search_request(['today', '!yolo'])
self.assertListEqual(second_note,
redis_notes.find_redis_notes(request))
request = get_search_request(['!note'])
self.assertListEqual(first_note, redis_notes.find_redis_notes(request))
request = get_search_request()
self.assertListEqual(both_notes, redis_notes.find_redis_notes(request))
request = get_search_request(['missing'])
self.assertListEqual([(None, None)],
redis_notes.find_redis_notes(request))
def test_delete_redis_note_miss(self, get_redis_connection):
"""Verifies add_redis_note is properly functioning"""
get_redis_connection.return_value = self.redis
redis_notes.add_redis_note(1373500800, "today: this is note 1")
redis_notes.add_redis_note(1450794188, "today: this is note 2")
redis_notes.delete_redis_note(1373500801)
self.assertEqual(b"today: this is note 1",
self.redis.get('note_1373500800'))
self.assertEqual(b"today: this is note 2",
self.redis.get('note_1450794188'))
def test_add_redis_note(self, get_redis_connection):
"""Verifies add_redis_note is properly functioning"""
get_redis_connection.return_value = self.redis
redis_notes.add_redis_note(1373500800,
"today: this is a simple test #yolo")
keys = self.redis.keys()
keys.sort()
self.assertListEqual([
b'day_11', b'hour_0', b'month_7', b'note_1373500800', b'w_#yolo',
b'w_a', b'w_is', b'w_simple', b'w_test', b'w_this', b'w_today',
b'w_yolo', b'weekday_3', b'year_2013'
], keys)
def test_get_common_words(self, get_redis_connection):
"""Verifies get_common_words is properly functioning"""
get_redis_connection.return_value = self.redis
redis_notes.add_redis_note(1373500800,
"today: this is a simple test #yolo")
redis_notes.add_redis_note(1450794188, "today: this is note 2")
words = list(redis_notes.get_common_words())
words.sort()
self.assertListEqual([
'#yolo', '2', 'a', 'is', 'note', 'simple', 'test', 'this', 'today',
'yolo'
], words)
def test_delete_redis_note_hit(self, get_redis_connection):
"""Verifies add_redis_note is properly functioning"""
get_redis_connection.return_value = self.redis
redis_notes.add_redis_note(1373500800,
"today: this is a simple test #yolo")
redis_notes.add_redis_note(1450794188, "today: this is note 2")
keys = self.redis.keys()
keys.sort()
self.assertListEqual([
b'day_11', b'day_22', b'hour_0', b'hour_14', b'month_12',
b'month_7', b'note_1373500800', b'note_1450794188', b'w_#yolo',
b'w_2', b'w_a', b'w_is', b'w_note', b'w_simple', b'w_test',
b'w_this', b'w_today', b'w_yolo', b'weekday_1', b'weekday_3',
b'year_2013', b'year_2015'
], keys)
redis_notes.delete_redis_note(1373500800)
keys.sort()
self.assertEqual(b"set()", self.redis.get('day_11'))
self.assertEqual(b"{b'1450794188'}", self.redis.get('day_22'))
self.assertEqual(b"set()", self.redis.get('hour_0'))
self.assertEqual(b"{b'1450794188'}", self.redis.get('hour_14'))
self.assertEqual(b"{b'1450794188'}", self.redis.get('month_12'))
self.assertEqual(b"set()", self.redis.get('month_7'))
self.assertEqual(None, self.redis.get('note_1373500800'))
self.assertEqual(b"today: this is note 2",
self.redis.get('note_1450794188'))
self.assertEqual(b"set()", self.redis.get('w_#yolo'))
self.assertEqual(b"{b'1450794188'}", self.redis.get('w_2'))
self.assertEqual(b"set()", self.redis.get('w_a'))
self.assertEqual(b"{b'1450794188'}", self.redis.get('w_is'))
self.assertEqual(b"{b'1450794188'}", self.redis.get('w_note'))
self.assertEqual(b"set()", self.redis.get('w_simple'))
self.assertEqual(b"set()", self.redis.get('w_test'))
self.assertEqual(b"{b'1450794188'}", self.redis.get('w_this'))
self.assertEqual(b"{b'1450794188'}", self.redis.get('w_today'))
self.assertEqual(b"set()", self.redis.get('w_yolo'))
self.assertEqual(b"{b'1450794188'}", self.redis.get('weekday_1'))
self.assertEqual(b"set()", self.redis.get('weekday_3'))
self.assertEqual(b"set()", self.redis.get('year_2013'))
self.assertEqual(b"{b'1450794188'}", self.redis.get('year_2015'))