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

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