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

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