feat: Location method

Docs/SupportedAPIs.md

@@ -41,7 +41,8 @@ Contributions to expand support to unimplemented functionality are always welcom
 | POST   | `/session/:sessionId/execute_async`                 | Not Supported| `Session.execute()` |
 | POST   | `/session/:sessionId/forward`                       | Supported    | `Session.forward()` |
 | POST   | `/session/:sessionId/keys`                          | Supported    | `Session.sendKeys()`|
-| GET    | `/session/:sessionId/location`                      | Supported    | Not implemented     |
+| POST   | `/session/:sessionId/location`                      | Supported    | `Session.setLocation`|
+| GET    | `/session/:sessionId/location`                      | Supported    | `Session.location`|
 | POST   | `/session/:sessionId/moveto`                        | Supported    | `Session.moveTo()`  |
 | GET    | `/session/:sessionId/orientation`                   | Supported    | Not implemented     |
 | POST   | `/session/:sessionId/refresh`                       | Not supported| `Session.refresh()` |

Sources/WebDriver/Location.swift

@@ -0,0 +1,12 @@
+public struct Location: Codable, Equatable {
+    public var latitude: Double
+    public var longitude: Double
+    public var altitude: Double
+
+    public init(latitude: Double, longitude: Double, altitude: Double) {
+        self.latitude = latitude
+        self.longitude = longitude
+        self.altitude = altitude
+    }
+
+}

Sources/WebDriver/Requests.swift

@@ -541,6 +541,30 @@ public enum Requests {
         }
     }
 
+    // https://www.selenium.dev/documentation/legacy/json_wire_protocol/#sessionsessionidlocation
+    public enum SessionLocation {
+        public struct Post: Request {
+            public var session: String
+            public var latitude: Double
+            public var longitude: Double
+            public var altitude: Double
+            public var location: Location
+
+            public var pathComponents: [String] { ["session", session, "location"] }
+            public var method: HTTPMethod { .post }
+            public var body: Location { .init(latitude: latitude, longitude: longitude, altitude: altitude) }
+        }
+
+        public struct Get: Request {
+            public var session: String
+
+            public var pathComponents: [String] { ["session", session, "location"] }
+            public var method: HTTPMethod {.get}
+
+            public typealias Response = ResponseWithValue<Location>
+        }
+    }
+
     // https://www.selenium.dev/documentation/legacy/json_wire_protocol/#sessionsessionidsource
     public struct SessionSource: Request {
         public var session: String 
@@ -554,6 +578,7 @@ public enum Requests {
             public var source: String
         }
 
+
     }
 
     // https://www.selenium.dev/documentation/legacy/json_wire_protocol/#status

Sources/WebDriver/Session.swift

@@ -66,6 +66,13 @@ public class Session {
             }
         }
     }
+
+    public var location: Location {
+        get throws {
+            let response = try webDriver.send(Requests.SessionLocation.Get(session: id))
+            return response.value
+        }
+    }
 
     /// Sets a a timeout value on this session.
     public func setTimeout(type: String, duration: TimeInterval) throws {
@@ -321,6 +328,11 @@ public class Session {
         try webDriver.send(Requests.SessionWindowSize.Post(session: id, windowHandle: handle, width: width, height: height))
     }
 
+    /// Set the current geolocation 
+    public func setLocation(latitude: Double, longitude: Double, altitude: Double, location: Location) throws {
+        try webDriver.send(Requests.SessionLocation.Post(session: id, latitude: latitude, longitude: longitude, altitude: altitude, location: location))
+    }
+
     /// - Returns: The current page source.
     public func source() throws -> String {
         let response = try webDriver.send(Requests.SessionSource(session: id))

Tests/UnitTests/APIToRequestMappingTests.swift

@@ -214,6 +214,20 @@ class APIToRequestMappingTests: XCTestCase {
         XCTAssert(try session.size(window: "myWindow") == (width: 500, height: 500))
     }
 
+    func testLocation() throws {
+        let mockWebDriver: MockWebDriver = MockWebDriver()
+        let session = Session(webDriver: mockWebDriver, existingId: "mySession")
+        let location = Location(latitude: 5, longitude: 20, altitude: 2003)
+
+        mockWebDriver.expect(path: "session/mySession/location", method: .post)
+        try session.setLocation(latitude: 5, longitude: 20, altitude: 2003, location: location)
+
+        mockWebDriver.expect(path: "session/mySession/location", method: .get, type: Requests.SessionLocation.Get.self) {
+            ResponseWithValue(.init(latitude: 5, longitude: 20, altitude: 2003))
+        }
+        XCTAssert(try session.location == location)
+    }
+
     func testSessionSource() throws {
         let mockWebDriver: MockWebDriver = MockWebDriver()
         let session = Session(webDriver: mockWebDriver, existingId: "mySession")
@@ -222,5 +236,6 @@ class APIToRequestMappingTests: XCTestCase {
             ResponseWithValue(.init(source: "currentSource"))
         }
         XCTAssert(try session.source() == "currentSource")
+
     }
 }