add provide response support

Author: aguspe

rb/lib/selenium/webdriver/bidi/network.rb

@@ -112,6 +112,18 @@ def continue_response(**args)
           )
         end
 
+        def provide_response(**args)
+          @bidi.send_cmd(
+            'network.provideResponse',
+            request: args[:id],
+            body: args[:body],
+            cookies: args[:cookies],
+            headers: args[:headers],
+            reasonPhrase: args[:reason],
+            statusCode: args[:status]
+          )
+        end
+
         def on(event, &)
           event = EVENTS[event] if event.is_a?(Symbol)
           @bidi.add_callback(event, &)

rb/lib/selenium/webdriver/bidi/network/intercepted_response.rb

@@ -25,11 +25,14 @@ module Selenium
   module WebDriver
     class BiDi
       class InterceptedResponse < InterceptedItem
-        attr_accessor :reason
+        attr_accessor :reason, :status
+        attr_reader :body
 
         def initialize(network, request)
           super
           @reason = nil
+          @status = nil
+          @body = nil
         end
 
         def continue
@@ -38,7 +41,19 @@ def continue
             cookies: set_cookie_headers.serialize,
             headers: headers.serialize,
             credentials: credentials.serialize,
-            reason: reason
+            reason: reason,
+            status: status
+          )
+        end
+
+        def provide_response
+          network.provide_response(
+            id: id,
+            cookies: set_cookie_headers.serialize,
+            headers: headers.serialize,
+            body: body,
+            reason: reason,
+            status: status
           )
         end
 
@@ -53,6 +68,13 @@ def headers
         def set_cookie_headers(set_cookie_headers = {})
           @set_cookie_headers ||= SetCookieHeaders.new(set_cookie_headers)
         end
+
+        def body=(value)
+          @body = {
+            type: 'string',
+            value: value.to_json
+          }
+        end
       end
     end # BiDi
   end # WebDriver

rb/sig/lib/selenium/webdriver/bidi/network.rbs

@@ -22,6 +22,8 @@ module Selenium
 
         def fail_request: -> Hash[nil, nil]
 
+        def provide_response: -> Hash[nil, nil]
+
         def remove_intercept: (String intercept) -> Hash[nil, nil]
 
         def continue_with_auth: (String request_id, String username, String password) -> Hash[nil, nil]

rb/sig/lib/selenium/webdriver/bidi/network/intercepted_response.rbs

@@ -2,6 +2,7 @@ module Selenium
   module WebDriver
     class BiDi
       class InterceptedResponse < InterceptedItem
+        @body: untyped
         @reason: untyped
 
         @credentials: untyped
@@ -10,16 +11,23 @@ module Selenium
 
         @set_cookie_headers: untyped
 
+        attr_reader body: untyped
         attr_accessor reason: untyped
 
+        attr_accessor status: Integer?
+
         def initialize: (untyped network, untyped request) -> void
 
+        def body=: -> untyped
+
         def continue: () -> untyped
 
         def credentials: (?username: untyped?, ?password: untyped?) -> untyped
 
         def headers: () -> untyped
 
+        def provide_response: -> untyped
+
         def set_cookie_headers: (?untyped? set_cookie_headers) -> untyped
       end
     end

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

@@ -144,6 +144,34 @@ class BiDi
             expect(driver.find_element(name: 'login')).to be_displayed
           end
         end
+
+        it 'provides response' do
+          reset_driver!(web_socket_url: true) do |driver|
+            network = described_class.new(driver.bidi)
+            network.add_intercept(phases: [described_class::PHASES[:response_started]])
+            network.on(:response_started) do |event|
+              request_id = event['request']['request']
+              network.provide_response(id: request_id,
+                                       status: 200,
+                                       headers: [
+                                         {
+                                           name: 'foo',
+                                           value: {
+                                             type: 'string',
+                                             value: 'bar'
+                                           }
+                                         }
+                                       ],
+                                       body: {
+                                         type: 'string',
+                                         value: '<html><head><title>Hello World!</title></head><body/></html>'
+                                       })
+            end
+
+            driver.navigate.to url_for('formPage.html')
+            expect(driver.page_source).to include('Hello World!')
+          end
+        end
       end
     end
   end

rb/spec/integration/selenium/webdriver/network_spec.rb

@@ -150,7 +150,7 @@ module WebDriver
             request.headers['foo'] = 'bar'
             request.headers['baz'] = 'qux'
             request.cookies['foo'] = 'bar'
-            request.body = ({test: 'example'})
+            request.body = ({ test: 'example' })
             request.continue
           end
           driver.navigate.to url_for('formPage.html')
@@ -274,6 +274,20 @@ module WebDriver
         end
       end
 
+      it 'adds a response handler that provides a response' do
+        reset_driver!(web_socket_url: true) do |driver|
+          network = described_class.new(driver)
+          network.add_response_handler do |response|
+            response.status = 200
+            response.headers['foo'] = 'bar'
+            response.body = '<html><head><title>Hello World!</title></head><body/></html>'
+            response.provide_response
+          end
+          driver.navigate.to url_for('formPage.html')
+          expect(driver.page_source).to include('Hello World!')
+        end
+      end
+
       it 'removes a response handler' do
         reset_driver!(web_socket_url: true) do |driver|
           network = described_class.new(driver)