From 7f75c2a1845b935da1d9695685c211d5bddf56e2 Mon Sep 17 00:00:00 2001
From: zhaozhixuan <zhaozhixuan2@hisilicon.com>
Date: Tue, 11 May 2021 09:15:08 +0800
Subject: [PATCH] Fix ut.

---
 tests/ut/ge/graph/manager/graph_manager_unittest.cc | 20 ++++++++++----------
 1 file changed, 10 insertions(+), 10 deletions(-)

diff --git a/tests/ut/ge/graph/manager/graph_manager_unittest.cc b/tests/ut/ge/graph/manager/graph_manager_unittest.cc
index 13637a87..8ee6de11 100644
--- a/tests/ut/ge/graph/manager/graph_manager_unittest.cc
+++ b/tests/ut/ge/graph/manager/graph_manager_unittest.cc
@@ -194,16 +194,16 @@ TEST_F(UtestGraphManagerTest, test_add_graph_3) {
   std::map<std::string, std::string> options;
   OmgContext context;
 
-  // std::future<Status> fut1 = std::async(std::launch::async,
-  //     &GraphManager::AddGraph, &graph_manager, graph_id, graph, options, context);
-  // std::future<Status> fut2 = std::async(std::launch::async,
-  //     &GraphManager::AddGraph, &graph_manager, graph_id, graph, options, context);
-  // fut1.wait();
-  // fut2.wait();
-  // Status status1 = fut1.get();
-  // Status status2 = fut2.get();
-  // EXPECT_EQ(status1, ge::SUCCESS);
-  // EXPECT_EQ(status2, ge::SUCCESS);
+  std::future<Status> fut1 = std::async(std::launch::async,
+      &GraphManager::AddGraph, &graph_manager, graph_id, graph, options, context);
+  std::future<Status> fut2 = std::async(std::launch::async,
+      &GraphManager::AddGraph, &graph_manager, graph_id, graph, options, context);
+  fut1.wait();
+  fut2.wait();
+  Status status1 = fut1.get();
+  Status status2 = fut2.get();
+  EXPECT_EQ(status1, ge::SUCCESS);
+  EXPECT_EQ(status2, ge::SUCCESS);
 }
 
 TEST_F(UtestGraphManagerTest, test_remove_graph_1) {