inference-engine/tests/unit/mocks/cpp_interfaces/interface/mock_iexecutable_network_internal.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/interface/mock_iinfer_request_internal.hpp"
  12 #include "cpp_interfaces/impl/mock_infer_request_internal.hpp"
  13
  14 #include "ie_plugin.hpp"
  15 #include "ie_input_info.hpp"
  16 #include "ie_icnn_network.hpp"
  17 #include "ie_iexecutable_network.hpp"
  18 #include <cpp_interfaces/impl/ie_executable_network_internal.hpp>
  19 #include <cpp_interfaces/impl/ie_infer_request_internal.hpp>
  20
  21 using namespace InferenceEngine;
  22
  23 class MockIExecutableNetworkInternal : public IExecutableNetworkInternal {
  24 public:
  25     MOCK_CONST_METHOD0(GetOutputsInfo, ConstOutputsDataMap ());
  26     MOCK_CONST_METHOD0(GetInputsInfo, ConstInputsDataMap ());
  27     MOCK_METHOD1(CreateInferRequest, void(IInferRequest::Ptr &));
  28     MOCK_METHOD1(Export, void(const std::string &));
  29     MOCK_METHOD1(GetMappedTopology, void(std::map<std::string, std::vector<PrimitiveInfo::Ptr>> &));
  30     MOCK_METHOD0(QueryState, std::vector<IMemoryStateInternal::Ptr>());
  31     MOCK_METHOD1(GetExecGraphInfo, void(ICNNNetwork::Ptr &));
  32 };