inference-engine/tests/unit/mocks/mock_allocator.hpp

   1 // Copyright (C) 2018-2019 Intel Corporation
   2 // SPDX-License-Identifier: Apache-2.0
   3 //
   4
   5 //
   6 // Created by user on 20.10.16.
   7 //
   8
   9 #pragma once
  10
  11 #include <gmock/gmock.h>
  12
  13 #include "ie_allocator.hpp"
  14
  15
  16 class MockAllocator : public InferenceEngine::IAllocator
  17 {
  18 public:
  19     MOCK_QUALIFIED_METHOD0(Release, noexcept, void());
  20     MOCK_QUALIFIED_METHOD2(lock, noexcept, void*(void*, InferenceEngine::LockOp));
  21     MOCK_QUALIFIED_METHOD1(unlock, noexcept, void(void * ));
  22     MOCK_QUALIFIED_METHOD1(alloc, noexcept, void*(size_t));
  23     MOCK_QUALIFIED_METHOD1(free, noexcept, bool(void*));
  24 };
  25