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

   1 // Copyright (C) 2018-2019 Intel Corporation
   2 // SPDX-License-Identifier: Apache-2.0
   3 //
   4
   5 /**
   6  * \brief header file for MockIExecutableNetwork
   7  * \file mock_iexecutable_network.hpp
   8  */
   9 #pragma once
  10
  11 #include "ie_iexecutable_network.hpp"
  12 #include <gmock/gmock-generated-function-mockers.h>
  13
  14 using namespace InferenceEngine;
  15
  16 class MockIExecutableNetwork : public IExecutableNetwork {
  17 public:
  18     MOCK_QUALIFIED_METHOD2(GetOutputsInfo, const noexcept, StatusCode (ConstOutputsDataMap  &, ResponseDesc *));
  19     MOCK_QUALIFIED_METHOD2(GetInputsInfo, const noexcept, StatusCode (ConstInputsDataMap &, ResponseDesc *));
  20     MOCK_QUALIFIED_METHOD2(CreateInferRequest, noexcept, StatusCode(IInferRequest::Ptr &, ResponseDesc*));
  21     MOCK_QUALIFIED_METHOD2(Export, noexcept, StatusCode(const std::string &, ResponseDesc*));
  22     MOCK_QUALIFIED_METHOD2(GetMappedTopology, noexcept, StatusCode(std::map<std::string, std::vector<PrimitiveInfo::Ptr>> &, ResponseDesc*));
  23     MOCK_QUALIFIED_METHOD0(Release, noexcept, void ());
  24     MOCK_QUALIFIED_METHOD3(QueryState, noexcept, StatusCode(IMemoryState::Ptr &, size_t  , ResponseDesc*));
  25     MOCK_QUALIFIED_METHOD2(GetExecGraphInfo, noexcept, StatusCode(ICNNNetwork::Ptr &, ResponseDesc*));
  26 };