From 52bb8f72952f6c1d266dffb7736374776363e953 Mon Sep 17 00:00:00 2001
From: kean <grebenyuk.alexander@gmail.com>
Date: Sun, 3 Mar 2024 13:30:47 -0500
Subject: [PATCH] Update testDataCacheMultipleThreadAccess

---
 Tests/NukeThreadSafetyTests/ThreadSafetyTests.swift | 13 ++++++++++---
 1 file changed, 10 insertions(+), 3 deletions(-)

diff --git a/Tests/NukeThreadSafetyTests/ThreadSafetyTests.swift b/Tests/NukeThreadSafetyTests/ThreadSafetyTests.swift
index 0f1a30ad4..f6e4af6ef 100644
--- a/Tests/NukeThreadSafetyTests/ThreadSafetyTests.swift
+++ b/Tests/NukeThreadSafetyTests/ThreadSafetyTests.swift
@@ -165,7 +165,9 @@ class ThreadSafetyTests: XCTestCase {
         queue.waitUntilAllOperationsAreFinished()
     }
 
-    func testDataCacheMultipleThreadAccess() {
+    func testDataCacheMultipleThreadAccess() throws {
+        let cache = try DataCache(name: UUID().uuidString)
+
         let aURL = URL(string: "https://example.com/image-01-small.jpeg")!
         let imageData = Test.data(name: "fixture", extension: "jpeg")
 
@@ -173,8 +175,11 @@ class ThreadSafetyTests: XCTestCase {
         expectSuccessFromCache.expectedFulfillmentCount = 1
         expectSuccessFromCache.assertForOverFulfill = true
 
-        let pipeline = ImagePipeline(configuration: .withDataCache)
-        pipeline.cache.storeCachedData(imageData, for: ImageRequest.init(url: aURL))
+        let pipeline = ImagePipeline {
+            $0.dataCache = cache
+            $0.dataLoader = MockDataLoader()
+        }
+        pipeline.cache.storeCachedData(imageData, for: ImageRequest(url: aURL))
         pipeline.loadImage(with: aURL) { result in
             switch result {
             case .success(let response):
@@ -189,6 +194,8 @@ class ThreadSafetyTests: XCTestCase {
         }
 
         wait(for: [expectSuccessFromCache], timeout: 2)
+
+        try? FileManager.default.removeItem(at: cache.path)
     }
 }