1 files changed, 43 insertions, 0 deletions
diff --git a/src/test/java/design/model/TestPlay.java b/src/test/java/design/model/TestPlay.java
new file mode 100644
index 0000000..c3282bf
--- /dev/null
+++ b/src/test/java/design/model/TestPlay.java
@@ -0,0 +1,43 @@
+package design.model;
+
+import static org.junit.jupiter.api.Assertions.assertEquals;
+
+import org.junit.jupiter.api.Tag;
+import org.junit.jupiter.api.Test;
+
+import design.model.Club.ClubType;
+
+import java.util.ArrayList;
+
+/** Unit Tests for the Play class.
+ * @author Willem Dalton
+ **/
+@Tag("Model-tier")
+public class TestPlay {
+    @Test
+    void testConstructor()
+    {
+        Play testPlay = new Play(0);
+        assertEquals(0, testPlay.getHoleNumber());
+    }
+
+    @Test
+    void testConstructorNull()
+    {
+        Play testPlay2 = new Play(0, null);
+        assertEquals(0, testPlay2.getHoleNumber());
+        assertEquals(0, testPlay2.getSwings().length);
+    }
+
+    @Test
+    void testConstructorNotNull()
+    {
+        Club newClub = new Club("John Doe Inc", "The Slammer", ClubType.DRIVER);
+        Swing newSwing = new Swing(100, newClub);
+        ArrayList<Swing> swings = new ArrayList<Swing>();
+        swings.add(newSwing);
+        Play testPlay3 = new Play(0, swings);
+        assertEquals(0, testPlay3.getHoleNumber());
+        assertEquals(swings.get(0), testPlay3.getSwings()[0]);
+    }
+}