test/test_subclass.rb in mechanize-0.7.8 vs test/test_subclass.rb in mechanize-0.8.0

@@ -1,21 +1,14 @@
-require File.dirname(__FILE__) + "/helper"
+require File.expand_path(File.join(File.dirname(__FILE__), "helper"))
 
-class MechSubclass < WWW::Mechanize
-  def set_headers(uri, request, cur_page)
-    super(uri, request, cur_page)
-    request.add_field('Cookie', 'name=Aaron')
-    request
-  end
-end
-
 class TestSubclass < Test::Unit::TestCase
   def setup
-    @agent = MechSubclass.new
+    @agent = WWW::Mechanize.new
   end
 
   def test_send_cookie
-    page = @agent.get("http://localhost/send_cookies")
+    page = @agent.get(  :url      => "http://localhost/send_cookies",
+                        :headers  => {'Cookie' => 'name=Aaron'} )
     assert_equal(1, page.links.length)
     assert_not_nil(page.links.find { |l| l.text == "name:Aaron" })
   end
 end