From 3f9523969984bd8a53b94a16a46b7c8005a05e82 Mon Sep 17 00:00:00 2001
From: kean <grebenyuk.alexander@gmail.com>
Date: Sat, 15 Jul 2023 09:21:20 -0400
Subject: [PATCH] Add DataCache write/flush performance tests

---
 Sources/Nuke/Caching/DataCache.swift          |  4 +-
 .../DataCachePeformanceTests.swift            | 39 +++++++++++++++++++
 2 files changed, 40 insertions(+), 3 deletions(-)

diff --git a/Sources/Nuke/Caching/DataCache.swift b/Sources/Nuke/Caching/DataCache.swift
index 1fbf5b87c..8b39014a5 100644
--- a/Sources/Nuke/Caching/DataCache.swift
+++ b/Sources/Nuke/Caching/DataCache.swift
@@ -230,9 +230,7 @@ public final class DataCache: DataCaching, @unchecked Sendable {
 
     /// Returns `url` for the given cache key.
     public func url(for key: String) -> URL? {
-        guard let filename = self.filename(for: key) else {
-            return nil
-        }
+        guard let filename = self.filename(for: key) else { return nil }
         return self.path.appendingPathComponent(filename, isDirectory: false)
     }
 
diff --git a/Tests/NukePerformanceTests/DataCachePeformanceTests.swift b/Tests/NukePerformanceTests/DataCachePeformanceTests.swift
index d48357175..71429f5d3 100644
--- a/Tests/NukePerformanceTests/DataCachePeformanceTests.swift
+++ b/Tests/NukePerformanceTests/DataCachePeformanceTests.swift
@@ -22,6 +22,43 @@ class DataCachePeformanceTests: XCTestCase {
         try? FileManager.default.removeItem(at: cache.path)
     }
 
+    // MARK: - Write
+
+    func testWriteWithFlush() {
+        let data = Array(0..<count).map { _ in generateRandomData() }
+
+        measure {
+            for index in data.indices {
+                cache["\(index)"] = data[index]
+            }
+            cache.flush()
+        }
+    }
+
+    func testWriteWithFlushIndividual() {
+        let data = Array(0..<count).map { _ in generateRandomData() }
+
+        measure {
+            for index in data.indices {
+                let key = "\(index)"
+                cache[key] = data[index]
+                cache.flush(for: key)
+            }
+        }
+    }
+
+    func testWriteWithoutFlush() {
+        let data = Array(0..<count).map { _ in generateRandomData() }
+
+        measure {
+            for index in data.indices {
+                cache["\(index)"] = data[index]
+            }
+        }
+    }
+
+    // MARK: - Read
+
     func testReadFlushedPerformance() {
         populate()
 
@@ -47,6 +84,8 @@ class DataCachePeformanceTests: XCTestCase {
         }
     }
 
+    // MARK: - Helpers
+
     func populate() {
         for idx in 0..<count {
             cache["\(idx)"] = generateRandomData()