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)
class TestNotifications(TestCase): not_cont = NotificationController() fan1 = Fan("user1", "123", "nameOne", datetime(1993, 1, 1), "1.1.1.1", "1") fan2 = Fan("user2", "123", "nameTwo", datetime(1993, 1, 1), "1.1.1.1", "2") fan3 = Fan("user3", "123", "nameThree", datetime(1993, 1, 1), "1.1.1.1", "3") team1 = Team("team1") team2 = Team("team2") game1 = Game(team1, team2, datetime.today(), "field1") game2 = Game(team1, team2, datetime.today(), "field2") referee1 = Referee(RefereeQualificationEnum.MAIN, "user4", "123", "userFour", datetime.today(), "1", "4") referee2 = Referee(RefereeQualificationEnum.MAIN, "user5", "123", "userFive", datetime.today(), "1", "5") not_cont.add_fan_follower_to_game(fan1, game1) not_cont.add_fan_follower_to_game(fan2, game1) not_cont.add_fan_follower_to_game(fan2, game2) game1.add_referee(referee1) game2.add_referee(referee2) game_event1 = GameEvent(game1, referee1, EventTypeEnum.YELLOW_CARD, "desc", datetime.today(), 5) game_event2 = GameEvent(game2, referee2, EventTypeEnum.GOAL, "desc", datetime.today(), 5) def test_add_follower(self): self.assertEqual(len(self.fan1.notifications), 2) self.assertEqual(len(self.fan2.notifications), 4) self.assertEqual(len(self.fan3.notifications), 0) def test_remove_follower(self): self.not_cont.remove_fan_follower_from_game(self.fan2, self.game1) self.game1.remove_referee(self.referee1) self.assertEqual(len(self.fan1.notifications), 3) self.assertEqual(len(self.fan2.notifications), 4) self.assertEqual(len(self.fan3.notifications), 0)
def test_add_event(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") game1.add_referee(ref) event1 = GameEvent(game1, ref, EventTypeEnum.GOAL, "Start!", d1, 22) self.assertIn(event1, ref.events) ref.remove_event(event1) self.assertNotIn(event1, ref.events)
def test_add_remove_referee(self): m_r = Referee(RefereeQualificationEnum.MAIN, 'Dor123', '12345678', 'Dor', datetime(1990, 8, 8), '1.1.1.1', '') r = Referee(RefereeQualificationEnum.REGULAR, 'Dor12', '12345678', 'Dor', datetime(1990, 8, 8), '1.1.1.1', '') self.game.main_referee = m_r self.game.add_referee(r) self.assertIn(r, self.game._Game__referees) self.game.remove_referee(r) self.assertNotIn(r, self.game._Game__referees)
def create_referee(user_dict): qualifications = { 'MAIN': RefereeQualificationEnum.MAIN, 'REGULAR': RefereeQualificationEnum.REGULAR } referee = Referee(qualifications[user_dict['qualification']], user_dict['user_name'], user_dict['password'], user_dict['name'], get_birth_date_object(user_dict['birth_date']), user_dict['user_id']) referee.referee_in_games = user_dict['referee_in_games'] referee.events = user_dict['events'] referee.notifications = user_dict['notifications'] return referee
def test_add_remove_referee(self): ref = Referee(RefereeQualificationEnum.MAIN, 'Dor123', '12345678', 'Dor', datetime(1990, 8, 8), '1.1.1.1', '') self.league.add_referee(ref) self.assertIn(ref, self.league._League__referees) self.league.remove_referee(ref) self.assertNotIn(ref, self.league._League__referees)
def add_referee(self, qualification, user_name, password, name, birth_date, using_user_id=""): try: referee_user = Referee(qualification, user_name, password, name, birth_date, self.__ID) self.__user_data_base.add_signed_user(referee_user, 'referee') self.update_counter() Logger.info_log("{0}:".format(using_user_id) + "add referee : {0}".format(user_name)) except Exception as err: Logger.error_log("{0}:".format(using_user_id) + err.__str__()) raise err
def test_add_remove_event(self): r = Referee(RefereeQualificationEnum.MAIN, 'Dor123', '12345678', 'Dor', datetime(1990, 8, 8), '1.1.1.1', '') self.game.add_referee(r) game_event = GameEvent(self.game, r, "type", "des", datetime(2020, 5, 5), 89) g = Game(self.home_team, self.away_team, self.d, self.field) g.main_referee = Referee(RefereeQualificationEnum.MAIN, 'Dor12', '12345678', 'Dor', datetime(1990, 8, 8), '1.1.1.1', '') not_game_event = GameEvent(g, g.main_referee, EventTypeEnum.GOAL, "des", datetime(2020, 5, 5), 89) self.assertRaises(ValueError, self.game.add_event, event=not_game_event) self.assertNotIn(not_game_event, self.game._Game__events) self.assertIn(game_event, self.game._Game__events) self.assertRaises(ValueError, self.game.add_event, event=game_event) self.game.remove_event(r) self.assertNotIn(r, self.game._Game__events) self.game.remove_event(game_event) self.assertNotIn(game_event, self.game._Game__events)
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())
def test_set_main_referee(self): referee = Referee(RefereeQualificationEnum.MAIN, 'Dor123', '12345678', 'Dor', datetime(1990, 8, 8), '1.1.1.1', '') self.assertRaises(TypeError, self.game.main_referee, main_referee=self.home_team) self.game.main_referee = referee self.assertEqual(referee, self.game._Game__main_referee)