diff --git a/computer-test/src/main/java/org/apache/hugegraph/computer/core/compute/ComputeManagerTest.java b/computer-test/src/main/java/org/apache/hugegraph/computer/core/compute/ComputeManagerTest.java
index c4052497b..a43baef84 100644
--- a/computer-test/src/main/java/org/apache/hugegraph/computer/core/compute/ComputeManagerTest.java
+++ b/computer-test/src/main/java/org/apache/hugegraph/computer/core/compute/ComputeManagerTest.java
@@ -47,6 +47,7 @@
 import org.apache.hugegraph.computer.core.receiver.MessageRecvManager;
 import org.apache.hugegraph.computer.core.receiver.ReceiverUtil;
 import org.apache.hugegraph.computer.core.sender.MessageSendManager;
+import org.apache.hugegraph.computer.core.snapshot.SnapshotManager;
 import org.apache.hugegraph.computer.core.sort.sorting.SendSortManager;
 import org.apache.hugegraph.computer.core.sort.sorting.SortManager;
 import org.apache.hugegraph.computer.core.store.FileManager;
@@ -104,6 +105,11 @@ public void setup() {
                                                                    fileManager,
                                                                    sortManager);
         this.managers.add(receiveManager);
+        SnapshotManager snapshotManager = new SnapshotManager(context(),
+                                                   null,
+                                                              receiveManager,
+                                                     null);
+        this.managers.add(snapshotManager);
         this.managers.initAll(this.config);
         this.connectionId = new ConnectionId(new InetSocketAddress("localhost",
                                                                    8081),
diff --git a/computer-test/src/main/java/org/apache/hugegraph/computer/core/compute/input/EdgesInputTest.java b/computer-test/src/main/java/org/apache/hugegraph/computer/core/compute/input/EdgesInputTest.java
index c2126d1bd..b6c427596 100644
--- a/computer-test/src/main/java/org/apache/hugegraph/computer/core/compute/input/EdgesInputTest.java
+++ b/computer-test/src/main/java/org/apache/hugegraph/computer/core/compute/input/EdgesInputTest.java
@@ -51,6 +51,7 @@
 import org.apache.hugegraph.computer.core.receiver.MessageRecvManager;
 import org.apache.hugegraph.computer.core.receiver.ReceiverUtil;
 import org.apache.hugegraph.computer.core.sender.MessageSendManager;
+import org.apache.hugegraph.computer.core.snapshot.SnapshotManager;
 import org.apache.hugegraph.computer.core.sort.flusher.PeekableIterator;
 import org.apache.hugegraph.computer.core.sort.sorting.SendSortManager;
 import org.apache.hugegraph.computer.core.sort.sorting.SortManager;
@@ -137,6 +138,11 @@ private void testEdgeFreq(EdgeFrequency freq)
                                                                    fileManager,
                                                                    sortManager);
         this.managers.add(receiveManager);
