def test_reaches_vote_threshold(self):
test_room = Room(None, None)
for entry in REACHES_VOTE_THRESHOLDS:
if USER_COUNT in entry:
for i in range(entry[USER_COUNT]):
test_room.add_user(User(None, sid=i))
test_room.vote_threshold = entry[THRESHOLD]
self.assertEqual(test_room.reaches_vote_threshold(entry[INPUT]), entry[OUTPUT])
if len(test_room) != 0:
test_room.empty_room()
def test_empty_room(self):
test_room = Room(None, None)
test_room.add_user(USERS[0])
test_room.empty_room()
self.assertTrue(len(test_room) == 0)