lib/webrat/core/session.rb in jwilger-webrat-0.4.4.2 vs lib/webrat/core/session.rb in jwilger-webrat-0.4.4.3

@@ -253,10 +253,11 @@
     def_delegators :current_scope, :click_button,       :clicks_button
     def_delegators :current_scope, :field_labeled
     def_delegators :current_scope, :field_by_xpath
     def_delegators :current_scope, :field_with_id
     def_delegators :current_scope, :select_option
+    def_delegators :current_scope, :field_named
 
   private
 
     def process_request(http_method, url, data, headers)
       if headers.empty?
@@ -269,10 +270,10 @@
     def response_location
       response.headers["Location"]
     end
 
     def current_host
-      URI.parse(current_url).host || "www.example.com"
+      URI.parse(current_url).host || @custom_headers["Host"] || "www.example.com"
     end
 
     def response_location_host
       URI.parse(response_location).host || "www.example.com"
     end