From ed88bc69f6d813615ef9bc8b276df9b2edf85bc3 Mon Sep 17 00:00:00 2001
From: Michal Mrozek <michal.mrozek@intel.com>
Date: Thu, 7 Nov 2019 09:27:44 +0100
Subject: [PATCH] Enhance kernel testing.

Change-Id: I7e9a3c8ba64df82e878a3514a8ed87604cd6f839
Signed-off-by: Michal Mrozek <michal.mrozek@intel.com>
---
 unit_tests/kernel/kernel_tests.cpp | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/unit_tests/kernel/kernel_tests.cpp b/unit_tests/kernel/kernel_tests.cpp
index d891416b69..e974b77194 100644
--- a/unit_tests/kernel/kernel_tests.cpp
+++ b/unit_tests/kernel/kernel_tests.cpp
@@ -2693,6 +2693,19 @@ TEST(KernelTest, givenDebugVariableSetWhenKernelHasStatefulBufferAccessThenMarkK
     }
 }
 
+TEST(KernelTest, givenKernelWithPairArgumentWhenItIsInitializedThenPatchImmediateIsUsedAsArgHandler) {
+    HardwareInfo localHwInfo = *platformDevices[0];
+
+    std::unique_ptr<MockDevice> device(MockDevice::createWithNewExecutionEnvironment<MockDevice>(&localHwInfo));
+    auto context = clUniquePtr(new MockContext(device.get()));
+    MockKernelWithInternals kernel(*device, context.get());
+    kernel.kernelInfo.kernelArgInfo.resize(1);
+    kernel.kernelInfo.kernelArgInfo.at(0).typeStr = "pair<char*, int>";
+
+    kernel.mockKernel->initialize();
+    EXPECT_EQ(&Kernel::setArgImmediate, kernel.mockKernel->kernelArgHandlers[0]);
+}
+
 TEST(KernelTest, whenNullAllocationThenAssignNullPointerToCacheFlushVector) {
     auto device = std::unique_ptr<MockDevice>(MockDevice::createWithNewExecutionEnvironment<MockDevice>(platformDevices[0]));
     MockKernelWithInternals kernel(*device);