inference-engine/tests/unit/mocks/cpp_interfaces/impl/mock_async_infer_request_default.hpp

   1 // Copyright (C) 2018 Intel Corporation
   2 //
   3 // SPDX-License-Identifier: Apache-2.0
   4 //
   5
   6 #pragma once
   7
   8 #include <string>
   9 #include <vector>
  10
  11 #include <gmock/gmock.h>
  12 #include "cpp_interfaces/impl/mock_infer_request_internal.hpp"
  13
  14 #include <ie_iinfer_request.hpp>
  15 #include <cpp_interfaces/impl/ie_infer_async_request_thread_safe_default.hpp>
  16
  17 using namespace InferenceEngine;
  18
  19 class MockAsyncInferRequestDefault : public AsyncInferRequestThreadSafeDefault {
  20 public:
  21     MockAsyncInferRequestDefault(InferRequestInternal::Ptr request,
  22                                  const ITaskExecutor::Ptr &taskExecutor,
  23                                  const TaskSynchronizer::Ptr &taskSynchronizer,
  24                                  const ITaskExecutor::Ptr &callbackExecutor)
  25             : AsyncInferRequestThreadSafeDefault(request, taskExecutor, taskSynchronizer, callbackExecutor) {}
  26
  27     MOCK_METHOD0(startAsyncTask, void());
  28
  29     MOCK_METHOD0(initNextAsyncTask, void());
  30 };