Skip to content

O
openvino
Switch branch/tag
Find file
Normal viewHistoryPermalink
mock_inference_engine.hpp 2.52 KB
Newer Older
Alexey Suhov's avatar
Publishing 2019 R1 content
72660e9a
 
Alexey Suhov committed
1 
// Copyright (C) 2018-2019 Intel Corporation
openvino-pushbot's avatar
Publishing R3
866530fb
 
openvino-pushbot committed
2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 
// SPDX-License-Identifier: Apache-2.0
//

#pragma once
#include "inference_engine.hpp"
#include <gmock/gmock.h>

class MockInferenceEngine : public InferenceEngine :: IInferencePlugin
{
public:
    MOCK_QUALIFIED_METHOD2(AddExtension, noexcept, InferenceEngine::StatusCode(InferenceEngine::IExtensionPtr,
                                                                               InferenceEngine::ResponseDesc *resp));
    MOCK_QUALIFIED_METHOD1(GetVersion, noexcept, void (const InferenceEngine::Version *&));
    MOCK_QUALIFIED_METHOD0(Release, noexcept, void ());
    MOCK_QUALIFIED_METHOD1(SetLogCallback, noexcept, void (InferenceEngine::IErrorListener &));
    MOCK_QUALIFIED_METHOD2(LoadNetwork, noexcept, InferenceEngine::StatusCode(InferenceEngine::ICNNNetwork &, InferenceEngine::ResponseDesc *resp));
    MOCK_QUALIFIED_METHOD4(LoadNetwork, noexcept, InferenceEngine::StatusCode(
            InferenceEngine::IExecutableNetwork::Ptr &, InferenceEngine::ICNNNetwork &, const std::map<std::string, std::string> &, InferenceEngine::ResponseDesc *));
    MOCK_QUALIFIED_METHOD0(Unload, noexcept, void ());
    MOCK_QUALIFIED_METHOD4(ImportNetwork, noexcept, InferenceEngine::StatusCode(
            InferenceEngine::IExecutableNetwork::Ptr &, const std::string &,
            const std::map<std::string, std::string> &, InferenceEngine::ResponseDesc *));
    MOCK_QUALIFIED_METHOD3(Infer, noexcept,
                           InferenceEngine::StatusCode(
                               const InferenceEngine::Blob &,
                               InferenceEngine::Blob &,
                               InferenceEngine::ResponseDesc *resp));
    MOCK_QUALIFIED_METHOD3(Infer, noexcept,
                           InferenceEngine::StatusCode(
                               const InferenceEngine::BlobMap&,
                               std::map<std::string, InferenceEngine::Blob::Ptr> &,
                               InferenceEngine::ResponseDesc *resp));
    MOCK_QUALIFIED_METHOD2(GetPerformanceCounts,
                           const noexcept,
                           InferenceEngine::StatusCode(
                               std::map<std::string, InferenceEngine::InferenceEngineProfileInfo> & ,
                               InferenceEngine::ResponseDesc * resp));
    MOCK_QUALIFIED_METHOD2(SetConfig, noexcept, InferenceEngine::StatusCode(const std::map<std::string, std::string> &,
                                                                            InferenceEngine::ResponseDesc *resp));
};