inference-engine/tests/unit/mocks/cpp_interfaces/mock_task_synchronizer.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 <cpp_interfaces/ie_task_synchronizer.hpp>
   9 #include <gmock/gmock.h>
  10 #include <string>
  11 #include <vector>
  12 #include <memory>
  13
  14 class MockTaskSynchronizer : public InferenceEngine::TaskSynchronizer {
  15 public:
  16     typedef std::shared_ptr<MockTaskSynchronizer> Ptr;
  17     MOCK_METHOD0(lock, void());
  18     MOCK_METHOD0(unlock, void());
  19 };
  20
  21 class MockTaskSynchronizerPrivate : public InferenceEngine::TaskSynchronizer {
  22 public:
  23     typedef std::shared_ptr<MockTaskSynchronizerPrivate> Ptr;
  24     MOCK_METHOD0(_addTaskToQueue, unsigned int());
  25     MOCK_METHOD1(_waitInQueue, void(unsigned int));
  26 };