1 files changed, 14 insertions, 0 deletions
diff --git a/src/test/java/design/model/GolferTest.java b/src/test/java/design/model/GolferTest.java
index a73997a..98dad20 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;
@@ -136,4 +137,17 @@ public class GolferTest {
         String expectedString = "John Doe (@jdoesgolf2)";
         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());
+    }
+
+
 }
 \ No newline at end of file