fix failing tests... heh

jamesgolick · May 6, 2009 · 97e95dc · 97e95dc
1 parent 2bbffc1
commit 97e95dc
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/test/session_test.rb b/test/session_test.rb
@@ -29,6 +29,7 @@ def setup
       stub(@session).last_response do
         response = RestClient::Response.new("", stub!)
         stub(response).cookies { {} }
+        stub(response).code { 200 }
       end
       @session.trample
     end
@@ -43,6 +44,7 @@ def setup
       stub(RestClient).get(anything, anything) do
         response = RestClient::Response.new("", stub!)
         stub(response).cookies { {"xyz" => "abc"} }
+        stub(response).code { 200 }
       end
 
       @config = Trample::Configuration.new do

diff --git a/test/test_helper.rb b/test/test_helper.rb
@@ -19,6 +19,7 @@ def mock_get(url, opts={})
     mock(RestClient).get(url, :cookies => opts[:cookies] || {}).times(opts[:times]) do
       response = RestClient::Response.new("", stub!)
       stub(response).cookies { opts[:return_cookies] || {} }
+      stub(response).code { 200 }
     end
   end
 
@@ -27,13 +28,15 @@ def mock_post(url, opts={})
                                :cookies => opts[:cookies] || {}).times(opts[:times]) do
       response = RestClient::Response.new("", stub!)
       stub(response).cookies { opts[:return_cookies] || {} }
+      stub(response).code { 200 }
     end
   end
 
   def stub_get(url, opts = {})
     stub(RestClient).get(url, :cookies => opts[:cookies] || {}).times(opts[:times]) do
       response = RestClient::Response.new("", stub!)
       stub(response).cookies { opts[:return_cookies] || {} }
+      stub(response).code { 200 }
     end
   end
 end