- final ShardDataTreeSnapshot actual = ShardDataTreeSnapshot.deserialize(reply.getSnapshot());
- assertNotNull(actual);
- assertEquals(snapshot.getClass(), actual.getClass());
+ final CaptureSnapshotReply reply = kit.expectMsgClass(Duration.ofSeconds(3), CaptureSnapshotReply.class);
+ assertNotNull("getSnapshotState is null", reply.getSnapshotState());
+ assertEquals("SnapshotState type", ShardSnapshotState.class, reply.getSnapshotState().getClass());
+ assertEquals("Snapshot", snapshot, ((ShardSnapshotState)reply.getSnapshotState()).getSnapshot());