test: addPlayerToTeamNewPlayer

src/test/java/cz/muni/fi/pa165/icehockeymanager/services/TeamServiceImplTest.java

@@ -2,6 +2,7 @@ package cz.muni.fi.pa165.icehockeymanager.services;
 
 import cz.muni.fi.pa165.icehockeymanager.config.ApplicationConfig;
 import cz.muni.fi.pa165.icehockeymanager.dao.TeamDao;
+import cz.muni.fi.pa165.icehockeymanager.model.Player;
 import cz.muni.fi.pa165.icehockeymanager.model.Team;
 import org.junit.jupiter.api.Test;
 import org.junit.jupiter.api.extension.ExtendWith;
@@ -10,9 +11,7 @@ import org.mockito.Mock;
 import org.springframework.test.context.ContextConfiguration;
 import org.springframework.test.context.junit.jupiter.SpringExtension;
 
-import java.util.ArrayList;
-import java.util.List;
-import java.util.Optional;
+import java.util.*;
 
 import static org.assertj.core.api.Assertions.assertThat;
 import static org.mockito.Mockito.verify;
@@ -60,6 +59,21 @@ class TeamServiceImplTest {
         verify(teamDao).read(1L);
     }
 
+    @Test
+    public void addPlayerToTeamNewPlayer() {
+        Team team = buildTeam("Florida Panthers");
+        var players = buildPlayers();
+        team.setPlayers(players);
+        Player player = buildPlayer(3L, "Miroslav Satan");
+        players = new HashSet<>(players);
+        players.add(player);
+        teamService.addPlayerToTeam(team, player);
+        assertThat(team.getPlayers()).hasSameElementsAs(players);
+        verify(teamDao).update(team);
+    }
+
+
+
 
     private Team buildTeam(String name) {
         var team = new Team();
@@ -75,4 +89,18 @@ class TeamServiceImplTest {
         teams.add(buildTeam("Colorado Avalanche"));
         return teams;
     }
+
+    private Player buildPlayer(Long id, String name) {
+        var player = new Player();
+        player.setId(id);
+        player.setName(name);
+        return player;
+    }
+
+    private Set<Player> buildPlayers() {
+        var players = new HashSet<Player>();
+        players.add(buildPlayer(1L, "Marian Hossa"));
+        players.add(buildPlayer(2L, "Dominik Hasek"));
+        return players;
+    }
 }
\ No newline at end of file