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

   1 // Copyright (C) 2018 Intel Corporation
   2 //
   3 // SPDX-License-Identifier: Apache-2.0
   4 //
   5
   6 #pragma once
   7
   8 #include "ie_plugin.hpp"
   9 #include "ie_iexecutable_network.hpp"
  10 #include <gmock/gmock.h>
  11 #include <string>
  12 #include <vector>
  13 #include <cpp_interfaces/impl/ie_executable_network_thread_safe_async_only.hpp>
  14
  15 using namespace InferenceEngine;
  16
  17 class MockExecutableNetworkThreadSafeAsyncOnly : public ExecutableNetworkThreadSafeAsyncOnly {
  18 public:
  19     MOCK_METHOD2(CreateAsyncInferRequestImpl,
  20                  AsyncInferRequestInternal::Ptr(InputsDataMap networkInputs, OutputsDataMap networkOutputs));
  21     MOCK_METHOD1(Export, void(const std::string &));
  22     MOCK_METHOD1(GetMappedTopology, void(std::map<std::string, std::vector<PrimitiveInfo::Ptr>> &));
  23 };