76 lines
2.5 KiB
C++
76 lines
2.5 KiB
C++
/*
|
|
* Copyright (C) 2019 Intel Corporation
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*
|
|
*/
|
|
|
|
#include "core/execution_environment/root_device_environment.h"
|
|
#include "runtime/context/context.h"
|
|
#include "runtime/device/device.h"
|
|
#include "runtime/execution_environment/execution_environment.h"
|
|
#include "runtime/platform/platform.h"
|
|
#include "unit_tests/api/cl_api_tests.h"
|
|
#include "unit_tests/mocks/mock_aub_center.h"
|
|
#include "unit_tests/mocks/mock_aub_manager.h"
|
|
|
|
using namespace NEO;
|
|
|
|
namespace ULT {
|
|
|
|
struct clAddCommentToAubTest : api_tests {
|
|
void SetUp() override {
|
|
api_tests::SetUp();
|
|
pDevice = pContext->getDevice(0);
|
|
}
|
|
void TearDown() override {
|
|
api_tests::TearDown();
|
|
}
|
|
|
|
Device *pDevice = nullptr;
|
|
};
|
|
|
|
TEST_F(clAddCommentToAubTest, givenProperCommentNullptrAubCenterWhenAddCommentToAubThenSuccessIsReturned) {
|
|
auto retVal = clAddCommentINTEL(pDevice, "comment");
|
|
EXPECT_EQ(CL_SUCCESS, retVal);
|
|
}
|
|
|
|
TEST_F(clAddCommentToAubTest, givenInvalidDeviceWhenAddCommentToAubThenErrorIsReturned) {
|
|
auto retVal = clAddCommentINTEL(nullptr, "comment");
|
|
EXPECT_EQ(CL_INVALID_DEVICE, retVal);
|
|
}
|
|
|
|
TEST_F(clAddCommentToAubTest, givenNullptrCommentWhenAddCommentToAubThenErrorIsReturned) {
|
|
auto retVal = clAddCommentINTEL(pDevice, nullptr);
|
|
EXPECT_EQ(CL_INVALID_VALUE, retVal);
|
|
}
|
|
|
|
TEST_F(clAddCommentToAubTest, givenAubCenterAndProperCommentButNullptrAubManagerWhenAddCommentToAubThenErrorIsReturned) {
|
|
pPlatform->peekExecutionEnvironment()->rootDeviceEnvironments[testedRootDeviceIndex]->aubCenter.reset(new MockAubCenter());
|
|
|
|
auto retVal = clAddCommentINTEL(pDevice, "comment");
|
|
EXPECT_EQ(CL_INVALID_VALUE, retVal);
|
|
}
|
|
|
|
TEST_F(clAddCommentToAubTest, givenProperCommentAubCenterAndAubManagerWhenAddCommentToAubThenSuccessIsReturned) {
|
|
struct AubManagerCommentMock : public MockAubManager {
|
|
using MockAubManager::MockAubManager;
|
|
void addComment(const char *message) override {
|
|
addCommentCalled = true;
|
|
EXPECT_STREQ("comment", message);
|
|
}
|
|
bool addCommentCalled = false;
|
|
};
|
|
auto mockAubCenter = new MockAubCenter();
|
|
auto mockAubManager = new AubManagerCommentMock;
|
|
mockAubCenter->aubManager.reset(mockAubManager);
|
|
pPlatform->peekExecutionEnvironment()->rootDeviceEnvironments[testedRootDeviceIndex]->aubCenter.reset(mockAubCenter);
|
|
|
|
EXPECT_FALSE(mockAubManager->addCommentCalled);
|
|
|
|
auto retVal = clAddCommentINTEL(pDevice, "comment");
|
|
EXPECT_EQ(CL_SUCCESS, retVal);
|
|
EXPECT_TRUE(mockAubManager->addCommentCalled);
|
|
}
|
|
} // namespace ULT
|