setAttribute() Method Implemented

Sources/SwiftWebDriver/API/Request/DevTools/AnyEncodable.swift

@@ -0,0 +1,16 @@
+// AnyEncodable.swift
+// Copyright (c) 2025 GetAutomaApp
+// All source code and related assets are the property of GetAutomaApp.
+// All rights reserved.
+
+public struct AnyEncodable: Encodable {
+    private let encodeFunc: (Encoder) throws -> Void
+
+    init(_ value: some Encodable) {
+        encodeFunc = value.encode
+    }
+
+    public func encode(to encoder: Encoder) throws {
+        try encodeFunc(encoder)
+    }
+}

Sources/SwiftWebDriver/API/Request/DevTools/PostExecuteAsyncRequest.swift

@@ -39,11 +39,9 @@ internal struct PostExecuteAsyncRequest: RequestType {
 
         return .data(data)
     }
-}
 
-internal extension PostExecuteAsyncRequest {
-    struct RequestBody: Codable {
-        public let script: String
-        public let args: [String]
+    struct RequestBody: Encodable {
+        let script: String
+        let args: [AnyEncodable]
     }
 }

Sources/SwiftWebDriver/API/Request/DevTools/PostExecuteSyncRequest.swift

@@ -41,8 +41,8 @@ internal struct PostExecuteSyncRequest: RequestType {
 }
 
 internal extension PostExecuteSyncRequest {
-    struct RequestBody: Codable {
-        public let script: String
-        public let args: [String]
+    struct RequestBody: Encodable {
+        let script: String
+        let args: [AnyEncodable]
     }
 }

Sources/SwiftWebDriver/WebDriver.swift

@@ -9,7 +9,7 @@
 import Foundation
 import NIOCore
 
 public class WebDriver<T: Driver> {
     private let driver: T
 
     /// init webDriver
@@ -46,7 +46,7 @@
     }
 
     @available(macOS 10.15, iOS 13.0, watchOS 6.0, tvOS 13.0, *)
     public func getNavigation() async throws -> GetNavigationResponse {
         try await driver.getNavigation()
     }
 
@@ -60,7 +60,7 @@
     }
 
     @discardableResult
     public func navigateTo(url: URL) async throws -> PostNavigationResponse {
         try await navigateTo(urlString: url.absoluteString)
     }
 
@@ -82,12 +82,12 @@
 
     @discardableResult
     @available(macOS 10.15, iOS 13.0, watchOS 6.0, tvOS 13.0, *)
     public func navigationRefresh() async throws -> PostNavigationRefreshResponse {
         try await driver.postNavigationRefresh()
     }
 
     @available(macOS 10.15, iOS 13.0, watchOS 6.0, tvOS 13.0, *)
     public func navigationTitle() async throws -> GetNavigationTitleResponse {
         try await driver.getNavigationTitle()
     }
 
@@ -120,7 +120,7 @@
     @discardableResult
     public func execute(
         _ script: String,
-        args: [String] = [],
+        args: [AnyEncodable] = [],
         type: DevToolTypes.JavascriptExecutionTypes = .sync
     ) async throws -> PostExecuteResponse {
         try await driver.execute(script, args: args, type: type)
@@ -131,4 +131,9 @@
     public func getActiveElement() async throws -> Element {
         try await driver.getActiveElement()
     }
+
+    @available(macOS 10.15, iOS 13.0, watchOS 6.0, tvOS 13.0, *)
+    func setAttribute(element: Element, attributeName: String, newValue: String) async throws {
+        try await driver.setAttribute(element: element, attributeName: attributeName, newValue: newValue)
+    }
 }

Sources/SwiftWebDriver/WebDrivers/Chrome/ChromeDriver.swift

@@ -247,7 +247,7 @@ public class ChromeDriver: Driver {
         }
     }
 
