tests

Author: grdsdev

Tests/AuthTests/SessionManagerTests.swift

@@ -5,7 +5,6 @@
 //  Created by Guilherme Souza on 23/10/23.
 //
 
-@testable import Auth
 import ConcurrencyExtras
 import CustomDump
 import Helpers
@@ -14,6 +13,8 @@ import TestHelpers
 import XCTest
 import XCTestDynamicOverlay
 
+@testable import Auth
+
 final class SessionManagerTests: XCTestCase {
   var http: HTTPClientMock!
 
@@ -42,6 +43,12 @@ final class SessionManagerTests: XCTestCase {
     )
   }
 
+  override func invokeTest() {
+    withMainSerialExecutor {
+      super.invokeTest()
+    }
+  }
+
   func testSession_shouldFailWithSessionNotFound() async {
     do {
       _ = try await sut.session()
@@ -57,57 +64,53 @@ final class SessionManagerTests: XCTestCase {
   }
 
   func testSession_shouldReturnValidSession() async throws {
-    try await withMainSerialExecutor {
-      let session = Session.validSession
-      Dependencies[clientID].sessionStorage.store(session)
+    let session = Session.validSession
+    Dependencies[clientID].sessionStorage.store(session)
 
-      let returnedSession = try await sut.session()
-      expectNoDifference(returnedSession, session)
-    }
+    let returnedSession = try await sut.session()
+    expectNoDifference(returnedSession, session)
   }
 
   func testSession_shouldRefreshSession_whenCurrentSessionExpired() async throws {
-    try await withMainSerialExecutor {
-      let currentSession = Session.expiredSession
-      Dependencies[clientID].sessionStorage.store(currentSession)
-
-      let validSession = Session.validSession
-
-      let refreshSessionCallCount = LockIsolated(0)
-
-      let (refreshSessionStream, refreshSessionContinuation) = AsyncStream<Session>.makeStream()
-
-      await http.when(
-        { $0.url.path.contains("/token") },
-        return: { _ in
-          refreshSessionCallCount.withValue { $0 += 1 }
-          let session = await refreshSessionStream.first(where: { _ in true })!
-          return .stub(session)
-        }
-      )
-
-      // Fire N tasks and call sut.session()
-      let tasks = (0 ..< 10).map { _ in
-        Task { [weak self] in
-          try await self?.sut.session()
-        }
-      }
+    let currentSession = Session.expiredSession
+    Dependencies[clientID].sessionStorage.store(currentSession)
 
-      await Task.yield()
+    let validSession = Session.validSession
 
-      refreshSessionContinuation.yield(validSession)
-      refreshSessionContinuation.finish()
+    let refreshSessionCallCount = LockIsolated(0)
 
-      // Await for all tasks to complete.
-      var result: [Result<Session?, Error>] = []
-      for task in tasks {
-        let value = await task.result
-        result.append(value)
+    let (refreshSessionStream, refreshSessionContinuation) = AsyncStream<Session>.makeStream()
+
+    await http.when(
+      { $0.url.path.contains("/token") },
+      return: { _ in
+        refreshSessionCallCount.withValue { $0 += 1 }
+        let session = await refreshSessionStream.first(where: { _ in true })!
+        return .stub(session)
+      }
+    )
+
+    // Fire N tasks and call sut.session()
+    let tasks = (0..<10).map { _ in
+      Task { [weak self] in
+        try await self?.sut.session()
       }
+    }
+
+    await Task.yield()
 
-      // Verify that refresher and storage was called only once.
-      XCTAssertEqual(refreshSessionCallCount.value, 1)
-      XCTAssertEqual(try result.map { try $0.get() }, (0 ..< 10).map { _ in validSession })
+    refreshSessionContinuation.yield(validSession)
+    refreshSessionContinuation.finish()
+
+    // Await for all tasks to complete.
+    var result: [Result<Session?, Error>] = []
+    for task in tasks {
+      let value = await task.result
+      result.append(value)
     }
+
+    // Verify that refresher and storage was called only once.
+    XCTAssertEqual(refreshSessionCallCount.value, 1)
+    XCTAssertEqual(try result.map { try $0.get() }, (0..<10).map { _ in validSession })
   }
 }