inference-engine/tests/unit/mocks/mock_inference_engine.hpp

   1 // Copyright (C) 2018 Intel Corporation
   2 //
   3 // SPDX-License-Identifier: Apache-2.0
   4 //
   5
   6 #pragma once
   7 #include "inference_engine.hpp"
   8 #include <gmock/gmock.h>
   9
  10 class MockInferenceEngine : public InferenceEngine :: IInferencePlugin
  11 {
  12 public:
  13     MOCK_QUALIFIED_METHOD2(AddExtension, noexcept, InferenceEngine::StatusCode(InferenceEngine::IExtensionPtr,
  14                                                                                InferenceEngine::ResponseDesc *resp));
  15     MOCK_QUALIFIED_METHOD1(GetVersion, noexcept, void (const InferenceEngine::Version *&));
  16     MOCK_QUALIFIED_METHOD0(Release, noexcept, void ());
  17     MOCK_QUALIFIED_METHOD1(SetLogCallback, noexcept, void (InferenceEngine::IErrorListener &));
  18     MOCK_QUALIFIED_METHOD2(LoadNetwork, noexcept, InferenceEngine::StatusCode(InferenceEngine::ICNNNetwork &, InferenceEngine::ResponseDesc *resp));
  19     MOCK_QUALIFIED_METHOD4(LoadNetwork, noexcept, InferenceEngine::StatusCode(
  20             InferenceEngine::IExecutableNetwork::Ptr &, InferenceEngine::ICNNNetwork &, const std::map<std::string, std::string> &, InferenceEngine::ResponseDesc *));
  21     MOCK_QUALIFIED_METHOD0(Unload, noexcept, void ());
  22     MOCK_QUALIFIED_METHOD4(ImportNetwork, noexcept, InferenceEngine::StatusCode(
  23             InferenceEngine::IExecutableNetwork::Ptr &, const std::string &,
  24             const std::map<std::string, std::string> &, InferenceEngine::ResponseDesc *));
  25     MOCK_QUALIFIED_METHOD3(Infer, noexcept,
  26                            InferenceEngine::StatusCode(
  27                                const InferenceEngine::Blob &,
  28                                InferenceEngine::Blob &,
  29                                InferenceEngine::ResponseDesc *resp));
  30     MOCK_QUALIFIED_METHOD3(Infer, noexcept,
  31                            InferenceEngine::StatusCode(
  32                                const InferenceEngine::BlobMap&,
  33                                std::map<std::string, InferenceEngine::Blob::Ptr> &,
  34                                InferenceEngine::ResponseDesc *resp));
  35     MOCK_QUALIFIED_METHOD2(GetPerformanceCounts,
  36                            const noexcept,
  37                            InferenceEngine::StatusCode(
  38                                std::map<std::string, InferenceEngine::InferenceEngineProfileInfo> & ,
  39                                InferenceEngine::ResponseDesc * resp));
  40     MOCK_QUALIFIED_METHOD2(SetConfig, noexcept, InferenceEngine::StatusCode(const std::map<std::string, std::string> &,
  41                                                                             InferenceEngine::ResponseDesc *resp));
  42 };
  43