From 99367eb363a4911a9eb098bb7ca6fb4f3c74f15b Mon Sep 17 00:00:00 2001
From: zhangxiaokun <zhang.xiaokun@huawei.com>
Date: Tue, 6 Jul 2021 10:03:38 +0800
Subject: [PATCH] Add UT for InnerSession

---
 tests/ut/ge/session/inner_session_unittest.cc | 19 +++++++------------
 1 file changed, 7 insertions(+), 12 deletions(-)

diff --git a/tests/ut/ge/session/inner_session_unittest.cc b/tests/ut/ge/session/inner_session_unittest.cc
index 0d20f06a..80cc2834 100644
--- a/tests/ut/ge/session/inner_session_unittest.cc
+++ b/tests/ut/ge/session/inner_session_unittest.cc
@@ -19,21 +19,18 @@
 #define private public
 #define protected public
 #include "session/inner_session.h"
-#undef private
-#undef protected
-
 
 using namespace std;
 
 namespace ge {
-class Utest_Inner_session : public testing::Test {
+class UtestInnerSession : public testing::Test {
  protected:
   void SetUp() override {}
 
   void TearDown() override {}
 };
 
-TEST_F(Utest_Inner_session, build_graph_success) {
+TEST_F(UtestInnerSession, build_graph_success) {
   std::map <string, string> options;
   uint64_t session_id = 1;
   InnerSession inner_seesion(session_id, options);
@@ -44,17 +41,15 @@ TEST_F(Utest_Inner_session, build_graph_success) {
   EXPECT_NE(ret, ge::SUCCESS);
 }
 
-TEST_F(Utest_Inner_session, initialize) {
-  std::map<std::string, std::string> options = {
-    {ge::MODIFY_MIXLIST, "/modify.json"}
-  };
+TEST_F(UtestInnerSession, initialize) {
+  std::map<std::string, std::string> options = {};
   uint64_t session_id = 1;
   InnerSession inner_session(session_id, options);
-  auto ret = inner_session.Initialize();
-  EXPECT_NE(ret, ge::SUCCESS);
+  EXPECT_EQ(inner_session.Initialize(), SUCCESS);
+  EXPECT_EQ(inner_session.Finalize(), SUCCESS);
 }
 
-TEST_F(Utest_Inner_session, check_op_precision_mode) {
+TEST_F(UtestInnerSession, check_op_precision_mode) {
   std::map<std::string, std::string> options = {
     {ge::OP_PRECISION_MODE, "./op_precision_mode.ini"}
   };