1 files changed, 36 insertions, 0 deletions
diff --git a/src/test/java/design/model/CourseTest.java b/src/test/java/design/model/CourseTest.java
new file mode 100644
index 0000000..b6dd4a5
--- /dev/null
+++ b/src/test/java/design/model/CourseTest.java
@@ -0,0 +1,36 @@
+package design.model;
+
+import static org.junit.jupiter.api.Assertions.assertEquals;
+
+import org.junit.jupiter.api.Tag;
+import org.junit.jupiter.api.Test;
+
+import java.util.ArrayList;
+
+
+/** Unit Tests for the Course class.
+ * @author Willem Dalton
+ **/
+@Tag("Model-tier")
+public class CourseTest{
+    
+    @Test
+    void testConstructor()
+    {
+        Course testCourse = new Course(0, "Rolling Waves", 62, "Rochester, NY", 9, 20, new ArrayList<Hole>());
+        assertEquals(0, testCourse.getId());
+        assertEquals("Rolling Waves", testCourse.getName());
+        assertEquals(62, testCourse.getDifficultyRating());
+        assertEquals("Rochester, NY", testCourse.getLocation());
+        assertEquals(9, testCourse.getHoleCount());
+        assertEquals(20, testCourse.getTotalPar());
+        assertEquals(0, testCourse.getHoles().size());
+    }
+
+    @Test
+    void testToString()
+    {
+        Course testCourse = new Course(0, "Rolling Waves", 62, "Rochester, NY", 9, 20, new ArrayList<Hole>());
+        assertEquals("Rolling Waves (Rochester, NY) | Holes: 9 | Total Par: 20 | Difficulty: 62.0", testCourse.toString());
+    }
+}