Refactoring default host

brynary · Mar 3, 2011 · 55c9a99 · 55c9a99
1 parent ef084a7
commit 55c9a99
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/lib/webrat/core/session.rb b/lib/webrat/core/session.rb
@@ -286,11 +286,15 @@ def response_location
  end
 
  def current_host
- URI.parse(current_url).host || @custom_headers["Host"] || adapter.session.rack_mock_session.default_host
+ URI.parse(current_url).host || @custom_headers["Host"] || default_current_host
  end
 
  def response_location_host
- URI.parse(response_location).host || adapter.session.rack_mock_session.default_host
+ URI.parse(response_location).host || default_current_host
+ end
+
+ def default_current_host
+ adapter.session.rack_mock_session.default_host
  end
 
  def reset