class UserTests(TestCase):
def setUp(self):
self._flush_redis()
self.user = User(username='******')
self.user.save()
def tearDown(self):
self._flush_redis()
def _flush_redis(self):
keys = settings.REDIS_CONN.keys('test_*')
if keys:
settings.REDIS_CONN.delete(*keys)
def _add_data_to_redis(self):
redis_key = settings.REDIS_KEY_USER_TAGS % self.user.id
settings.REDIS_CONN.zincrby(redis_key, 1, 'books')
settings.REDIS_CONN.zincrby(redis_key, 1, 'books')
settings.REDIS_CONN.zincrby(redis_key, 1, 'books')
settings.REDIS_CONN.zincrby(redis_key, 1, 'tax')
settings.REDIS_CONN.zincrby(redis_key, 1, 'tax')
settings.REDIS_CONN.zincrby(redis_key, 1, 'home')
def test_01_empty_user_tags_list(self):
self.assertEqual(list(self.user.get_tags_order_by_frequency()), [])
def test_02_get_user_tags_list(self):
self._add_data_to_redis()
self.assertEqual(list(self.user.get_tags_order_by_frequency()),
['books', 'tax', 'home'])
def test_03_ordered_tags(self):
self.user.tags = ['tax', 'new', 'books'] # tag 'home' is old.
self.user.save()
self._add_data_to_redis()
self.assertEqual(list(self.user.get_ordered_tags()),
['books', 'tax', 'new'])
