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

   1 // Copyright (C) 2018-2019 Intel Corporation
   2 // SPDX-License-Identifier: Apache-2.0
   3 //
   4
   5 #pragma once
   6
   7 #include <ie_iinfer_request.hpp>
   8 #include "mock_executable_network_internal.hpp"
   9 #include <gmock/gmock.h>
  10 #include <string>
  11 #include <vector>
  12 #include <cpp_interfaces/impl/ie_plugin_internal.hpp>
  13 #include <cpp_interfaces/impl/ie_executable_network_internal.hpp>
  14
  15 class MockInferencePluginInternal2 : public InferenceEngine::InferencePluginInternal {
  16 public:
  17     using InferenceEngine::IInferencePluginInternal::Infer;
  18     using InferenceEngine::InferencePluginInternal::LoadNetwork;
  19     MOCK_METHOD2(LoadExeNetworkImpl, std::shared_ptr<InferenceEngine::ExecutableNetworkInternal>(
  20             InferenceEngine::ICNNNetwork &,const std::map<std::string, std::string> &));
  21     MOCK_METHOD3(LoadNetwork, void(
  22             InferenceEngine::IExecutableNetwork::Ptr &,
  23             InferenceEngine::ICNNNetwork &,
  24             const std::map<std::string, std::string> &));
  25     MOCK_METHOD1(AddExtension, void(InferenceEngine::IExtensionPtr ext_ptr));
  26     MOCK_METHOD1(SetConfig, void ( const std::map <std::string, std::string> &));
  27 };
  28
  29 class MockInferencePluginInternal : public InferenceEngine::InferencePluginInternal {
  30 public:
  31     using InferenceEngine::IInferencePluginInternal::Infer;
  32     using InferenceEngine::IInferencePluginInternal::LoadNetwork;
  33     MOCK_METHOD2(LoadExeNetworkImpl, std::shared_ptr<InferenceEngine::ExecutableNetworkInternal>(
  34             InferenceEngine::ICNNNetwork &,const std::map<std::string, std::string> &));
  35     MOCK_METHOD2(Infer, void(const InferenceEngine::BlobMap &, InferenceEngine::BlobMap&));
  36     MOCK_METHOD1(AddExtension, void(InferenceEngine::IExtensionPtr ext_ptr));
  37     MOCK_METHOD1(SetConfig, void ( const std::map <std::string, std::string> &));
  38 };
  39
  40 class MockInferencePluginInternal3 : public InferenceEngine::InferencePluginInternal {
  41 public:
  42     using InferenceEngine::IInferencePluginInternal::Infer;
  43     using InferenceEngine::IInferencePluginInternal::LoadNetwork;
  44     MOCK_METHOD2(LoadExeNetworkImpl, std::shared_ptr<InferenceEngine::ExecutableNetworkInternal>(
  45             InferenceEngine::ICNNNetwork &,const std::map<std::string, std::string> &));
  46     MOCK_METHOD1(AddExtension, void(InferenceEngine::IExtensionPtr ext_ptr));
  47     MOCK_METHOD1(SetConfig, void ( const std::map <std::string, std::string> &));
  48 };