Test closing GEM worker is blocking during cleanup

Added test checking if closing GEM worker during
DRM memory manager clean up is blocking.

Related-To: NEO-6213

Signed-off-by: Krystian Chmielewski <[email protected]>

Author: KrystianChmielewski

opencl/test/unit_test/os_interface/linux/drm_memory_manager_tests.cpp

@@ -3188,6 +3188,15 @@ TEST_F(DrmMemoryManagerBasic, givenDisabledAsyncDeleterFlagWhenMemoryManagerIsCr
     memoryManager.commonCleanup();
 }
 
+TEST_F(DrmMemoryManagerBasic, givenWorkerToCloseWhenCommonCleanupIsCalledThenClosingIsBlocking) {
+    MockDrmMemoryManager memoryManager(gemCloseWorkerMode::gemCloseWorkerInactive, false, true, executionEnvironment);
+    memoryManager.gemCloseWorker.reset(new MockDrmGemCloseWorker(memoryManager));
+    auto pWorker = static_cast<MockDrmGemCloseWorker *>(memoryManager.gemCloseWorker.get());
+
+    memoryManager.commonCleanup();
+    EXPECT_TRUE(pWorker->wasBlocking);
+}
+
 TEST_F(DrmMemoryManagerBasic, givenDefaultDrmMemoryManagerWhenItIsQueriedForInternalHeapBaseThenInternalHeapBaseIsReturned) {
     std::unique_ptr<TestedDrmMemoryManager> memoryManager(new (std::nothrow) TestedDrmMemoryManager(false,
                                                                                                     true,

shared/source/os_interface/linux/drm_gem_close_worker.h

@@ -1,5 +1,5 @@
 /*
- * Copyright (C) 2018-2021 Intel Corporation
+ * Copyright (C) 2018-2022 Intel Corporation
  *
  * SPDX-License-Identifier: MIT
  *
@@ -27,13 +27,13 @@ enum class gemCloseWorkerMode {
 class DrmGemCloseWorker {
   public:
     DrmGemCloseWorker(DrmMemoryManager &memoryManager);
-    ~DrmGemCloseWorker();
+    MOCKABLE_VIRTUAL ~DrmGemCloseWorker();
 
     DrmGemCloseWorker(const DrmGemCloseWorker &) = delete;
     DrmGemCloseWorker &operator=(const DrmGemCloseWorker &) = delete;
 
     void push(BufferObject *allocation);
-    void close(bool blocking);
+    MOCKABLE_VIRTUAL void close(bool blocking);
 
     bool isEmpty();
 

shared/test/common/mocks/linux/mock_drm_memory_manager.h

@@ -1,5 +1,5 @@
 /*
- * Copyright (C) 2018-2021 Intel Corporation
+ * Copyright (C) 2018-2022 Intel Corporation
  *
  * SPDX-License-Identifier: MIT
  *
@@ -124,4 +124,20 @@ class TestedDrmMemoryManager : public MemoryManagerCreate<DrmMemoryManager> {
     bool alignedMallocShouldFail = false;
     size_t alignedMallocSizeRequired = 0u;
 };
+
+struct MockDrmGemCloseWorker : DrmGemCloseWorker {
+    using DrmGemCloseWorker::DrmGemCloseWorker;
+
+    void close(bool blocking) override {
+        wasBlocking = blocking;
+        DrmGemCloseWorker::close(blocking);
+    }
+    bool wasBlocking = false;
+};
+
+struct MockDrmMemoryManager : DrmMemoryManager {
+    using DrmMemoryManager::DrmMemoryManager;
+    using DrmMemoryManager::gemCloseWorker;
+};
+
 } // namespace NEO