-    private func executeJavascriptSync(_ script: String, args: [String]) async throws -> PostExecuteResponse {
+    private func executeJavascriptSync(_ script: String, args: [AnyEncodable]) async throws -> PostExecuteResponse {
         guard let sessionId else {
             throw WebDriverError.sessionIdIsNil
         }
@@ -261,7 +261,7 @@ public class ChromeDriver: Driver {
         return try await client.request(request)
     }
 
-    private func executeJavascriptAsync(_ script: String, args: [String]) async throws -> PostExecuteResponse {
+    private func executeJavascriptAsync(_ script: String, args: [AnyEncodable]) async throws -> PostExecuteResponse {
         guard let sessionId else {
             throw WebDriverError.sessionIdIsNil
         }
@@ -279,7 +279,7 @@ public class ChromeDriver: Driver {
     @discardableResult
     public func execute(
         _ script: String,
-        args: [String],
+        args: [AnyEncodable],
         type: DevToolTypes.JavascriptExecutionTypes
     ) async throws -> PostExecuteResponse {
         try await type == .sync ?
@@ -301,6 +301,17 @@ public class ChromeDriver: Driver {
         return Element(baseURL: url, sessionId: sessionId, elementId: response.elementId)
     }
 
+    public func setAttribute(element: Element, attributeName: String, newValue: String) async throws {
+        let script = "arguments[0].setAttribute(arguments[1], arguments[2]);"
+
+        let args: [AnyEncodable] = [
+            AnyEncodable(["element-6066-11e4-a52e-4f735466cecf": element.elementId]),
+            AnyEncodable(attributeName),
+            AnyEncodable(newValue)
+        ]
+        try await execute(script, args: args, type: .sync)
+    }
+
     deinit {
         let url = url
         let sessionId = sessionId

Sources/SwiftWebDriver/WebDrivers/Driver.swift

@@ -44,9 +44,11 @@ public protocol Driver: FindElementProtocol {
 
     func execute(
         _ script: String,
-        args: [String],
+        args: [AnyEncodable],
         type: DevToolTypes.JavascriptExecutionTypes
     ) async throws -> PostExecuteResponse
 
     func getActiveElement() async throws -> Element
+
+    func setAttribute(element: Element, attributeName: String, newValue: String) async throws
 }

Tests/SwiftWebDriverIntegrationTests/ChromeDriver/Element/ChromeDriverSetAttributeIntegrationTests.swift

@@ -0,0 +1,27 @@
+// ChromeDriverSetAttributeIntegrationTests.swift
+// Copyright (c) 2025 GetAutomaApp
+// All source code and related assets are the property of GetAutomaApp.
+// All rights reserved.
+
+@testable import SwiftWebDriver
+import Testing
+
+@Suite("Chrome Driver Set Attribute", .serialized)
+internal class ChromeDriverSetAttributeIntegrationTests: ChromeDriverTest {
+    @Test("Set Attribute")
+    public func setAttribute() async throws {
+        page = "elementHandleTestPage.html"
+        try await driver.navigateTo(urlString: testPageURL.absoluteString)
+
+        let element = try await driver.findElement(.css(.id("attribute")))
+        let newIdentifier = "newidentifier"
+        try await driver.setAttribute(element: element, attributeName: "id", newValue: newIdentifier)
+
+        let elementId = try await element.attribute(name: "id")
+        #expect(elementId == newIdentifier)
+    }
+
+    deinit {
+        // Add deinit logic here
+    }
+}

package.json

@@ -9,6 +9,14 @@
     "install:all": "npx npm-run-all install:swiftformat install:swiftlint config",
     "install:swiftformat": "brew install swiftformat",
     "install:swiftlint": "brew install swiftlint",
+    "compose:up": "npm run compose -- up -d",
+    "compose:build": "npm run compose -- build",
+    "compose:build:no-cache": "npm run compose -- build --no-cache",
+    "compose:build-and-up": "npx npm-run-all --sequential compose:build compose:up",
+    "compose:build-no-cache-and-up": "npx npm-run-all --sequential compose:build:no-cache compose:up",
+    "compose:down": "npm run compose -- down",
+    "compose:ps": "npm run compose -- ps -a",
+    "compose": "docker compose",
     "config": "./.dotfiles/config.sh",
     "update:submodules": "git submodule foreach --recursive 'branch=$(git remote show origin | awk \"/HEAD branch/ {print \\$NF}\"); git checkout $branch && git pull origin $branch' && CHANGED=$(git status --porcelain | grep '^ M \\.dotfiles' || true) && if [ -n \"$CHANGED\" ]; then npm run config; fi && git add -A && git commit -m \"chore: update submodules\" || echo 'No changes to commit'"
   },