inference-engine/tests/unit/mocks/cpp_interfaces/interface/mock_iasync_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 <gmock/gmock.h>
   8 #include <string>
   9 #include <vector>
  10 #include <ie_iinfer_request.hpp>
  11 #include <cpp_interfaces/interface/ie_iinfer_async_request_internal.hpp>
  12
  13 class MockIAsyncInferRequestInternal : public InferenceEngine::IAsyncInferRequestInternal {
  14 public:
  15     MOCK_METHOD0(StartAsync, void());
  16     MOCK_METHOD1(Wait, InferenceEngine::StatusCode(int64_t));
  17     MOCK_METHOD1(GetUserData, void(void **));
  18     MOCK_METHOD1(SetUserData, void(void *));
  19     MOCK_METHOD0(Infer, void());
  20     MOCK_CONST_METHOD1(GetPerformanceCounts, void(std::map<std::string, InferenceEngine::InferenceEngineProfileInfo> &));
  21     MOCK_METHOD2(SetBlob, void(const char *name, const InferenceEngine::Blob::Ptr &));
  22     MOCK_METHOD2(GetBlob, void(const char *name, InferenceEngine::Blob::Ptr &));
  23     MOCK_METHOD1(SetCompletionCallback, void(InferenceEngine::IInferRequest::CompletionCallback));
  24         MOCK_METHOD1(SetBatch, void(int));
  25 };