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

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