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_users_logged(self): users_ids = [random.randrange(0, 100000) for i in range(100)] ls = LoginMetrics(self.date) # Initial State for i in users_ids: self.assertEquals(LoginMetrics.FLAG_DOWN, ls.get_flag(i)) # After login ls.set_flags(users_ids) correct_list = [LoginMetrics.FLAG_UP for i in range(len(users_ids))] self.assertEquals(correct_list, ls.get_flags(users_ids))
def test_users_state(self): users_ids = [random.randrange(0, 100000) for i in range(100)] ls = LoginMetrics(self.date) # Initial State for i in users_ids: self.assertEquals(LoginMetrics.FLAG_DOWN, ls.get_flag(i)) # After login ls.set_flags(users_ids) correct_list = [LoginMetrics.FLAG_UP for i in range(len(users_ids))] self.assertEquals(correct_list, ls.get_flags(users_ids)) # take some random users out black_ships = [random.randrange(0, len(users_ids)) for i in range( len(users_ids))] for i in black_ships: ls.unset_flag(users_ids[i]) correct_list[i] = LoginMetrics.FLAG_DOWN # Test again self.assertEquals(correct_list, ls.get_flags(users_ids))