diff --git a/src/main/java/edu/ntnu/idatt1002/k1g4/Match.java b/src/main/java/edu/ntnu/idatt1002/k1g4/Match.java index 8178830e8214badae59bfed23d38c05057847516..f27e890a177588e097f729146fbd2112553f4867 100644 --- a/src/main/java/edu/ntnu/idatt1002/k1g4/Match.java +++ b/src/main/java/edu/ntnu/idatt1002/k1g4/Match.java @@ -161,7 +161,6 @@ public class Match { public void setFinished() { setFinished(LocalDateTime.now()); if (isKnockout) { - System.out.println("WTF"); setWinner(); } } diff --git a/src/main/java/edu/ntnu/idatt1002/k1g4/client/Model.java b/src/main/java/edu/ntnu/idatt1002/k1g4/client/Model.java index 5f34f26b348786e38c30e0c9d7ab329a9b3d5d23..00e74491b9e1b624332664e9fe56cdd2f8b66abc 100644 --- a/src/main/java/edu/ntnu/idatt1002/k1g4/client/Model.java +++ b/src/main/java/edu/ntnu/idatt1002/k1g4/client/Model.java @@ -370,7 +370,7 @@ public class Model { team.setName(teamName); team.setCompeting(true); try { - teamDAO.addTeam(team); + teamDAO.addTeam(team, currentDivision.getDivisionId()); teamDAO.linkTeamToDivision(teamDAO.getTeamByName(teamName).getTeamId(), currentDivision.getDivisionId()); } catch (IllegalArgumentException e) { result = false; diff --git a/src/main/java/edu/ntnu/idatt1002/k1g4/dao/TeamDAO.java b/src/main/java/edu/ntnu/idatt1002/k1g4/dao/TeamDAO.java index 8ed8f372a50c75c4539548ef68f7fe0ad6531d85..c85cd45622676f4b5d9a6eb522002caeab961aa3 100644 --- a/src/main/java/edu/ntnu/idatt1002/k1g4/dao/TeamDAO.java +++ b/src/main/java/edu/ntnu/idatt1002/k1g4/dao/TeamDAO.java @@ -186,7 +186,7 @@ public class TeamDAO { * * @param team Team: the team to be added to the database. */ - public void addTeam(Team team) throws IllegalArgumentException { + public void addTeam(Team team, int divisionId) throws IllegalArgumentException { String sql = "INSERT INTO teams(teamName, competing) VALUES(?, ?)"; Team teamFromDb = getTeamByName(team.getName()); ResultSet resultSet = null; @@ -205,7 +205,25 @@ public class TeamDAO { close(connection, preparedStatement, resultSet); } } else { - throw new IllegalArgumentException(); + String sqlLink = "SELECT * FROM teams, divisionsteamslink WHERE teams.teamId = divisionsteamslink.teamId AND teams.teamName = ? AND divisionsteamslink.divisionId = ?"; + Team newTeamFromDB; + try { + connection = Database.instance().getConnection(); + preparedStatement = connection.prepareStatement(sqlLink); + preparedStatement.setString(1, team.getName()); + preparedStatement.setInt(2, divisionId); + resultSet = preparedStatement.executeQuery(); + newTeamFromDB = getTeamFromResultSet(resultSet); + if (newTeamFromDB == null) { + linkTeamToDivision(teamFromDb.getTeamId(), divisionId); + } else { + throw new IllegalArgumentException(); + } + } catch (SQLException e) { + e.printStackTrace(); + } finally { + close(connection, preparedStatement, resultSet); + } } } diff --git a/src/test/java/edu/ntnu/idatt1002/k1g4/MatchDAOTest.java b/src/test/java/edu/ntnu/idatt1002/k1g4/MatchDAOTest.java index 74f4172468195564ee17b970995457d21e23573e..bfc40d09fe5af66008f36f7684ddb6ffc2054775 100644 --- a/src/test/java/edu/ntnu/idatt1002/k1g4/MatchDAOTest.java +++ b/src/test/java/edu/ntnu/idatt1002/k1g4/MatchDAOTest.java @@ -33,7 +33,6 @@ public class MatchDAOTest { expectedFieldList.add("A1"); expectedFieldList.add("A2"); expectedFieldList.add("A3"); - System.out.println(actualMatches.size()); assertTrue(actualMatches.size() == expectedFieldList.size() && actualMatches.containsAll(expectedFieldList)); } diff --git a/src/test/java/edu/ntnu/idatt1002/k1g4/TeamDAOTest.java b/src/test/java/edu/ntnu/idatt1002/k1g4/TeamDAOTest.java index 3960fb8441b06d8f4ae057cedc3650e7671191db..24dffb29f9417c894395fcf67aecae26c7059935 100644 --- a/src/test/java/edu/ntnu/idatt1002/k1g4/TeamDAOTest.java +++ b/src/test/java/edu/ntnu/idatt1002/k1g4/TeamDAOTest.java @@ -64,7 +64,7 @@ public class TeamDAOTest { public void testAddTeam() { Team team = new Team("Callums Javaer", true); try { - teamDAO.addTeam(team); + teamDAO.addTeam(team, 1); } catch (IllegalArgumentException e) { System.out.println(e.getMessage()); }