inference-engine/tests/unit/mocks/mock_plugin_dispatcher.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_dispatcher.hpp"
   8 #include "ie_plugin_ptr.hpp"
   9 #include <gmock/gmock.h>
  10 #include <string>
  11 #include <vector>
  12
  13 class MockDispatcher : public InferenceEngine::PluginDispatcher {
  14 public:
  15     MockDispatcher(const std::vector<std::string>& pp) : PluginDispatcher(pp) {}
  16     MOCK_CONST_METHOD1(getPluginByName, InferenceEngine::InferencePlugin(const std::string& name));
  17 };