diff --git a/patterns/design-patterns-2/src/test/java/com/baeldung/memento/TextEditorTest.java b/patterns/design-patterns-2/src/test/java/com/baeldung/memento/TextEditorTest.java
new file mode 100644
index 0000000000..b985753d91
--- /dev/null
+++ b/patterns/design-patterns-2/src/test/java/com/baeldung/memento/TextEditorTest.java
@@ -0,0 +1,20 @@
+package com.baeldung.memento;
+
+import org.junit.jupiter.api.Test;
+
+import static org.assertj.core.api.Assertions.assertThat;
+
+class TextEditorTest {
+
+    @Test
+    void givenTextEditor_whenAddTextSaveAddMoreAndUndo_thenSavecStateRestored() {
+        TextEditor textEditor = new TextEditor(new TextWindow());
+        textEditor.write("The Memento Design Pattern\n");
+        textEditor.write("How to implement it in Java?\n");
+        textEditor.hitSave();
+        textEditor.write("Buy milk and eggs before coming home\n");
+        textEditor.hitUndo();
+
+        assertThat(textEditor.print()).isEqualTo("The Memento Design Pattern\nHow to implement it in Java?\n");
+    }
+}
\ No newline at end of file