Refactor FXIOS-13481 [Swift 6 Migration] Turn on Concurrency for SummarizerKit Tests in BrowserKit (#31033)

* Turn on concurrency for summarizerTests

* Fix summarizeControllerTests

* Fix SummarizeViewModelTests

* Make MockSummarizerService unchecked Sendable

* Make MockLanguageModelSession unchecked Sendable

* Add SummarizerServiceTests

* Make TrackMemoryLeak extension MainActor isolated

* use setup and teardown async throws

* remove unneeded change and version bump

* Update SummarizerServiceTests

Author: dataports

BrowserKit/Package.swift

@@ -220,7 +220,11 @@ let package = Package(
             ]
         ),
         .testTarget(name: "SummarizeKitTests",
-                    dependencies: ["SummarizeKit"]),
+                    dependencies: ["SummarizeKit"],
+                    swiftSettings: [
+                        .unsafeFlags(["-enable-testing"]),
+                        .enableExperimentalFeature("StrictConcurrency")
+                    ]),
         .target(
             name: "JWTKit",
             dependencies: ["Common", "Shared"],

BrowserKit/Tests/SummarizeKitTests/LiteLLMClientTests.swift

@@ -6,6 +6,7 @@ import XCTest
 import Common
 @testable import SummarizeKit
 
+@MainActor
 final class LiteLLMClientTests: XCTestCase {
     private static let mockAPIEndpoint = "https://test-api-url.com"
     private static let mockAPIKey =  "test-api-key"

BrowserKit/Tests/SummarizeKitTests/Mocks/MockLanguageModelSession.swift

@@ -25,7 +25,7 @@ struct MockLanguageModelResponseSnapshot: LanguageModelResponseSnapshotProtocol
 /// Mock implementation of a language model session for testing the session and responses.
 /// This allows injecting controlled outputs or errors without calling the real inference backend.
 @available(iOS 26, *)
-final class MockLanguageModelSession: LanguageModelSessionProtocol {
+final class MockLanguageModelSession: LanguageModelSessionProtocol, @unchecked Sendable {
     var respondWith: [String] = [""]
     var respondWithError: Error?
 

BrowserKit/Tests/SummarizeKitTests/Mocks/MockWebView.swift

@@ -4,7 +4,7 @@
 
 import WebKit
 
-class MockWebView: WKWebView {
+final class MockWebView: WKWebView, @unchecked Sendable {
     var overridenURL: URL
 
     init(_ url: URL) {

BrowserKit/Tests/SummarizeKitTests/Mocks/XCTest+TrackMemoryLeak.swift

@@ -6,6 +6,7 @@ import Foundation
 import XCTest
 
 public extension XCTestCase {
+    @MainActor
     func trackForMemoryLeaks(_ object: AnyObject?, file: StaticString = #filePath, line: UInt = #line) {
         addTeardownBlock { [weak object] in
             XCTAssertNil(object, "Memory leak detected in \(file):\(line)")

BrowserKit/Tests/SummarizeKitTests/SSEDataParserTests.swift

@@ -5,6 +5,7 @@
 import XCTest
 @testable import SummarizeKit
 
+@MainActor
 final class SSEDataParserTests: XCTestCase {
     private struct EventExampleType: Decodable, Equatable {
         let id: Int

BrowserKit/Tests/SummarizeKitTests/SummarizationCheckerTests.swift

@@ -5,6 +5,7 @@
 @testable import SummarizeKit
 import XCTest
 
+@MainActor
 final class SummarizationCheckerTests: XCTestCase {
     func jsStubBuilder(
         canSummarize: Bool,

BrowserKit/Tests/SummarizeKitTests/SummarizeControllerTests.swift

@@ -33,7 +33,7 @@ final class MockSummarizeNavigationHandler: SummarizeNavigationHandler {
 }
 
 @MainActor
-final class SummarizeControllerTests: XCTestCase {
+final class SummarizeControllerTests: XCTestCase, @unchecked Sendable {
     private var summarizer: MockSummarizer!
     private var navigationHandler: MockSummarizeNavigationHandler!
     private var webView: MockWebView!
@@ -87,8 +87,8 @@ final class SummarizeControllerTests: XCTestCase {
     )
     private let maxWords = 5000
 
-    override func setUp() {
-        super.setUp()
+    override func setUp() async throws {
+        try await super.setUp()
         UIView.setAnimationsEnabled(false)
         summarizer = MockSummarizer(shouldRespond: ["Response"], shouldThrowError: nil)
         navigationHandler = MockSummarizeNavigationHandler()
@@ -97,14 +97,14 @@ final class SummarizeControllerTests: XCTestCase {
         AppContainer.shared.register(service: DefaultThemeManager(sharedContainerIdentifier: "") as ThemeManager)
     }
 
-    override func tearDown() {
+    override func tearDown() async throws {
         UIView.setAnimationsEnabled(true)
         summarizer = nil
         navigationHandler = nil
         webView = nil
         viewModel = nil
         AppContainer.shared.reset()
-        super.tearDown()
+        try await super.tearDown()
     }
 
     func test_viewDidLoad_startSummarizing() async {

BrowserKit/Tests/SummarizeKitTests/SummarizeViewModelTests.swift

@@ -3,6 +3,7 @@
 // file, You can obtain one at http://mozilla.org/MPL/2.0/
 
 import XCTest
+import Common
 @testable import SummarizeKit
 
 extension Result {
@@ -25,28 +26,28 @@ class MockDateProvider: DateProvider {
 }
 
 @MainActor
-final class SummarizeViewModelTests: XCTestCase {
+final class SummarizeViewModelTests: XCTestCase, @unchecked Sendable {
     private var tosAcceptor: MockSummarizeToSAcceptor!
     private var summarizerService: MockSummarizerService!
     private var webView: MockWebView!
     private var dateProvider: MockDateProvider!
     private let maxWords = 5000
     private let url = URL(string: "https://example.com")!
 
-    override func setUp() {
-        super.setUp()
+    override func setUp() async throws {
+        try await super.setUp()
         webView = MockWebView(url)
         dateProvider = MockDateProvider()
         summarizerService = MockSummarizerService()
         tosAcceptor = MockSummarizeToSAcceptor()
     }
 
-    override func tearDown() {
+    override func tearDown() async throws {
         tosAcceptor = nil
         dateProvider = nil
         summarizerService = nil
         webView = nil
-        super.tearDown()
+        try await super.tearDown()
     }
 
     func test_summarize_whenTosNotAccepted_whenTriggerIsShake_showsToSScreen() {

BrowserKit/Tests/SummarizeKitTests/SummarizerServiceTests.swift

@@ -4,12 +4,24 @@
 
 @testable import SummarizeKit
 import XCTest
+import Common
 
+@MainActor
 final class SummarizerServiceTests: XCTestCase {
     static let mockResponse = ["Summarized", "content"]
     static let maxWords = 100
     static let maxWordCount = 50
-    var mockWebView = MockWebView(URL(string: "https://foo.com")!)
+    private var mockWebView: MockWebView!
+
+    override func setUp() async throws {
+        try await super.setUp()
+        mockWebView = MockWebView(URL(string: "https://foo.com")!)
+    }
+
+    override func tearDown() async throws {
+        mockWebView = nil
+        try await super.tearDown()
+    }
 
     func testSummarizerServiceReturnsSummary() async throws {
         let subject = createSubject()