intel/compute-runtime
1
0
Fork 0
mirror of https://github.com/intel/compute-runtime.git synced 2026-01-05 17:20:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

Rename L0 Device method

Browse Source 
remove gmock from mock device
add macros to define mock methods with ref return

Related-To: NEO-4914
Signed-off-by: Mateusz Jablonski <mateusz.jablonski@intel.com>
This commit is contained in:
Mateusz Jablonski
2021-11-16 18:37:15 +00:00
committed by Compute-Runtime-Automation
parent 35f135bbc8
commit babdbfb0cf
7 changed files with 33 additions and 43 deletions
Download Patch File Download Diff File Expand all files Collapse all files

40
level_zero/core/test/unit_tests/mocks/mock_device.h
View File

@@ -6,19 +6,12 @@
*/
#pragma once
#include "shared/source/command_stream/preemption_mode.h"
#include "shared/source/device/device.h"
#include "shared/source/device/device_info.h"
#include "shared/test/common/test_macros/mock_method_macros.h"
#include "level_zero/core/source/device/device.h"
#include "level_zero/core/source/device/device_imp.h"
#include "level_zero/core/source/driver/driver_handle.h"
#include "level_zero/core/test/unit_tests/mock.h"
#include "level_zero/core/test/unit_tests/white_box.h"
#include "level_zero/tools/source/metrics/metric.h"
#include "gmock/gmock.h"
namespace L0 {
namespace ult {
@@ -58,40 +51,21 @@ struct Mock<Device> : public Device {
ADDMETHOD_NOBASE(systemBarrier, ze_result_t, ZE_RESULT_SUCCESS, ());
// Runtime internal methods
ADDMETHOD_NOBASE(getExecEnvironment, void *, nullptr, ());
MOCK_METHOD(NEO::HwHelper &,
getHwHelper,
(),
(override));
ADDMETHOD_NOBASE_REFRETURN(getHwHelper, NEO::HwHelper &, ());
ADDMETHOD_CONST_NOBASE(isMultiDeviceCapable, bool, false, ());
ADDMETHOD_NOBASE(getBuiltinFunctionsLib, BuiltinFunctionsLib *, nullptr, ());
ADDMETHOD_CONST_NOBASE(getMaxNumHwThreads, uint32_t, 16u, ());
ADDMETHOD_NOBASE(activateMetricGroups, ze_result_t, ZE_RESULT_SUCCESS, (uint32_t count, zet_metric_group_handle_t *phMetricGroups));
MOCK_METHOD(NEO::OSInterface &,
getOsInterface,
(),
(override));
ADDMETHOD_NOBASE(activateMetricGroupsDeferred, ze_result_t, ZE_RESULT_SUCCESS, (uint32_t count, zet_metric_group_handle_t *phMetricGroups));
ADDMETHOD_NOBASE_REFRETURN(getOsInterface, NEO::OSInterface &, ());
ADDMETHOD_CONST_NOBASE(getPlatformInfo, uint32_t, 0u, ());
MOCK_METHOD(MetricContext &,
getMetricContext,
(),
(override));
MOCK_METHOD(const NEO::HardwareInfo &,
getHwInfo,
(),
(const, override));
ADDMETHOD_NOBASE_REFRETURN(getMetricContext, MetricContext &, ());
ADDMETHOD_CONST_NOBASE_REFRETURN(getHwInfo, const NEO::HardwareInfo &, ());
ADDMETHOD_NOBASE(getDriverHandle, L0::DriverHandle *, nullptr, ());
ADDMETHOD_NOBASE_VOIDRETURN(setDriverHandle, (L0::DriverHandle *));
ADDMETHOD_CONST_NOBASE(getDevicePreemptionMode, NEO::PreemptionMode, NEO::PreemptionMode::Initial, ());
MOCK_METHOD(const NEO::DeviceInfo &,
getDeviceInfo,
(),
(const, override));
ADDMETHOD_CONST_NOBASE_REFRETURN(getDeviceInfo, const NEO::DeviceInfo &, ());
ADDMETHOD_NOBASE(getNEODevice, NEO::Device *, nullptr, ());
MOCK_METHOD(void,
activateMetricGroups,
(),
(override));
ADDMETHOD_NOBASE_VOIDRETURN(activateMetricGroups, ());
ADDMETHOD_CONST_NOBASE(getDebugSurface, NEO::GraphicsAllocation *, nullptr, ());
ADDMETHOD_NOBASE(allocateManagedMemoryFromHostPtr, NEO::GraphicsAllocation *, nullptr, (void *buffer, size_t size, struct L0::CommandList *commandList));
ADDMETHOD_NOBASE(allocateMemoryFromHostPtr, NEO::GraphicsAllocation *, nullptr, (const void *buffer, size_t size, bool hostCopyAllowed));

3
level_zero/core/test/unit_tests/mocks/mock_driver_handle.cpp
View File

@@ -15,9 +15,6 @@ namespace L0 {
namespace ult {
using MockDriverHandle = Mock<L0::ult::DriverHandle>;
using namespace testing;
using ::testing::Invoke;
using ::testing::Return;
Mock<DriverHandle>::Mock() = default;