Bug 7521: Convert DatastoreSnapshot.ShardSnapshot to store Snapshot

[controller.git] / opendaylight / md-sal / sal-distributed-datastore / src / test / java / org / opendaylight / controller / cluster / datastore / actors / ShardSnapshotActorTest.java

diff --git a/opendaylight/md-sal/sal-distributed-datastore/src/test/java/org/opendaylight/controller/cluster/datastore/actors/ShardSnapshotActorTest.java b/opendaylight/md-sal/sal-distributed-datastore/src/test/java/org/opendaylight/controller/cluster/datastore/actors/ShardSnapshotActorTest.java
index 5cf81834d8d40c4ec2c5ca9cac78cb987492d025..ae9499744f924fd1f932565e4057c9e3519b9091 100644 (file)
--- a/opendaylight/md-sal/sal-distributed-datastore/src/test/java/org/opendaylight/controller/cluster/datastore/actors/ShardSnapshotActorTest.java
+++ b/opendaylight/md-sal/sal-distributed-datastore/src/test/java/org/opendaylight/controller/cluster/datastore/actors/ShardSnapshotActorTest.java
@@ -14,6 +14,7 @@ import akka.actor.ActorRef;
 import akka.testkit.JavaTestKit;
 import java.io.ByteArrayInputStream;
 import java.io.ByteArrayOutputStream;
+import java.io.ObjectInputStream;
 import java.util.Optional;
 import org.junit.Test;
 import org.opendaylight.controller.cluster.datastore.AbstractActorTest;
@@ -47,8 +48,12 @@ public class ShardSnapshotActorTest extends AbstractActorTest {
                 assertEquals("Snapshot", snapshot, ((ShardSnapshotState)reply.getSnapshotState()).getSnapshot());
 
                 if (installSnapshotStream != null) {
-                    final ShardDataTreeSnapshot deserialized = ShardDataTreeSnapshot.deserialize(
-                            new ByteArrayInputStream(installSnapshotStream.toByteArray()));
+                    final ShardDataTreeSnapshot deserialized;
+                    try (final ObjectInputStream in = new ObjectInputStream(new ByteArrayInputStream(
+                            installSnapshotStream.toByteArray()))) {
+                        deserialized = ShardDataTreeSnapshot.deserialize(in);
+                    }
+
                     assertEquals("Deserialized snapshot type", snapshot.getClass(), deserialized.getClass());
 
                     final Optional<NormalizedNode<?, ?>> maybeNode = deserialized.getRootNode();