finished unit testing for teams and matches

1 files changed, 11 insertions, 0 deletions

diff --git a/src/test/java/design/model/GolferTest.java b/src/test/java/design/model/GolferTest.java
index 65229f3..5f113ab 100644
--- a/src/test/java/design/model/GolferTest.java
+++ b/src/test/java/design/model/GolferTest.java
@@ -10,6 +10,7 @@ import org.junit.jupiter.api.Tag;
 import org.junit.jupiter.api.Test;
 
 import design.model.Club.ClubType;
+import design.model.undo.Memento;
 
 import java.lang.reflect.Constructor;
 import java.time.LocalDateTime;
@@ -137,6 +138,16 @@ public class GolferTest {
         assertEquals(expectedString, testGolfer.toString());
     }
 
+    @Test
+    void testMemento()
+    {
+        Golfer testGolfer = new Golfer("John Doe", "jdoesgolf2", "weback4321");
+        Memento memento = testGolfer.createMemento();
+        testGolfer.setFullName("Joe Doe");
+        assertEquals("Joe Doe", testGolfer.getFullName());
+        testGolfer.restore(memento);
+        assertEquals("John Doe", testGolfer.getFullName());
+    }
 
 
 }