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 b4ff58287a3c8cfa24f38dc587c05c5a1571982e..0036e218f766eb83c0024a4a9241a60337df9a6f 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 @@ -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