[rb][bidi] network module events

Author: TamsilAmani

rb/lib/selenium/webdriver/bidi.rb

@@ -23,6 +23,7 @@ class BiDi
       autoload :Session, 'selenium/webdriver/bidi/session'
       autoload :LogInspector, 'selenium/webdriver/bidi/log_inspector'
       autoload :BrowsingContext, 'selenium/webdriver/bidi/browsing_context'
+      autoload :NetworkInspector, 'selenium/webdriver/bidi/network_inspector'
 
       def initialize(url:)
         @ws = WebSocketConnection.new(url: url)

rb/lib/selenium/webdriver/bidi/network_inspector.rb

@@ -0,0 +1,60 @@
+# 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
+    class BiDi
+      class NetworkInspector
+        EVENTS = {
+          before_request_sent: 'beforeRequestSent',
+          fetch_error: 'fetchError',
+          response_completed: 'responseCompleted',
+          response_started: 'responseStarted'
+        }.freeze
+
+        def initialize(driver, browsing_context_ids = nil)
+          unless driver.capabilities.web_socket_url
+            raise Error::WebDriverError,
+                  'WebDriver instance must support BiDi protocol'
+          end
+
+          @bidi = driver.bidi
+          @bidi.session.subscribe('network.beforeRequestSent', browsing_context_ids)
+        end
+
+        def before_request_sent(&block)
+          on(:before_request_sent) do |params|
+            before_request_sent_event(params, &block)
+          end
+        end
+
+        private
+
+        def on(event, &block)
+          event = EVENTS[event] if event.is_a?(Symbol)
+          @bidi.callbacks["network.#{event}"] << block
+        end
+
+        def before_request_sent_event(params)
+          yield(params)
+        end
+      end # NetworkInspector
+    end # Bidi
+  end # WebDriver
+end # Selenium

rb/spec/integration/selenium/webdriver/bidi/network_inspector_spec.rb

@@ -0,0 +1,66 @@
+# 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
+    class BiDi
+      describe NetworkInspector, only: {browser: %i[chrome edge firefox]} do
+        let(:empty_page) { '/bidi/emptyPage.html' }
+        let(:empty_text) { '/bidi/emptyText.txt' }
+        let(:redirected_http_equiv) { '/bidi/redirected_http_equiv.html' }
+
+        it 'can listen to event before request is sent' do
+          reset_driver!(web_socket_url: true) do |driver|
+            before_request_event = []
+            inspector = described_class.new(driver)
+            inspector.before_request_sent { |event| before_request_event.push(event) }
+
+            driver.navigate.to url_for(empty_page)
+
+            wait.until { !before_request_event.nil? }
+
+            expect(before_request_event[0].dig('request', 'method')).to eq 'GET'
+            url = before_request_event[0].dig('request', 'url')
+            expect(url).to eq driver.current_url
+          end
+        end
+
+        it 'can request cookies' do
+          reset_driver!(web_socket_url: true) do |driver|
+            before_request_event = []
+            inspector = described_class.new(driver)
+            inspector.before_request_sent { |event| before_request_event.push(event) }
+
+            driver.navigate.to url_for(empty_text)
+            driver.manage.add_cookie name: 'north',
+                                     value: 'biryani'
+            driver.navigate.refresh
+            wait.until { !before_request_event.nil? }
+
+            cookies = before_request_event[0]['request']['cookies']
+            puts "cookies = \n", cookies # prints nil
+            expect(before_request_event[0].dig('request', 'method')).to eq 'GET'
+          end
+        end
+      end
+    end
+  end
+end