def test_add_game(self):
d1 = datetime.datetime(2020, 4, 23)
d2 = datetime.datetime(2001, 4, 23)
ref = Referee(RefereeQualificationEnum.MAIN, "name_u1", "1234", "ro", d1, "0.0.0.5", 23)
game = Game(Team("Real Madrid"), Team("Barcelona"), d2, "Camp Nou")
ref.add_game(game)
self.assertIn(game, ref.referee_in_games)
def test_show_games_by_referee(self):
d1 = datetime.datetime(2020, 4, 23)
d2 = datetime.datetime(2001, 4, 23)
ref = Referee(RefereeQualificationEnum.MAIN, "name_u1", "1234", "ro", d1, "0.0.0.5", 23)
game1 = Game(Team("Real Madrid"), Team("Barcelona"), d2, "Camp Nou")
game2 = Game(Team("Real Madrid2"), Team("Barcelona2"), d1, "Camp Nou2")
game1.add_referee(ref)
ref.add_game(game1)
self.assertIn(game1, ref.show_games_by_referee())
self.assertNotIn(game2, ref.show_games_by_referee())