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))
