Update testDataCacheMultipleThreadAccess

kean · Mar 3, 2024 · 52bb8f7 · 52bb8f7
1 parent 3b4c97f
commit 52bb8f7
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/Tests/NukeThreadSafetyTests/ThreadSafetyTests.swift b/Tests/NukeThreadSafetyTests/ThreadSafetyTests.swift
@@ -165,16 +165,21 @@ 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")
 
         let expectSuccessFromCache = self.expectation(description: "one successful load, from cache")
         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)
     }
 }