Add memory allocate in device pool tests for DG1 platform

Related-To: NEO-4744

Change-Id: Ia4b557e8aec40033ce05ca313c0a9fe545c31388
Signed-off-by: Slawomir Milczarek <slawomir.milczarek@intel.com>

Author: smilczar

opencl/test/unit_test/os_interface/linux/CMakeLists.txt

@@ -21,7 +21,6 @@ set(IGDRCL_SRCS_tests_os_interface_linux
   ${CMAKE_CURRENT_SOURCE_DIR}${BRANCH_DIR_SUFFIX}/drm_mapper_tests.cpp
   ${CMAKE_CURRENT_SOURCE_DIR}/drm_memory_manager_tests.cpp
   ${CMAKE_CURRENT_SOURCE_DIR}/drm_memory_manager_tests.h
-  ${CMAKE_CURRENT_SOURCE_DIR}${BRANCH_DIR_SUFFIX}/drm_memory_manager_allocate_in_device_pool_tests.cpp
   ${CMAKE_CURRENT_SOURCE_DIR}/drm_neo_create.cpp
   ${CMAKE_CURRENT_SOURCE_DIR}/drm_os_memory_tests.cpp
   ${CMAKE_CURRENT_SOURCE_DIR}/drm_residency_handler_tests.cpp
@@ -41,10 +40,12 @@ set(IGDRCL_SRCS_tests_os_interface_linux
 if(SUPPORT_DG1 AND "${BRANCH_TYPE}" STREQUAL "")
 list(APPEND IGDRCL_SRCS_tests_os_interface_linux
   ${CMAKE_CURRENT_SOURCE_DIR}/drm_memory_info_tests_dg1.cpp
+  ${CMAKE_CURRENT_SOURCE_DIR}/drm_memory_manager_allocate_in_device_pool_tests_dg1.cpp
 )
 else()
 list(APPEND IGDRCL_SRCS_tests_os_interface_linux
   ${CMAKE_CURRENT_SOURCE_DIR}${BRANCH_DIR_SUFFIX}/drm_memory_info_tests.cpp
+  ${CMAKE_CURRENT_SOURCE_DIR}${BRANCH_DIR_SUFFIX}/drm_memory_manager_allocate_in_device_pool_tests.cpp
 )
 endif()
 if(UNIX)

opencl/test/unit_test/os_interface/linux/device_command_stream_fixture.h

@@ -72,6 +72,8 @@ class DrmMockTime : public DrmMockSuccess {
 
 class DrmMockCustom : public Drm {
   public:
+    using Drm::memoryInfo;
+
     struct IoctlResExt {
         int32_t no;
         int32_t res;

opencl/test/unit_test/os_interface/linux/device_command_stream_fixture_dg1.h

@@ -0,0 +1,78 @@
+/*
+ * Copyright (C) 2020 Intel Corporation
+ *
+ * SPDX-License-Identifier: MIT
+ *
+ */
+
+#pragma once
+#include "opencl/test/unit_test/os_interface/linux/device_command_stream_fixture.h"
+
+class DrmMockCustomDg1 : public DrmMockCustom {
+  public:
+    using Drm::memoryInfo;
+
+    class IoctlsDg1 {
+      public:
+        void reset() {
+            gemCreateExt = 0;
+            gemMmapOffset = 0;
+        }
+        std::atomic<int32_t> gemCreateExt;
+        std::atomic<int32_t> gemMmapOffset;
+    };
+
+    IoctlsDg1 ioctlDg1_cnt;
+    IoctlsDg1 ioctlDg1_expected;
+
+    void testIoctlsDg1() {
+#define NEO_IOCTL_EXPECT_EQ(PARAM)                                          \
+    if (this->ioctlDg1_expected.PARAM >= 0) {                               \
+        EXPECT_EQ(this->ioctlDg1_expected.PARAM, this->ioctlDg1_cnt.PARAM); \
+    }
+        NEO_IOCTL_EXPECT_EQ(gemMmapOffset);
+#undef NEO_IOCTL_EXPECT_EQ
+    }
+
+    //DRM_IOCTL_I915_GEM_CREATE_EXT
+    __u64 createExtSize = 0;
+    __u32 createExtHandle = 0;
+    __u64 createExtExtensions = 0;
+
+    //DRM_IOCTL_I915_GEM_MMAP_OFFSET
+    __u32 mmapOffsetHandle = 0;
+    __u32 mmapOffsetPad = 0;
+    __u64 mmapOffsetOffset = 0;
+    __u64 mmapOffsetFlags = 0;
+
+    int ioctlExtra(unsigned long request, void *arg) override {
+        switch (request) {
+        case DRM_IOCTL_I915_GEM_CREATE_EXT: {
+            auto createExtParams = reinterpret_cast<drm_i915_gem_create_ext *>(arg);
+            createExtSize = createExtParams->size;
+            createExtHandle = createExtParams->handle;
+            createExtExtensions = createExtParams->extensions;
+            ioctlDg1_cnt.gemCreateExt++;
+        } break;
+        case DRM_IOCTL_I915_GEM_MMAP_OFFSET: {
+            auto mmapOffsetParams = reinterpret_cast<drm_i915_gem_mmap_offset *>(arg);
+            mmapOffsetHandle = mmapOffsetParams->handle;
+            mmapOffsetPad = mmapOffsetParams->pad;
+            mmapOffsetOffset = mmapOffsetParams->offset;
+            mmapOffsetFlags = mmapOffsetParams->flags;
+            ioctlDg1_cnt.gemMmapOffset++;
+        } break;
+        default: {
+            std::cout << std::hex << DRM_IOCTL_I915_GEM_WAIT << std::endl;
+            std::cout << "unexpected IOCTL: " << std::hex << request << std::endl;
+            UNRECOVERABLE_IF(true);
+        } break;
+        }
+        return 0;
+    }
+
+    DrmMockCustomDg1() : DrmMockCustom() {
+        ioctlDg1_cnt.reset();
+        ioctlDg1_expected.reset();
+    }
+};