From 826b1e25045d982700d7f681dc4e884c878fcb44 Mon Sep 17 00:00:00 2001
From: Andrii Vysotskyi <Andril.Vysotskyi@checkout.com>
Date: Thu, 10 Oct 2024 16:47:14 +0200
Subject: [PATCH] Test throttled session

---
 ...bAuthenticationSessionDecoratorTests.swift | 58 +++++++++++++++++++
 1 file changed, 58 insertions(+)
 create mode 100644 Tests/ProcessOutTests/Sources/Unit/Sessions/WebAuthentication/ThrottledWebAuthenticationSessionDecoratorTests.swift

diff --git a/Tests/ProcessOutTests/Sources/Unit/Sessions/WebAuthentication/ThrottledWebAuthenticationSessionDecoratorTests.swift b/Tests/ProcessOutTests/Sources/Unit/Sessions/WebAuthentication/ThrottledWebAuthenticationSessionDecoratorTests.swift
new file mode 100644
index 00000000..4d6f58ec
--- /dev/null
+++ b/Tests/ProcessOutTests/Sources/Unit/Sessions/WebAuthentication/ThrottledWebAuthenticationSessionDecoratorTests.swift
@@ -0,0 +1,58 @@
+//
+//  ThrottledWebAuthenticationSessionDecoratorTests.swift
+//  ProcessOut
+//
+//  Created by Andrii Vysotskyi on 10.10.2024.
+//
+
+@testable import ProcessOut
+import XCTest
+
+final class ThrottledWebAuthenticationSessionDecoratorTests: XCTestCase {
+
+    // MARK: - Wait
+
+    func test_authenticate_allowsOneAuthenticationAtTime() async throws {
+        // Given
+        let mockSession = MockWebAuthenticationSession()
+        mockSession.authenticateFromClosure = { _, _, _ in
+            try await Task.sleep(for: .seconds(5))
+            return URL(string: "response.com")!
+        }
+        let sut = ThrottledWebAuthenticationSessionDecorator(session: mockSession)
+
+        // When
+        Task {
+            _ = try await sut.authenticate(using: URL(string: "request1.com")!)
+        }
+        Task {
+            _ = try await sut.authenticate(using: URL(string: "request2.com")!)
+        }
+        try await Task.sleep(for: .seconds(1))
+
+        // Then
+        XCTAssertEqual(mockSession.authenticateCallsCount, 1)
+    }
+
+    func test_authenticate_throttlesAuthentications() async throws {
+        var lastAuthenticationStartTime: DispatchTime?
+
+        // Given
+        let mockSession = MockWebAuthenticationSession()
+        mockSession.authenticateFromClosure = { _, _, _ in
+            // Then
+            if let lastAuthenticationStartTime {
+                let delay = Int(DispatchTime.now().uptimeNanoseconds - lastAuthenticationStartTime.uptimeNanoseconds)
+                let expectedDelay = 1_000_000_000, tolerance = 500_000_000 // 1 and 0.5 seconds respectfully
+                XCTAssertTrue(abs(delay - expectedDelay) <= tolerance)
+            }
+            lastAuthenticationStartTime = DispatchTime.now()
+            return URL(string: "response.com")!
+        }
+        let sut = ThrottledWebAuthenticationSessionDecorator(session: mockSession)
+
+        // When
+        _ = try await sut.authenticate(using: URL(string: "request1.com")!)
+        _ = try await sut.authenticate(using: URL(string: "request2.com")!)
+    }
+}