diff --git a/src/test/java/cz/muni/fi/pa165/icehockeymanager/services/TeamServiceImplTest.java b/src/test/java/cz/muni/fi/pa165/icehockeymanager/services/TeamServiceImplTest.java index 8031823ddc080a84ec672b8f3713d0b8773473dc..74b7782d4cd7ee7b5cf7d07f5bc767eecda56723 100644 --- a/src/test/java/cz/muni/fi/pa165/icehockeymanager/services/TeamServiceImplTest.java +++ b/src/test/java/cz/muni/fi/pa165/icehockeymanager/services/TeamServiceImplTest.java @@ -166,7 +166,7 @@ class TeamServiceImplTest { Team team = buildTeam("Florida Panthers"); Game game = buildGame(); teamService.addAwayGame(team, game); - assertThat(team.getHomeGames()).hasSameElementsAs(List.of(game)); + assertThat(team.getAwayGames()).hasSameElementsAs(List.of(game)); verify(teamDao).update(team); } @@ -176,7 +176,7 @@ class TeamServiceImplTest { assertThatExceptionOfType(NullPointerException.class).isThrownBy( () -> teamService.addAwayGame(team, null) ); - assertThat(team.getHomeGames()).hasSameElementsAs(List.of()); + assertThat(team.getAwayGames()).hasSameElementsAs(List.of()); } @Test @@ -187,6 +187,33 @@ class TeamServiceImplTest { ); } + @Test + public void removeHomeGame() { + Team team = buildTeam("Florida Panthers"); + Game game = buildGame(); + team.setHomeGames(Set.of(game)); + teamService.removeHomeGame(team, game); + assertThat(team.getHomeGames()).hasSameElementsAs(List.of()); + verify(teamDao).update(team); + } + + @Test + public void removeHomeGameGameIsNull() { + Team team = buildTeam("Florida Panthers"); + assertThatExceptionOfType(NullPointerException.class).isThrownBy( + () -> teamService.addHomeGame(team, null) + ); + assertThat(team.getHomeGames()).hasSameElementsAs(List.of()); + } + + @Test + public void removeHomeGameTeamIsNull() { + Game game = buildGame(); + assertThatExceptionOfType(NullPointerException.class).isThrownBy( + () -> teamService.addHomeGame(null, game) + ); + } + private Game buildGame() { var game = new Game(); game.setId(1L);