66 */
77
88#pragma once
9- #include " level_zero/core/source/device/device.h"
9+ #include " shared/test/common/test_macros/mock_method_macros.h"
10+
1011#include " level_zero/core/source/event/event.h"
1112#include " level_zero/core/test/unit_tests/mock.h"
1213#include " level_zero/core/test/unit_tests/white_box.h"
1314
14- #include " gmock/gmock.h"
15-
16- #include < vector>
17-
18- #if defined(__clang__)
19- #pragma clang diagnostic push
20- #pragma clang diagnostic ignored "-Winconsistent-missing-override"
21- #endif
22-
2315namespace L0 {
2416namespace ult {
2517
@@ -43,14 +35,13 @@ struct Mock<Event> : public Event {
4335 Mock ();
4436 ~Mock () override ;
4537
46- MOCK_METHOD3 (create, L0::Event *(::L0::EventPool *eventPool, const ze_event_desc_t *desc, ::L0::Device *device));
47- MOCK_METHOD0 (destroy, ze_result_t ());
48- MOCK_METHOD0 (hostSignal, ze_result_t ());
49- MOCK_METHOD1 (hostSynchronize, ze_result_t (uint64_t timeout));
50- MOCK_METHOD0 (queryStatus, ze_result_t ());
51- MOCK_METHOD0 (reset, ze_result_t ());
52- MOCK_METHOD1 (queryKernelTimestamp, ze_result_t (ze_kernel_timestamp_result_t *dstptr));
53- MOCK_METHOD3 (queryTimestampsExp, ze_result_t (::L0::Device *device, uint32_t *pCount, ze_kernel_timestamp_result_t *pTimestamps));
38+ ADDMETHOD_NOBASE (destroy, ze_result_t , ZE_RESULT_SUCCESS, ());
39+ ADDMETHOD_NOBASE (hostSignal, ze_result_t , ZE_RESULT_SUCCESS, ());
40+ ADDMETHOD_NOBASE (hostSynchronize, ze_result_t , ZE_RESULT_SUCCESS, (uint64_t timeout));
41+ ADDMETHOD_NOBASE (queryStatus, ze_result_t , ZE_RESULT_SUCCESS, ());
42+ ADDMETHOD_NOBASE (reset, ze_result_t , ZE_RESULT_SUCCESS, ());
43+ ADDMETHOD_NOBASE (queryKernelTimestamp, ze_result_t , ZE_RESULT_SUCCESS, (ze_kernel_timestamp_result_t * dstptr));
44+ ADDMETHOD_NOBASE (queryTimestampsExp, ze_result_t , ZE_RESULT_SUCCESS, (::L0::Device * device, uint32_t *pCount, ze_kernel_timestamp_result_t *pTimestamps));
5445
5546 // Fake an allocation for event memory
5647 alignas (16 ) uint32_t memory = -1 ;
@@ -61,28 +52,17 @@ struct Mock<Event> : public Event {
6152
6253template <>
6354struct Mock <EventPool> : public EventPool {
64- Mock ();
65- ~Mock () override ;
55+ Mock () = default ;
6656
67- MOCK_METHOD0 (destroy, ze_result_t ());
68- MOCK_METHOD0 (getPoolSize, size_t ());
69- MOCK_METHOD0 (getPoolUsedCount, uint32_t ());
70- MOCK_METHOD1 (getIpcHandle, ze_result_t (ze_ipc_event_pool_handle_t *pIpcHandle));
71- MOCK_METHOD0 (closeIpcHandle, ze_result_t ());
72- MOCK_METHOD2 (createEvent, ze_result_t (const ze_event_desc_t *desc, ze_event_handle_t *phEvent));
73- MOCK_METHOD2 (reserveEventFromPool, ze_result_t (int index, ::L0::Event *event));
74- MOCK_METHOD1 (releaseEventToPool, ze_result_t (::L0::Event *event));
75- MOCK_METHOD0 (getDevice, Device *());
76- MOCK_METHOD0 (getEventSize, uint32_t ());
77-
78- std::vector<int > pool;
57+ ADDMETHOD_NOBASE (destroy, ze_result_t , ZE_RESULT_SUCCESS, ());
58+ ADDMETHOD_NOBASE (getIpcHandle, ze_result_t , ZE_RESULT_SUCCESS, (ze_ipc_event_pool_handle_t * pIpcHandle));
59+ ADDMETHOD_NOBASE (closeIpcHandle, ze_result_t , ZE_RESULT_SUCCESS, ());
60+ ADDMETHOD_NOBASE (createEvent, ze_result_t , ZE_RESULT_SUCCESS, (const ze_event_desc_t *desc, ze_event_handle_t *phEvent));
61+ ADDMETHOD_NOBASE (getDevice, Device *, nullptr , ());
62+ ADDMETHOD_NOBASE (getEventSize, uint32_t , 0u , ());
7963
8064 using EventPool::eventPoolAllocations;
8165};
8266
8367} // namespace ult
8468} // namespace L0
85-
86- #if defined(__clang__)
87- #pragma clang diagnostic pop
88- #endif
0 commit comments