def test_winner(self): config = Config() config.add_place('Sisaket') config.add_place('Mall') session = Session(config) session.join(User('test')) session.finish() self.assertIn(session.winner, ('Mall', 'Sisaket'))
def test_finished(self): session = Session(Config()) self.assertEqual(session.finished, False) session.join(User('test')) session.finish() self.assertEqual(session.finished, True) with self.assertRaises(SessionFinished): session.finish() with self.assertRaises(SessionFinished): session.join(object()) with self.assertRaises(SessionFinished): session.leave(object())
def test_empty(self): session = Session(object()) with self.assertRaises(SessionEmpty): session.finish()