Update intercept request and response and improve headers and cookies

Author: aguspe

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

@@ -21,16 +21,13 @@ module Selenium
   module WebDriver
     class BiDi
       class Cookies < Hash
-        def initialize(cookies = {})
-          super()
-          merge!(cookies)
-        end
-
         def as_json
-          self[:name] = self[:name].to_s
-          self[:value] = {type: 'string', value: self[:value].to_s}
+          map do |name, val|
+            self[:name] = name.to_s
+            self[:value] = {type: 'string', value: val.to_s}
 
-          [compact]
+            [compact]
+          end
         end
       end
     end # BiDi

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

@@ -25,21 +25,23 @@ module WebDriver
     class BiDi
       class InterceptedRequest < InterceptedItem
         attr_accessor :method, :url
-        attr_reader :body
+        attr_reader :body, :headers, :cookies
 
         def initialize(network, request)
           super
           @method = nil
           @url = nil
           @body = nil
+          @headers = nil
+          @cookies = nil
         end
 
         def continue
           network.continue_request(
             id: id,
             body: body,
-            cookies: cookies.as_json,
-            headers: headers.as_json,
+            cookies: cookies&.as_json,
+            headers: headers&.as_json,
             method: method,
             url: url
           )
@@ -56,12 +58,12 @@ def body=(value)
           }
         end
 
-        def headers
-          @headers ||= Headers.new
+        def headers=(headers = {})
+          @headers = Headers.new(headers)
         end
 
-        def cookies(cookies = {})
-          @cookies ||= Cookies.new(cookies)
+        def cookies=(cookies = {})
+          @cookies = Cookies.new(cookies)
         end
       end
     end # BiDi

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

@@ -26,20 +26,22 @@ module WebDriver
     class BiDi
       class InterceptedResponse < InterceptedItem
         attr_accessor :reason, :status
-        attr_reader :body
+        attr_reader :body, :headers, :cookies
 
         def initialize(network, request)
           super
           @reason = nil
           @status = nil
           @body = nil
