1 files changed, 42 insertions, 0 deletions
diff --git a/src/test/java/design/model/TeamTest.java b/src/test/java/design/model/TeamTest.java
new file mode 100644
index 0000000..c81aa1e
--- /dev/null
+++ b/src/test/java/design/model/TeamTest.java
@@ -0,0 +1,42 @@
+package design.model;
+
+import static org.junit.jupiter.api.Assertions.assertEquals;
+
+import org.junit.jupiter.api.Tag;
+import org.junit.jupiter.api.Test;
+
+/** Unit Tests for the Team Class.
+ * @author Willem Dalton
+ **/
+@Tag("Model-tier")
+public class TeamTest {
+    @Test
+    void testConstructor()
+    {
+        Golfer testGolfer = new Golfer("John Doe", "j_doe_golfs", "ilovegolf123");
+        Team testTeam = new Team("A Team", testGolfer);
+        assertEquals("A Team", testTeam.getName());
+        assertEquals(testGolfer, testTeam.getOwner());
+    }
+
+    @Test
+    void testSetName()
+    {
+        Golfer testGolfer = new Golfer("John Doe", "j_doe_golfs", "ilovegolf123");
+        Team testTeam = new Team("A Team", testGolfer);
+        testTeam.setName("B Team");
+        assertEquals("B Team", testTeam.getName());
+    }
+
+    @Test
+    void testRemoveMember()
+    {
+        Golfer testGolfer = new Golfer("John Doe", "j_doe_golfs", "ilovegolf123");
+        Golfer newGolfer = new Golfer("Jane Doe", "j_doe_golfs2", "ilovegolf321");
+        Team testTeam = new Team("A Team", testGolfer);
+        testTeam.addMember(newGolfer);
+        assertEquals(1, testTeam.getMembers().length);
+        testTeam.removeMember(newGolfer);
+        assertEquals(0, testTeam.getMembers().length);
+    }
+}