diff options
Diffstat (limited to 'src/test/java')
| -rw-r--r-- | src/test/java/design/model/GolferTest.java | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/src/test/java/design/model/GolferTest.java b/src/test/java/design/model/GolferTest.java index 4dce705..3a79070 100644 --- a/src/test/java/design/model/GolferTest.java +++ b/src/test/java/design/model/GolferTest.java @@ -103,7 +103,8 @@ public class GolferTest { { Golfer testGolfer = new Golfer("John Doe", "jdoesgolf2", "weback4321"); - testGolfer.addClub("John Doe", "The Slammer", ClubType.DRIVER); + Club c = new Club("John Doe", "The Slammer", ClubType.DRIVER); + testGolfer.addClub(c); Club addedClub = testGolfer.getClubs()[0]; assertTrue(testGolfer.hasClub(addedClub)); assertTrue(testGolfer.hasClubs()); @@ -114,7 +115,8 @@ public class GolferTest { { Golfer testGolfer = new Golfer("John Doe", "jdoesgolf2", "weback4321"); - testGolfer.addClub("John Doe", "The Slammer", ClubType.DRIVER); + Club c = new Club("John Doe", "The Slammer", ClubType.DRIVER); + testGolfer.addClub(c); Club addedClub = testGolfer.getClubs()[0]; testGolfer.removeClub(addedClub); assertFalse(testGolfer.hasClub(addedClub)); |