From 826a6bddeb4368896ee7ccec91dfc742375b94e1 Mon Sep 17 00:00:00 2001
From: Augustin Gottlieb Pequeno <33221555+aguspe@users.noreply.github.com>
Date: Mon, 24 Jun 2024 17:56:14 +0200
Subject: [PATCH] [rb] Add FedCM support to the ruby selenium client (#13796)

Co-authored-by: aguspe <agustin.pe94@gmail.com>
---
 common/src/web/fedcm/fedcm.html               |  2 +-
 common/src/web/fedcm/fedcm.json               |  3 +-
 rb/lib/selenium/webdriver/chromium/driver.rb  |  1 +
 rb/lib/selenium/webdriver/common.rb           |  3 +
 .../driver_extensions/has_fedcm_dialog.rb     | 55 +++++++++++
 rb/lib/selenium/webdriver/common/fedcm.rb     | 27 ++++++
 .../webdriver/common/fedcm/account.rb         | 50 ++++++++++
 .../selenium/webdriver/common/fedcm/dialog.rb | 74 +++++++++++++++
 rb/lib/selenium/webdriver/remote/bridge.rb    | 40 ++++++++
 .../webdriver/remote/bridge/commands.rb       | 14 ++-
 .../lib/selenium/webdriver/fedcm/account.rbs  | 52 +++++++++++
 .../lib/selenium/webdriver/fedcm/dialog.rbs   | 26 ++++++
 .../lib/selenium/webdriver/remote/bridge.rbs  | 18 ++++
 .../selenium/webdriver/fedcm_spec.rb          | 90 ++++++++++++++++++
 .../webdriver/common/fedcm/account_spec.rb    | 54 +++++++++++
 .../webdriver/common/fedcm/dialog_spec.rb     | 93 +++++++++++++++++++
 16 files changed, 599 insertions(+), 3 deletions(-)
 create mode 100644 rb/lib/selenium/webdriver/common/driver_extensions/has_fedcm_dialog.rb
 create mode 100644 rb/lib/selenium/webdriver/common/fedcm.rb
 create mode 100644 rb/lib/selenium/webdriver/common/fedcm/account.rb
 create mode 100644 rb/lib/selenium/webdriver/common/fedcm/dialog.rb
 create mode 100644 rb/sig/lib/selenium/webdriver/fedcm/account.rbs
 create mode 100644 rb/sig/lib/selenium/webdriver/fedcm/dialog.rbs
 create mode 100644 rb/spec/integration/selenium/webdriver/fedcm_spec.rb
 create mode 100644 rb/spec/unit/selenium/webdriver/common/fedcm/account_spec.rb
 create mode 100644 rb/spec/unit/selenium/webdriver/common/fedcm/dialog_spec.rb
diff --git a/common/src/web/fedcm/fedcm.html b/common/src/web/fedcm/fedcm.html
index 186d9311c70b8..bdf10e2fa8f2d 100644
--- a/common/src/web/fedcm/fedcm.html
+++ b/common/src/web/fedcm/fedcm.html
@@ -1,7 +1,7 @@
 <!DOCTYPE html>
 <script>
 
-let configURL = `https://${location.host}/fedcm/fedcm.json`;
+let configURL = `http://${location.host}/fedcm/fedcm.json`;
 let promise = null;
 
 function triggerFedCm() {
diff --git a/common/src/web/fedcm/fedcm.json b/common/src/web/fedcm/fedcm.json
index 2512240f1e942..0bb2ca15f0d1f 100644
--- a/common/src/web/fedcm/fedcm.json
+++ b/common/src/web/fedcm/fedcm.json
@@ -2,5 +2,6 @@
   "accounts_endpoint": "accounts.json",
   "client_metadata_endpoint": "client_metadata.json",
   "id_assertion_endpoint": "id_assertion",
-  "signin_url": "/signin"
+  "signin_url": "/signin",
+  "login_url": "/login"
 }
diff --git a/rb/lib/selenium/webdriver/chromium/driver.rb b/rb/lib/selenium/webdriver/chromium/driver.rb
index 8633f1e9867d9..766571dbb2a1f 100644
--- a/rb/lib/selenium/webdriver/chromium/driver.rb
+++ b/rb/lib/selenium/webdriver/chromium/driver.rb
@@ -29,6 +29,7 @@ class Driver < WebDriver::Driver
         EXTENSIONS = [DriverExtensions::HasCDP,
                       DriverExtensions::HasBiDi,
                       DriverExtensions::HasCasting,
+                      DriverExtensions::HasFedCmDialog,
                       DriverExtensions::HasNetworkConditions,
                       DriverExtensions::HasNetworkInterception,
                       DriverExtensions::HasWebStorage,
diff --git a/rb/lib/selenium/webdriver/common.rb b/rb/lib/selenium/webdriver/common.rb
index 7473825892268..0a778214f38d0 100644
--- a/rb/lib/selenium/webdriver/common.rb
+++ b/rb/lib/selenium/webdriver/common.rb
@@ -89,6 +89,7 @@
 require 'selenium/webdriver/common/driver_extensions/has_cdp'
 require 'selenium/webdriver/common/driver_extensions/has_casting'
 require 'selenium/webdriver/common/driver_extensions/has_launching'
+require 'selenium/webdriver/common/driver_extensions/has_fedcm_dialog'
 require 'selenium/webdriver/common/keys'
 require 'selenium/webdriver/common/profile_helper'
 require 'selenium/webdriver/common/options'
@@ -99,3 +100,5 @@
 require 'selenium/webdriver/common/websocket_connection'
 require 'selenium/webdriver/common/child_process'
 require 'selenium/webdriver/common/script'
+require 'selenium/webdriver/common/fedcm/account'
+require 'selenium/webdriver/common/fedcm/dialog'
diff --git a/rb/lib/selenium/webdriver/common/driver_extensions/has_fedcm_dialog.rb b/rb/lib/selenium/webdriver/common/driver_extensions/has_fedcm_dialog.rb
new file mode 100644
index 0000000000000..c22eaf9a37898
--- /dev/null
+++ b/rb/lib/selenium/webdriver/common/driver_extensions/has_fedcm_dialog.rb
@@ -0,0 +1,55 @@
+# frozen_string_literal: true
+
+# Licensed to the Software Freedom Conservancy (SFC) under one
+# or more contributor license agreements.  See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership.  The SFC licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License.  You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
+module Selenium
+  module WebDriver
+    module DriverExtensions
+      module HasFedCmDialog
+        # Disables the promise rejection delay for FedCm.
+        #
+        # FedCm by default delays promise resolution in failure cases for privacy reasons.
+        # This method allows turning it off to let tests run faster where this is not relevant.
+        def enable_fedcm_delay=(enable)
+          @bridge.fedcm_delay(enable)
+        end
+
+        # Resets the FedCm dialog cooldown.
+        #
+        # If a user agent triggers a cooldown when the account chooser is dismissed,
+        # this method resets that cooldown so that the dialog can be triggered again immediately.
+        def reset_fedcm_cooldown
+          @bridge.reset_fedcm_cooldown
+        end
+
+        def fedcm_dialog
+          @fedcm_dialog ||= FedCM::Dialog.new(@bridge)
+        end
+
+        def wait_for_fedcm_dialog(timeout: 5, interval: 0.2, message: nil, ignore: nil)
+          wait = Wait.new(timeout: timeout, interval: interval, message: message, ignore: ignore)
+          wait.until do
+            fedcm_dialog if fedcm_dialog.type
+          rescue Error::NoSuchAlertError
+            nil
+          end
+        end
+      end # HasFedCmDialog
+    end # DriverExtensions
+  end # WebDriver
+end # Selenium
diff --git a/rb/lib/selenium/webdriver/common/fedcm.rb b/rb/lib/selenium/webdriver/common/fedcm.rb
new file mode 100644
index 0000000000000..12de1cdfd6f5d
--- /dev/null
+++ b/rb/lib/selenium/webdriver/common/fedcm.rb
@@ -0,0 +1,27 @@
+# frozen_string_literal: true
+
+# Licensed to the Software Freedom Conservancy (SFC) under one
+# or more contributor license agreements.  See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership.  The SFC licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License.  You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
+module Selenium
+  module WebDriver
+    module FedCM
+      autoload :Account, 'fedcm/account'
+      autoload :Dialog, 'fedcm/dialog'
+    end # FedCM
+  end # WebDriver
+end # Selenium
diff --git a/rb/lib/selenium/webdriver/common/fedcm/account.rb b/rb/lib/selenium/webdriver/common/fedcm/account.rb
new file mode 100644
index 0000000000000..95c724e390357
--- /dev/null
+++ b/rb/lib/selenium/webdriver/common/fedcm/account.rb
@@ -0,0 +1,50 @@
+# frozen_string_literal: true
+
+# Licensed to the Software Freedom Conservancy (SFC) under one
+# or more contributor license agreements.  See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership.  The SFC licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License.  You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
+module Selenium
+  module WebDriver
+    module FedCM
+      # Represents an account displayed in a FedCm account list.
+      # See: https://fedidcg.github.io/FedCM/#dictdef-identityprovideraccount
+      #      https://fedidcg.github.io/FedCM/#webdriver-accountlist
+      class Account
+        LOGIN_STATE_SIGNIN = 'SignIn'
+        LOGIN_STATE_SIGNUP = 'SignUp'
+
+        attr_reader :account_id, :email, :name, :given_name, :picture_url,
+                    :idp_config_url, :login_state, :terms_of_service_url, :privacy_policy_url
+
+        # Initializes a new account with the provided attributes.
+        #
+        # @param [Hash]
+        def initialize(**args)
+          @account_id = args['accountId']
+          @email = args['email']
+          @name = args['name']
+          @given_name = args['givenName']
+          @picture_url = args['pictureUrl']
+          @idp_config_url = args['idpConfigUrl']
+          @login_state = args['loginState']
+          @terms_of_service_url = args['termsOfServiceUrl']
+          @privacy_policy_url = args['privacyPolicyUrl']
+        end
+      end # Account
+    end # FedCM
+  end # WebDriver
+end # Selenium
diff --git a/rb/lib/selenium/webdriver/common/fedcm/dialog.rb b/rb/lib/selenium/webdriver/common/fedcm/dialog.rb
new file mode 100644
index 0000000000000..8ef245cd31c57
--- /dev/null
+++ b/rb/lib/selenium/webdriver/common/fedcm/dialog.rb
@@ -0,0 +1,74 @@
+# frozen_string_literal: true
+
+# Licensed to the Software Freedom Conservancy (SFC) under one
+# or more contributor license agreements.  See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership.  The SFC licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License.  You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
+module Selenium
+  module WebDriver
+    module FedCM
+      class Dialog
+        def initialize(bridge)
+          @bridge = bridge
+        end
+
+        DIALOG_TYPE_ACCOUNT_LIST = 'AccountChooser'
+        DIALOG_TYPE_AUTO_REAUTH = 'AutoReauthn'
+
+        # Closes the dialog as if the user had clicked X.
+        def click
+          @bridge.click_fedcm_dialog_button
+        end
+
+        # Closes the dialog as if the user had clicked X.
+        def cancel
+          @bridge.cancel_fedcm_dialog
+        end
+
+        # Selects an account as if the user had clicked on it.
+        #
+        # @param [Integer] index The index of the account to select from the list returned by get_accounts.
+        def select_account(index)
+          @bridge.select_fedcm_account index
+        end
+
+        # Returns the type of the open dialog.
+        #
+        # One of DIALOG_TYPE_ACCOUNT_LIST and DIALOG_TYPE_AUTO_REAUTH.
+        def type
+          @bridge.fedcm_dialog_type
+        end
+
+        # Returns the title of the dialog.
+        def title
+          @bridge.fedcm_title
+        end
+
+        # Returns the subtitle of the dialog or nil if none.
+        def subtitle
+          @bridge.fedcm_subtitle
+        end
+
+        # Returns the accounts shown in the account chooser.
+        #
+        # If this is an auto reauth dialog, returns the single account that is being signed in.
+        def accounts
+          @bridge.fedcm_account_list.map { |account| Account.new(**account) }
+        end
+      end # Dialog
+    end # FedCM
+  end # WebDriver
+end # Selenium
diff --git a/rb/lib/selenium/webdriver/remote/bridge.rb b/rb/lib/selenium/webdriver/remote/bridge.rb
index 3dd8a799c5249..74385dabb920c 100644
--- a/rb/lib/selenium/webdriver/remote/bridge.rb
+++ b/rb/lib/selenium/webdriver/remote/bridge.rb
@@ -602,6 +602,46 @@ def user_verified(verified, authenticator_id)
           execute :set_user_verified, {authenticatorId: authenticator_id}, {isUserVerified: verified}
         end
 
+        #
+        # federated-credential management
+        #
+
+        def cancel_fedcm_dialog
+          execute :cancel_fedcm_dialog
+        end
+
+        def select_fedcm_account(index)
+          execute :select_fedcm_account, {}, {accountIndex: index}
+        end
+
+        def fedcm_dialog_type
+          execute :get_fedcm_dialog_type
+        end
+
+        def fedcm_title
+          execute(:get_fedcm_title).fetch('title')
+        end
+
+        def fedcm_subtitle
+          execute(:get_fedcm_title).fetch('subtitle', nil)
+        end
+
+        def fedcm_account_list
+          execute :get_fedcm_account_list
+        end
+
+        def fedcm_delay(enabled)
+          execute :set_fedcm_delay, {}, {enabled: enabled}
+        end
+
+        def reset_fedcm_cooldown
+          execute :reset_fedcm_cooldown
+        end
+
+        def click_fedcm_dialog_button
+          execute :click_fedcm_dialog_button, {}, {dialogButton: 'ConfirmIdpLoginContinue'}
+        end
+
         def bidi
           msg = 'BiDi must be enabled by setting #web_socket_url to true in options class'
           raise(WebDriver::Error::WebDriverError, msg)
diff --git a/rb/lib/selenium/webdriver/remote/bridge/commands.rb b/rb/lib/selenium/webdriver/remote/bridge/commands.rb
index dee2c041c0626..5eeb61f755961 100644
--- a/rb/lib/selenium/webdriver/remote/bridge/commands.rb
+++ b/rb/lib/selenium/webdriver/remote/bridge/commands.rb
@@ -155,8 +155,20 @@ class Bridge
           remove_credential: [:delete,
                               'session/:session_id/webauthn/authenticator/:authenticatorId/credentials/:credentialId'],
           remove_all_credentials: [:delete, 'session/:session_id/webauthn/authenticator/:authenticatorId/credentials'],
-          set_user_verified: [:post, 'session/:session_id/webauthn/authenticator/:authenticatorId/uv']
+          set_user_verified: [:post, 'session/:session_id/webauthn/authenticator/:authenticatorId/uv'],
 
+          #
+          # federated-credential management
+          #
+
+          get_fedcm_title: [:get, 'session/:session_id/fedcm/gettitle'],
+          get_fedcm_dialog_type: [:get, 'session/:session_id/fedcm/getdialogtype'],
+          get_fedcm_account_list: [:get, 'session/:session_id/fedcm/accountlist'],
+          click_fedcm_dialog_button: [:post, 'session/:session_id/fedcm/clickdialogbutton'],
+          cancel_fedcm_dialog: [:post, 'session/:session_id/fedcm/canceldialog'],
+          select_fedcm_account: [:post, 'session/:session_id/fedcm/selectaccount'],
+          set_fedcm_delay: [:post, 'session/:session_id/fedcm/setdelayenabled'],
+          reset_fedcm_cooldown: [:post, 'session/:session_id/fedcm/resetcooldown']
         }.freeze
       end # Bridge
     end # Remote
diff --git a/rb/sig/lib/selenium/webdriver/fedcm/account.rbs b/rb/sig/lib/selenium/webdriver/fedcm/account.rbs
new file mode 100644
index 0000000000000..46dd9837b2d34
--- /dev/null
+++ b/rb/sig/lib/selenium/webdriver/fedcm/account.rbs
@@ -0,0 +1,52 @@
+module Selenium
+  module WebDriver
+    module FedCM
+      # Represents an account displayed in a FedCm account list.
+      # See: https://fedidcg.github.io/FedCM/#dictdef-identityprovideraccount
+      #      https://fedidcg.github.io/FedCM/#webdriver-accountlist
+      class Account
+        @account_id: String
+
+        @email: String
+
+        @name: String
+
+        @given_name: String
+
+        @picture_url: String
+
+        @idp_config_url: String
+
+        @login_state: String
+
+        @terms_of_service_url: String
+
+        @privacy_policy_url: String
+
+        LOGIN_STATE_SIGNIN: String
+
+        LOGIN_STATE_SIGNUP: String
+
+        attr_reader account_id: String
+
+        attr_reader email: String
+
+        attr_reader name: String
+
+        attr_reader given_name: String
+
+        attr_reader picture_url: String
+
+        attr_reader idp_config_url: String
+
+        attr_reader login_state: String
+
+        attr_reader terms_of_service_url: String
+
+        attr_reader privacy_policy_url: String
+
+        def initialize: (**untyped args) -> void
+      end
+    end
+  end
+end
diff --git a/rb/sig/lib/selenium/webdriver/fedcm/dialog.rbs b/rb/sig/lib/selenium/webdriver/fedcm/dialog.rbs
new file mode 100644
index 0000000000000..c16ef91d6c930
--- /dev/null
+++ b/rb/sig/lib/selenium/webdriver/fedcm/dialog.rbs
@@ -0,0 +1,26 @@
+module Selenium
+  module WebDriver
+    module FedCM
+      class Dialog
+        DIALOG_TYPE_ACCOUNT_LIST: String
+        DIALOG_TYPE_AUTO_REAUTH: String
+
+        @bridge: Remote::Bridge
+
+        def accounts: -> Array[Account]
+
+        def cancel: -> Remote::Response?
+
+        def click: -> untyped
+
+        def select_account: (Integer index) -> Remote::Response?
+
+        def subtitle: -> (String | Remote::Response)?
+
+        def title: -> (String | Remote::Response)
+
+        def type: -> (String | Remote::Response)
+      end
+    end
+  end
+end
diff --git a/rb/sig/lib/selenium/webdriver/remote/bridge.rbs b/rb/sig/lib/selenium/webdriver/remote/bridge.rbs
index 32aaa1a21b8ef..4e06cbd61ed7e 100644
--- a/rb/sig/lib/selenium/webdriver/remote/bridge.rbs
+++ b/rb/sig/lib/selenium/webdriver/remote/bridge.rbs
@@ -32,6 +32,10 @@ module Selenium
 
         def initialize: (url: String | URI, ?http_client: untyped?) -> void
 
+        def cancel_fedcm_dialog: -> nil
+
+        def click_fedcm_dialog_button: -> nil
+
         def create_session: (untyped capabilities) -> untyped
 
         extend WebDriver::Chrome::Features
@@ -42,10 +46,24 @@ module Selenium
 
         extend WebDriver::Safari::Features
 
+        def fedcm_account_list: -> [FedCM::Account]
+
+        def fedcm_dialog_type: -> String
+
+        def fedcm_subtitle: -> String?
+
+        def fedcm_title: -> String
+
+        def reset_fedcm_cooldown: -> nil
+
+        def select_fedcm_account: -> nil
+
         def session_id: () -> untyped
 
         def browser: () -> untyped
 
+        def fedcm_delay: -> bool
+
         def status: () -> untyped
 
         def get: (untyped url) -> untyped
diff --git a/rb/spec/integration/selenium/webdriver/fedcm_spec.rb b/rb/spec/integration/selenium/webdriver/fedcm_spec.rb
new file mode 100644
index 0000000000000..608f3a4361294
--- /dev/null
+++ b/rb/spec/integration/selenium/webdriver/fedcm_spec.rb
@@ -0,0 +1,90 @@
+# frozen_string_literal: true
+
+# Licensed to the Software Freedom Conservancy (SFC) under one
+# or more contributor license agreements.  See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership.  The SFC licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License.  You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
+require_relative 'spec_helper'
+
+module Selenium
+  module WebDriver
+    module FedCM
+      describe FedCM, exclusive: [{bidi: false, reason: 'Not yet implemented with BiDi'}, {browser: %i[chrome edge]}] do
+        let(:dialog) { driver.fedcm_dialog }
+
+        before do
+          driver.get url_for('fedcm/fedcm.html')
+        end
+
+        context 'without dialog present' do
+          it 'throws an error' do
+            expect { dialog.title }.to raise_error(Error::NoSuchAlertError)
+            expect { dialog.subtitle }.to raise_error(Error::NoSuchAlertError)
+            expect { dialog.type }.to raise_error(Error::NoSuchAlertError)
+            expect { dialog.accounts }.to raise_error(Error::NoSuchAlertError)
+            expect { dialog.select_account(1) }.to raise_error(Error::NoSuchAlertError)
+            expect { dialog.cancel }.to raise_error(Error::NoSuchAlertError)
+          end
+        end
+
+        context 'with dialog present' do
+          before do
+            driver.execute_script('triggerFedCm();')
+            driver.wait_for_fedcm_dialog
+          end
+
+          it 'returns the title' do
+            expect(dialog.title).to eq('Sign in to localhost with localhost')
+          end
+
+          it 'returns the subtitle' do
+            expect(dialog.subtitle).to be_nil
+          end
+
+          it 'returns the type' do
+            expect(dialog.type).to eq('AccountChooser')
+          end
+
+          it 'returns the accounts' do
+            first_account = dialog.accounts.first
+            expect(first_account.name).to eq 'John Doe'
+          end
+
+          it 'selects an account' do
+            expect(dialog.select_account(1)).to be_nil
+          end
+
+          it 'clicks the dialog', skip: 'Investigate IDP config issue' do
+            expect(dialog.click).to be_nil
+          end
+
+          it 'cancels the dialog' do
+            dialog.cancel
+            expect { dialog.title }.to raise_error(Error::NoSuchAlertError)
+          end
+
+          it 'sets the delay' do
+            expect(driver.enable_fedcm_delay = true).to be_truthy
+          end
+
+          it 'resets the cooldown' do
+            expect(driver.reset_fedcm_cooldown).to be_nil
+          end
+        end
+      end
+    end # FedCm
+  end # WebDriver
+end # Selenium
diff --git a/rb/spec/unit/selenium/webdriver/common/fedcm/account_spec.rb b/rb/spec/unit/selenium/webdriver/common/fedcm/account_spec.rb
new file mode 100644
index 0000000000000..0f855c3aa5c09
--- /dev/null
+++ b/rb/spec/unit/selenium/webdriver/common/fedcm/account_spec.rb
@@ -0,0 +1,54 @@
+# frozen_string_literal: true
+
+# Licensed to the Software Freedom Conservancy (SFC) under one
+# or more contributor license agreements.  See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership.  The SFC licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License.  You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
+require File.expand_path('../../spec_helper', __dir__)
+
+module Selenium
+  module WebDriver
+    module FedCM
+      describe Account do
+        let(:account) do
+          described_class.new(
+            'accountId' => '12341234',
+            'email' => 'fake@email.com',
+            'name' => 'Real Name',
+            'givenName' => 'Fake Name',
+            'pictureUrl' => 'picture-url',
+            'idpConfigUrl' => 'idp-config-url',
+            'loginState' => 'login-state',
+            'termsOfServiceUrl' => 'terms-of-service-url',
+            'privacyPolicyUrl' => 'privacy-policy-url'
+          )
+        end
+
+        it 'sets the provided attributes' do
+          expect(account.account_id).to eq('12341234')
+          expect(account.email).to eq('fake@email.com')
+          expect(account.name).to eq('Real Name')
+          expect(account.given_name).to eq('Fake Name')
+          expect(account.picture_url).to eq('picture-url')
+          expect(account.idp_config_url).to eq('idp-config-url')
+          expect(account.login_state).to eq('login-state')
+          expect(account.terms_of_service_url).to eq('terms-of-service-url')
+          expect(account.privacy_policy_url).to eq('privacy-policy-url')
+        end
+      end # Account
+    end # FedCM
+  end # WebDriver
+end # Selenium
diff --git a/rb/spec/unit/selenium/webdriver/common/fedcm/dialog_spec.rb b/rb/spec/unit/selenium/webdriver/common/fedcm/dialog_spec.rb
new file mode 100644
index 0000000000000..e15e4e455c5c3
--- /dev/null
+++ b/rb/spec/unit/selenium/webdriver/common/fedcm/dialog_spec.rb
@@ -0,0 +1,93 @@
+# frozen_string_literal: true
+
+# Licensed to the Software Freedom Conservancy (SFC) under one
+# or more contributor license agreements.  See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership.  The SFC licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License.  You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
+require File.expand_path('../../spec_helper', __dir__)
+
+module Selenium
+  module WebDriver
+    module FedCM
+      describe Dialog do
+        let(:bridge) { instance_double(Remote::Bridge) }
+        let(:dialog) { described_class.new(bridge) }
+
+        describe '#click' do
+          it 'calls click_fedcm_dialog on the bridge' do
+            allow(bridge).to receive(:click_fedcm_dialog_button).and_return(nil)
+            expect(dialog.click).to be_nil
+          end
+        end
+
+        describe '#cancel' do
+          it 'calls cancel_fedcm_dialog on the bridge' do
+            allow(bridge).to receive(:cancel_fedcm_dialog).and_return(nil)
+            expect(dialog.cancel).to be_nil
+          end
+        end
+
+        describe '#select_account' do
+          it 'calls select_fedcm_account on the bridge with the given index' do
+            index = 1
+            allow(bridge).to receive(:select_fedcm_account).with(index).and_return(nil)
+            expect(dialog.select_account(index)).to be_nil
+          end
+        end
+
+        describe '#type' do
+          it 'returns the type of the open dialog' do
+            allow(bridge).to receive(:fedcm_dialog_type).and_return('AccountChooser')
+            expect(dialog.type).to eq('AccountChooser')
+          end
+        end
+
+        describe '#title' do
+          it 'returns the title of the dialog' do
+            allow(bridge).to receive(:fedcm_title).and_return('Sign in')
+            expect(dialog.title).to eq('Sign in')
+          end
+        end
+
+        describe '#subtitle' do
+          it 'returns the subtitle of the dialog' do
+            allow(bridge).to receive(:fedcm_subtitle).and_return('Choose an account')
+            expect(dialog.subtitle).to eq('Choose an account')
+          end
+
+          it 'returns nil if there is no subtitle' do
+            allow(bridge).to receive(:fedcm_subtitle).and_return(nil)
+            expect(dialog.subtitle).to be_nil
+          end
+        end
+
+        describe '#accounts' do
+          it 'returns the accounts shown in the account chooser' do
+            accounts_data = [{'name' => 'Account1', 'email' => 'account1@example.com'},
+                             {'name' => 'Account2', 'email' => 'account2@example.com'}]
+            allow(bridge).to receive(:fedcm_account_list).and_return(accounts_data)
+            accounts = dialog.accounts
+            expect(accounts.size).to eq(2)
+            expect(accounts[0].name).to eq('Account1')
+            expect(accounts[0].email).to eq('account1@example.com')
+            expect(accounts[1].name).to eq('Account2')
+            expect(accounts[1].email).to eq('account2@example.com')
+          end
+        end
+      end # Dialog
+    end # FedCM
+  end # WebDriver
+end # Selenium
