def test_user_logged(self):
user_id = random.randrange(0, 100000)
ls = LoginMetrics(self.date)
# Initial State
self.assertEquals(LoginMetrics.FLAG_DOWN, ls.get_flag(user_id))
# After login
ls.set_flag(user_id)
self.assertEquals(LoginMetrics.FLAG_UP, ls.get_flag(user_id))
def test_login_user(self):
user_id = random.randrange(0, 100000)
ls = LoginMetrics(self.date)
# Initial State
bit = self.r.getbit(ls.key, user_id)
self.assertEquals(LoginMetrics.FLAG_DOWN, bit)
# After login
ls.set_flag(user_id)
bit = self.r.getbit(ls.key, user_id)
self.assertEquals(LoginMetrics.FLAG_UP, bit)
