fix: Failing tests

thebrowsercompany · Nov 29, 2023 · 150162f · 150162f
1 parent 53b0029
commit 150162f
Showing 1 changed file with 12 additions and 3 deletions.
diff --git a/Tests/UnitTests/APIToRequestMappingTests.swift b/Tests/UnitTests/APIToRequestMappingTests.swift
@@ -174,16 +174,25 @@ class APIToRequestMappingTests: XCTestCase {
         let mockWebDriver = MockWebDriver()
         let session = Session(webDriver: mockWebDriver, existingId: "mySession")
         mockWebDriver.expect(path: "session/mySession/execute", method: .post)
+        try session.execute(javascript: "return document.body", args: [], async: false) {
+            XCTAssertEqual($0.javascript, "return document.body")
+            XCTAssertEqual($0.args, [])
+            XCTAssertEqual($0.async, true)
+            return CodableNone()
+        }
         try session.execute(javascript: "return document.body", args: [], async: false)
-        XCTAssert(try session.execute(javascript: "return document.body", args: [], async: false) == (javascript: "return document.body", args: [], async: false))
     }
 
     func testSessionScriptAsync() throws {
         let mockWebDriver = MockWebDriver()
         let session = Session(webDriver: mockWebDriver, existingId: "mySession")
-        mockWebDriver.expect(path: "session/mySession/execute_async", method: .post)
+        mockWebDriver.expect(path: "session/mySession/execute_async", method: .post) {
+            XCTAssertEqual($0.javascript, "return document.body")
+            XCTAssertEqual($0.args, [])
+            XCTAssertEqual($0.async, true)
+            return CodableNone()
+        }
         try session.execute(javascript: "return document.body", args: [], async: true)
-        XCTAssert(try session.execute(javascript: "return document.body", args: [], async: true) == (javascript: "return document.body", args: [], async: true))
     }
 
     func testWindow() throws {