From 7a8a3599e141ecad170e5348e9a8cc7bd58d2700 Mon Sep 17 00:00:00 2001
From: zhaozhixuan <zhaozhixuan2@hisilicon.com>
Date: Mon, 31 May 2021 13:55:50 +0800
Subject: [PATCH] Add ut.

---
 .../hybrid_model_async_executor_unittest.cc    | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/tests/ut/ge/hybrid/executor/hybrid_model_async_executor_unittest.cc b/tests/ut/ge/hybrid/executor/hybrid_model_async_executor_unittest.cc
index b4091a50..d2679439 100644
--- a/tests/ut/ge/hybrid/executor/hybrid_model_async_executor_unittest.cc
+++ b/tests/ut/ge/hybrid/executor/hybrid_model_async_executor_unittest.cc
@@ -86,4 +86,22 @@ TEST_F(UtestHybridModelAsyncExecutor, BuildDeviceTensor) {
   auto size = tensor.GetSize();
   ASSERT_EQ(size, 100);
 }
+
+TEST_F(UtestHybridModelAsyncExecutor, Test_execute_internal) {
+  ComputeGraphPtr graph = std::make_shared<ComputeGraph>("test");
+  GeRootModelPtr ge_root_model = make_shared<GeRootModel>(graph);
+  ge_root_model->SetModelName("test_name");
+  HybridModel hybrid_model(ge_root_model);
+
+  HybridModelExecutor executor(&hybrid_model, 0, nullptr);
+  ASSERT_EQ(executor.Init(), SUCCESS);
+  auto &context = executor.context_;
+  GraphItem graph_item;
+  SubgraphExecutor subgraph_executor(&graph_item, &context);
+  HybridModelExecutor::ExecuteArgs args;
+  std::pair<rtEvent_t, std::pair<rtCallback_t, void *>> eof_entry;
+  eof_entry.first = nullptr;
+  context.callback_manager->callback_queue_.Push(eof_entry);
+  ASSERT_EQ(executor.ExecuteGraphInternal(subgraph_executor, args), SUCCESS);
+}
 } // namespace ge
\ No newline at end of file