compute-runtime/unit_tests/command_queue/enqueue_map_buffer_fixture.h

56 lines
1.4 KiB
C++

/*
* Copyright (C) 2017-2019 Intel Corporation
*
* SPDX-License-Identifier: MIT
*
*/
#pragma once
#include "unit_tests/command_queue/command_enqueue_fixture.h"
#include "unit_tests/command_queue/enqueue_fixture.h"
#include "unit_tests/fixtures/buffer_fixture.h"
#include "unit_tests/mocks/mock_context.h"
#include "gtest/gtest.h"
namespace NEO {
struct EnqueueMapBufferTypeTest : public CommandEnqueueFixture,
public ::testing::Test {
EnqueueMapBufferTypeTest(void)
: srcBuffer(nullptr) {
}
void SetUp() override {
CommandEnqueueFixture::SetUp();
BufferDefaults::context = new MockContext;
srcBuffer = BufferHelper<>::create();
}
void TearDown() override {
delete srcBuffer;
delete BufferDefaults::context;
CommandEnqueueFixture::TearDown();
}
protected:
template <typename FamilyType>
void enqueueMapBuffer(cl_bool blocking = CL_TRUE) {
cl_int retVal;
EnqueueMapBufferHelper<>::Traits::errcodeRet = &retVal;
auto mappedPointer = EnqueueMapBufferHelper<>::enqueueMapBuffer(
pCmdQ,
srcBuffer,
blocking);
EXPECT_EQ(CL_SUCCESS, *EnqueueMapBufferHelper<>::Traits::errcodeRet);
EXPECT_NE(nullptr, mappedPointer);
EnqueueMapBufferHelper<>::Traits::errcodeRet = nullptr;
parseCommands<FamilyType>(*pCmdQ);
}
Buffer *srcBuffer;
};
} // namespace NEO