diff --git a/OGP1718-Worms/src/worms/model/Team.java b/OGP1718-Worms/src/worms/model/Team.java index 1076f39..5fef8bd 100644 --- a/OGP1718-Worms/src/worms/model/Team.java +++ b/OGP1718-Worms/src/worms/model/Team.java @@ -27,14 +27,15 @@ public class Team { /** * - * TODO Facade: addWormsToTeam(Team team, Worm... worms) * * @param worm */ - public void addWormsToTeam(Worm worm) { - if (canHaveAsWorm(worm)) { - team.add(worm); - sortList(team); + public void addWormsToTeam(Worm... worm) { + for (Worm w: worm) { + if (canHaveAsWorm(w)) { + team.add(w); + sortList(team); + } } } @@ -61,12 +62,13 @@ public class Team { /** * - * TODO Facade: removeWormsFromTeam(Team team, Worm... worms) * * @param worm */ - public void removeWormsFromTeam(Worm worm) { - team.remove(worm); + public void removeWormsFromTeam(Worm... worm) { + for (Worm w: worm) { + team.remove(w); + } } /**