1414#include " unit_tests/fixtures/buffer_fixture.h"
1515#include " unit_tests/fixtures/device_fixture.h"
1616#include " unit_tests/mocks/mock_buffer.h"
17+ #include " unit_tests/mocks/mock_command_queue.h"
1718#include " unit_tests/mocks/mock_context.h"
1819#include " unit_tests/mocks/mock_kernel.h"
1920
@@ -260,7 +261,7 @@ TEST_F(EnqueueMapBufferTest, GivenValidArgsWhenMappingBufferThenSuccessIsReturne
260261 EXPECT_EQ (CL_SUCCESS, retVal);
261262}
262263
263- TEST_F (EnqueueMapBufferTest, givenNonBlockingReadOnlyMapBufferOnZeroCopyBufferWhenItIsCalledThenSynchronizationIsNotMadeUntilWaitForEvents) {
264+ HWTEST_F (EnqueueMapBufferTest, givenNonBlockingReadOnlyMapBufferOnZeroCopyBufferWhenItIsCalledThenSynchronizationIsNotMadeUntilWaitForEvents) {
264265 DebugManagerStateRestore dbgRestore;
265266 DebugManager.flags .EnableAsyncEventsHandler .set (false );
266267 cl_event mapEventReturned = nullptr ;
@@ -285,18 +286,20 @@ TEST_F(EnqueueMapBufferTest, givenNonBlockingReadOnlyMapBufferOnZeroCopyBufferWh
285286 EXPECT_EQ (CL_SUCCESS, retVal);
286287 EXPECT_NE (nullptr , buffer);
287288
288- auto &commandStreamReceiver = pCmdQ->getGpgpuCommandStreamReceiver ();
289+ MockCommandQueueHw<FamilyType> mockCmdQueue (context, pClDevice, nullptr );
290+
291+ auto &commandStreamReceiver = mockCmdQueue.getGpgpuCommandStreamReceiver ();
289292 uint32_t taskCount = commandStreamReceiver.peekTaskCount ();
290293 EXPECT_EQ (0u , taskCount);
291294
292295 // enqueue something that can be finished...
293- retVal = clEnqueueNDRangeKernel (pCmdQ , kernel, 1 , 0 , &GWS, nullptr , 0 , nullptr , nullptr );
296+ retVal = clEnqueueNDRangeKernel (&mockCmdQueue , kernel, 1 , 0 , &GWS, nullptr , 0 , nullptr , nullptr );
294297 EXPECT_EQ (retVal, CL_SUCCESS);
295298
296299 EXPECT_EQ (1u , commandStreamReceiver.peekTaskCount ());
297300
298301 auto ptrResult = clEnqueueMapBuffer (
299- pCmdQ ,
302+ &mockCmdQueue ,
300303 buffer,
301304 CL_FALSE,
302305 CL_MAP_READ,
@@ -334,14 +337,14 @@ TEST_F(EnqueueMapBufferTest, givenNonBlockingReadOnlyMapBufferOnZeroCopyBufferWh
334337
335338 // wait for event do not sent flushTask
336339 EXPECT_EQ (1u , commandStreamReceiver.peekTaskCount ());
337- EXPECT_EQ (1u , pCmdQ-> latestTaskCountWaited );
340+ EXPECT_EQ (1u , mockCmdQueue. latestTaskCountWaited );
338341
339342 EXPECT_TRUE (neoEvent->updateStatusAndCheckCompletion ());
340343
341344 EXPECT_EQ (1u , callbackCalled);
342345
343346 retVal = clEnqueueUnmapMemObject (
344- pCmdQ ,
347+ &mockCmdQueue ,
345348 buffer,
346349 ptrResult,
347350 0 ,
@@ -514,7 +517,7 @@ TEST_F(EnqueueMapBufferTest, givenNonBlockingMapBufferAfterL3IsAlreadyFlushedThe
514517 clReleaseEvent (eventReturned);
515518}
516519
517- TEST_F (EnqueueMapBufferTest, GivenBufferThatIsNotZeroCopyWhenNonBlockingMapIsCalledThenFinishIsCalledAndDataTransferred) {
520+ HWTEST_F (EnqueueMapBufferTest, GivenBufferThatIsNotZeroCopyWhenNonBlockingMapIsCalledThenFinishIsCalledAndDataTransferred) {
518521 const auto bufferSize = 100 ;
519522 auto localSize = bufferSize;
520523 char misaligned[bufferSize] = {1 };
@@ -540,16 +543,18 @@ TEST_F(EnqueueMapBufferTest, GivenBufferThatIsNotZeroCopyWhenNonBlockingMapIsCal
540543 auto pBuffer = castToObject<Buffer>(buffer);
541544 ASSERT_FALSE (pBuffer->isMemObjZeroCopy ());
542545
546+ MockCommandQueueHw<FamilyType> mockCmdQueue (context, pClDevice, nullptr );
547+
543548 // enqueue something that can be finished
544- retVal = clEnqueueNDRangeKernel (pCmdQ , kernel, 1 , 0 , &GWS, nullptr , 0 , nullptr , nullptr );
549+ retVal = clEnqueueNDRangeKernel (&mockCmdQueue , kernel, 1 , 0 , &GWS, nullptr , 0 , nullptr , nullptr );
545550 EXPECT_EQ (retVal, CL_SUCCESS);
546551
547- auto &commandStreamReceiver = pCmdQ-> getGpgpuCommandStreamReceiver ();
552+ auto &commandStreamReceiver = mockCmdQueue. getGpgpuCommandStreamReceiver ();
548553 uint32_t taskCount = commandStreamReceiver.peekTaskCount ();
549554 EXPECT_EQ (1u , taskCount);
550555
551556 auto ptrResult = clEnqueueMapBuffer (
552- pCmdQ ,
557+ &mockCmdQueue ,
553558 buffer,
554559 CL_FALSE,
555560 CL_MAP_READ,
@@ -566,7 +571,7 @@ TEST_F(EnqueueMapBufferTest, GivenBufferThatIsNotZeroCopyWhenNonBlockingMapIsCal
566571 commandStreamReceiver.peekTaskCount ();
567572
568573 EXPECT_EQ (1u , commandStreamReceiver.peekLatestSentTaskCount ());
569- EXPECT_EQ (1u , pCmdQ-> latestTaskCountWaited );
574+ EXPECT_EQ (1u , mockCmdQueue. latestTaskCountWaited );
570575
571576 retVal = clReleaseMemObject (buffer);
572577 EXPECT_EQ (CL_SUCCESS, retVal);
0 commit comments