+          @headers = nil
+          @cookies = nil
         end
 
         def continue
           network.continue_response(
             id: id,
-            cookies: cookies.as_json,
-            headers: headers.as_json,
+            cookies: cookies&.as_json,
+            headers: headers&.as_json,
             credentials: credentials.as_json,
             reason: reason,
             status: status
@@ -49,8 +51,8 @@ def continue
         def provide_response
           network.provide_response(
             id: id,
-            cookies: cookies.as_json,
-            headers: headers.as_json,
+            cookies: cookies&.as_json,
+            headers: headers&.as_json,
             body: body,
             reason: reason,
             status: status
@@ -61,11 +63,11 @@ def credentials(username: nil, password: nil)
           @credentials ||= Credentials.new(username: username, password: password)
         end
 
-        def headers
-          @headers ||= Headers.new
+        def headers=(*headers)
+          @headers = Headers.new(headers)
         end
 
-        def cookies(cookies = {})
+        def cookies=(cookies = {})
           @cookies ||= Cookies.new(cookies)
         end
 

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

@@ -2,15 +2,15 @@ module Selenium
   module WebDriver
     class BiDi
       class InterceptedRequest < InterceptedItem
-        @method: String
+        @method: String?
 
-        @url: String
+        @url: String?
 
-        @body: Hash[untyped, untyped]
+        @body: Hash[untyped, untyped]?
 
-        @headers: Hash[untyped, untyped]
+        @headers: Hash[untyped, untyped]?
 
-        @cookies: Hash[untyped, untyped]
+        @cookies: Hash[untyped, untyped]?
 
         attr_accessor method: String
 
@@ -22,13 +22,17 @@ module Selenium
 
         def continue: () -> Hash[nil, nil]
 
+        def cookies=: -> Hash[nil, nil]
+
         def fail: () -> Hash[nil, nil]
 
         def body=: (Hash[untyped, untyped] value) -> Hash[untyped, untyped]
 
         def headers: () -> Hash[untyped, untyped]
 
         def cookies: () -> Hash[untyped, untyped]
+
+        def headers=: -> Hash[untyped, untyped]
       end
     end
   end

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

@@ -5,7 +5,7 @@ module Selenium
         @body: untyped
         @cookies: Hash[Symbol, String | Integer]?
 
-        @reason: String
+        @reason: String?
 
         @credentials: untyped
 
@@ -24,10 +24,14 @@ module Selenium
 
         def cookies:(?Hash[Symbol, String | Integer]? set_cookie_headers) -> untyped
 
+        def cookies=: -> Hash[Symbol, String | Integer]
+
         def credentials: (?username: untyped?, ?password: untyped?) -> untyped
 
         def headers: () -> untyped
 
+        def headers=: -> Hash[untyped, untyped]
+
         def provide_response: -> untyped
 
         def set_cookie_headers: (?untyped? set_cookie_headers) -> untyped

rb/spec/unit/selenium/webdriver/bidi/intercepted_request_spec.rb

@@ -0,0 +1,104 @@
+# 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__)
+require File.expand_path('../../../../../lib/selenium/webdriver/bidi/network/cookies', __dir__)
+require File.expand_path('../../../../../lib/selenium/webdriver/bidi/network/headers', __dir__)
+require File.expand_path('../../../../../lib/selenium/webdriver/bidi/network/intercepted_request', __dir__)
+
+module Selenium
+  module WebDriver
+    class BiDi
+      describe InterceptedRequest do
+        let(:mock_network) { instance_double(Network) }
+        let(:mock_request) { {'request' => 'req-123'} }
+        let(:request_id) { 'req-123' }
+        let(:intercepted_request) { described_class.new(mock_network, mock_request) }
+        let(:mock_headers) { [{name: 'Auth', value: 'token'}] }
+        let(:mock_cookies) { [{name: 'session', value: {type: 'string', value: '123'}}] }
+
+        describe '#continue' do
+          before do
+            allow(mock_network).to receive(:continue_request)
+          end
+
+          it 'sends only the mandatory ID when no optional fields are set' do
+            expected_payload = {id: request_id, body: nil, cookies: nil, headers: nil, method: nil, url: nil}
+            intercepted_request.continue
+
+            expect(mock_network).to have_received(:continue_request).with(expected_payload)
+          end
+
+          it 'sends headers payload when headers are explicitly set' do
+            intercepted_request.headers = mock_headers
+
+            expected_payload = {
+              id: request_id,
+              body: nil,
+              cookies: nil,
+              headers: Headers.new(mock_headers).as_json,
+              method: nil,
+              url: nil
+            }
+
+            intercepted_request.continue
+
+            expect(mock_network).to have_received(:continue_request).with(expected_payload)
+          end
+
+          it 'sends cookies payload when cookies are explicitly set' do
+            intercepted_request.cookies = mock_cookies
+
+            expected_payload = {
+              id: request_id,
+              body: nil,
+              cookies: Cookies.new(mock_cookies).as_json,
+              headers: nil,
+              method: nil,
+              url: nil
+            }
+
+            intercepted_request.continue
+
+            expect(mock_network).to have_received(:continue_request).with(expected_payload)
+          end
+
+          it 'sends full custom payload when all fields are set' do
+            intercepted_request.headers = mock_headers
+            intercepted_request.cookies = mock_cookies
+            intercepted_request.method = 'POST'
+
+            expected_payload = {
+              id: request_id,
+              body: nil,
+              cookies: Cookies.new(mock_cookies).as_json,
+              headers: Headers.new(mock_headers).as_json,
+              method: 'POST',
+              url: nil
+            }
+
+            intercepted_request.continue
+
+            expect(mock_network).to have_received(:continue_request).with(expected_payload)
+          end
+        end
+      end
+    end
+  end
+end