Update SerializationUnitTest.java

amitiw4u · amitiw4u · commit c902fc97caa9 · 2021-10-24T07:03:35.000+05:30
Changes to add more tests.
diff --git a/core-java-modules/core-java/src/test/java/com/baeldung/serialization/SerializationUnitTest.java b/core-java-modules/core-java/src/test/java/com/baeldung/serialization/SerializationUnitTest.java
@@ -3,9 +3,11 @@
 import static org.junit.Assert.assertTrue;
 import static org.junit.Assert.assertFalse;
 
+import java.io.FileInputStream;
 import java.io.FileOutputStream;
 import java.io.IOException;
 import java.io.NotSerializableException;
+import java.io.ObjectInputStream;
 import java.io.ObjectOutputStream;
 import java.io.Serializable;
 
@@ -23,6 +25,25 @@ public void whenSerializing_ThenThrowsError() throws IOException {
 			objectOutputStream.writeObject(address);
 		}
 	}
+	
+	@Test
+    public void whenSerializingAndDeserializing_ThenObjectIsTheSame() throws IOException, ClassNotFoundException {
+        Person p = new Person();
+        p.setAge(20);
+        p.setName("Joe");
+
+        FileOutputStream fileOutputStream = new FileOutputStream("yofile.txt");
+		try (ObjectOutputStream objectOutputStream = new ObjectOutputStream(fileOutputStream)) {
+			objectOutputStream.writeObject(p);
+		}
+
+		FileInputStream fileInputStream = new FileInputStream("yofile.txt");
+		try ( ObjectInputStream objectInputStream = new ObjectInputStream(fileInputStream)) {
+			Person p2 = (Person) objectInputStream.readObject();
+			assertTrue(p2.getAge() == p.getAge());
+	        assertTrue(p2.getName().equals(p.getName()));
+		}
+	}
 
 	@Test(expected = ClassCastException.class)
 	public void whenSerializingUsingApacheCommons_ThenThrowsError() {
