Fix tests

forem · Oct 19, 2023 · c8582f8 · c8582f8
1 parent c5fa28a
commit c8582f8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/DEV-SimpleTests/ViewControllerTests.swift b/DEV-SimpleTests/ViewControllerTests.swift
@@ -24,8 +24,8 @@ class ViewControllerTests: XCTestCase {
         _ = viewController.view
 
         let promise = expectation(description: "User Agent expectation")
-        DispatchQueue.main.asyncAfter(deadline: .now() + 3) {
-            let usesForemWebViewUserAgent = self.viewController.webView.customUserAgent?.contains("ForemWebView")
+        viewController.webView.evaluateJavaScript("navigator.userAgent") { agent, _ in
+            let usesForemWebViewUserAgent = (agent as? String)?.contains("ForemWebView")
             XCTAssertTrue(usesForemWebViewUserAgent ?? false)
             promise.fulfill()
         }