inference-engine/tests/unit/mocks/cpp_interfaces/impl/mock_infer_request_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_plugin.hpp"
   8 #include "ie_iexecutable_network.hpp"
   9 #include <gmock/gmock.h>
  10 #include <string>
  11 #include <vector>
  12 #include <cpp_interfaces/impl/ie_infer_request_internal.hpp>
  13
  14 using namespace InferenceEngine;
  15
  16 class MockInferRequestInternal : public InferRequestInternal {
  17 public:
  18     MockInferRequestInternal(InputsDataMap networkInputs, OutputsDataMap networkOutputs)
  19             : InferRequestInternal(networkInputs, networkOutputs) {}
  20     using InferRequestInternal::SetBlob;
  21     using InferRequestInternal::GetBlob;
  22     MOCK_METHOD0(InferImpl, void());
  23     MOCK_CONST_METHOD1(GetPerformanceCounts, void(std::map<std::string, InferenceEngineProfileInfo> &));
  24 };