From f8c481c0a022ca3392b03557d491d49e1671fa9e Mon Sep 17 00:00:00 2001
From: Erik Moravec <xmoravec@fi.muni.cz>
Date: Wed, 28 Apr 2021 17:39:06 +0200
Subject: [PATCH] test: fixed test for add away games and added remove home
 games tests

---
 .../services/TeamServiceImplTest.java         | 31 +++++++++++++++++--
 1 file changed, 29 insertions(+), 2 deletions(-)

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 8031823..74b7782 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);
-- 
GitLab