inference-engine/tests/unit/mocks/cpp_interfaces/interface/mock_imemory_state_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/interface/ie_imemory_state_internal.hpp>
  13
  14 class MockIMemoryStateInternal : public InferenceEngine::IMemoryStateInternal {
  15  public:
  16     MOCK_CONST_METHOD0(GetName, std::string ());
  17     MOCK_METHOD0(Reset, void ());
  18     MOCK_METHOD1(SetState, void (Blob::Ptr ));
  19     MOCK_CONST_METHOD0(GetLastState, Blob::CPtr ());
  20 };