+        SnapshotManager snapshotManager = new SnapshotManager(context(),
+                                                   null,
+                                                              receiveManager,
+                                                     null);
+        this.managers.add(snapshotManager);
         this.managers.initAll(this.config);
         ConnectionId connectionId = new ConnectionId(new InetSocketAddress(
                                                      "localhost", 8081),
diff --git a/computer-test/src/main/java/org/apache/hugegraph/computer/core/compute/input/MessageInputTest.java b/computer-test/src/main/java/org/apache/hugegraph/computer/core/compute/input/MessageInputTest.java
index 36361380f..3889f80f7 100644
--- a/computer-test/src/main/java/org/apache/hugegraph/computer/core/compute/input/MessageInputTest.java
+++ b/computer-test/src/main/java/org/apache/hugegraph/computer/core/compute/input/MessageInputTest.java
@@ -41,6 +41,7 @@
 import org.apache.hugegraph.computer.core.network.message.MessageType;
 import org.apache.hugegraph.computer.core.receiver.MessageRecvManager;
 import org.apache.hugegraph.computer.core.receiver.ReceiverUtil;
+import org.apache.hugegraph.computer.core.snapshot.SnapshotManager;
 import org.apache.hugegraph.computer.core.sort.flusher.PeekableIterator;
 import org.apache.hugegraph.computer.core.sort.sorting.RecvSortManager;
 import org.apache.hugegraph.computer.core.sort.sorting.SortManager;
@@ -90,6 +91,11 @@ public void setup() {
                                                                    fileManager,
                                                                    sortManager);
         this.managers.add(receiveManager);
+        SnapshotManager snapshotManager = new SnapshotManager(context(),
+                                                   null,
+                                                              receiveManager,
+                                                     null);
+        this.managers.add(snapshotManager);
         this.managers.initAll(this.config);
         this.connectionId = new ConnectionId(new InetSocketAddress("localhost",
                                                                    8081),
diff --git a/computer-test/src/main/java/org/apache/hugegraph/computer/core/network/DataServerManagerTest.java b/computer-test/src/main/java/org/apache/hugegraph/computer/core/network/DataServerManagerTest.java
index 89029ec14..6c8673c54 100644
--- a/computer-test/src/main/java/org/apache/hugegraph/computer/core/network/DataServerManagerTest.java
+++ b/computer-test/src/main/java/org/apache/hugegraph/computer/core/network/DataServerManagerTest.java
@@ -25,6 +25,7 @@
 import org.apache.hugegraph.computer.core.network.connection.ConnectionManager;
 import org.apache.hugegraph.computer.core.network.connection.TransportConnectionManager;
 import org.apache.hugegraph.computer.core.receiver.MessageRecvManager;
+import org.apache.hugegraph.computer.core.snapshot.SnapshotManager;
 import org.apache.hugegraph.computer.core.sort.sorting.RecvSortManager;
 import org.apache.hugegraph.computer.core.sort.sorting.SortManager;
 import org.apache.hugegraph.computer.core.store.FileManager;
@@ -55,6 +56,10 @@ public void test() {
         MessageRecvManager recvManager = new MessageRecvManager(context(),
                                                                 fileManager,
                                                                 sortManager);
+        SnapshotManager snapshotManager = new SnapshotManager(context(),
+                                                   null,
+                                                              recvManager,
+                                                     null);
         recvManager.init(config);
         ConnectionManager connManager = new TransportConnectionManager();
         DataServerManager serverManager = new DataServerManager(connManager,
diff --git a/computer-test/src/main/java/org/apache/hugegraph/computer/core/receiver/MessageRecvManagerTest.java b/computer-test/src/main/java/org/apache/hugegraph/computer/core/receiver/MessageRecvManagerTest.java
index 3942b4eff..8887b7395 100644
--- a/computer-test/src/main/java/org/apache/hugegraph/computer/core/receiver/MessageRecvManagerTest.java
+++ b/computer-test/src/main/java/org/apache/hugegraph/computer/core/receiver/MessageRecvManagerTest.java
@@ -32,6 +32,7 @@
 import org.apache.hugegraph.computer.core.receiver.edge.EdgeMessageRecvPartitionTest;
 import org.apache.hugegraph.computer.core.receiver.message.ComputeMessageRecvPartitionTest;
 import org.apache.hugegraph.computer.core.receiver.vertex.VertexMessageRecvPartitionTest;
+import org.apache.hugegraph.computer.core.snapshot.SnapshotManager;
 import org.apache.hugegraph.computer.core.sort.flusher.PeekableIterator;
 import org.apache.hugegraph.computer.core.sort.sorting.RecvSortManager;
 import org.apache.hugegraph.computer.core.sort.sorting.SortManager;
@@ -49,6 +50,7 @@ public class MessageRecvManagerTest extends UnitTestBase {
     private FileManager fileManager;
     private SortManager sortManager;
     private MessageRecvManager receiveManager;
+    private SnapshotManager snapshotManager;
     private ConnectionId connectionId;
 
     @Before
@@ -73,6 +75,10 @@ public void setup() {
         this.receiveManager = new MessageRecvManager(context(),
                                                      this.fileManager,
                                                      this.sortManager);
+        this.snapshotManager = new SnapshotManager(context(),
+                                        null,
+                                                   receiveManager,
+                                          null);
         this.receiveManager.init(this.config);
         this.connectionId = new ConnectionId(
                             new InetSocketAddress("localhost",8081),