mechanize 0.3.1 → 0.4.0

data/test/tc_cookies.rb

@@ -0,0 +1,125 @@
+$:.unshift File.join(File.dirname(__FILE__), "..", "lib")
+
+require 'webrick'
+require 'test/unit'
+require 'rubygems'
+require 'mechanize'
+require 'servlets'
+require 'net/http'
+require 'uri'
+
+class FormsMechTest < Test::Unit::TestCase
+  def setup
+    @port = 2000
+  end
+
+  def test_send_cookies
+    agent = WWW::Mechanize.new { |a| a.log = Logger.new(nil) }
+    page = agent.get("http://localhost:#{@port}/many_cookies")
+    page = agent.get("http://localhost:#{@port}/send_cookies")
+    assert_equal(2, page.links.length)
+    assert_not_nil(page.links.find { |l| l.text == "name:Aaron" })
+    assert_not_nil(page.links.find { |l| l.text == "no_expires:nope" })
+  end
+
+  def test_many_cookies_as_string
+    agent = WWW::Mechanize.new { |a| a.log = Logger.new(nil) }
+    page = agent.get("http://localhost:#{@port}/many_cookies_as_string")
+    assert_equal(5, agent.cookies.length)
+
+    name_cookie = agent.cookies.find { |k| k.name == "name" }
+    assert_not_nil(name_cookie, "Name cookie was nil")
+    assert_equal("Aaron", name_cookie.value)
+    assert_equal("/", name_cookie.path)
+    assert_equal(true, DateTime.now < name_cookie.expires)
+
+    expired_cookie = agent.cookies.find { |k| k.name == "expired" }
+    assert_not_nil(expired_cookie, "Expired cookie was nil")
+    assert_equal("doh", expired_cookie.value)
+    assert_equal("/", expired_cookie.path)
+    assert_equal(true, DateTime.now > expired_cookie.expires)
+
+    no_exp_cookie = agent.cookies.find { |k| k.name == "no_expires" }
+    assert_not_nil(no_exp_cookie, "No expires cookie is nil")
+    assert_equal("nope", no_exp_cookie.value)
+    assert_equal("/", no_exp_cookie.path)
+    assert_nil(no_exp_cookie.expires)
+
+    path_cookie = agent.cookies.find { |k| k.name == "a_path" }
+    assert_not_nil(path_cookie, "Path cookie is nil")
+    assert_equal("some_path", path_cookie.value)
+    assert_equal(true, DateTime.now < path_cookie.expires)
+
+    no_path_cookie = agent.cookies.find { |k| k.name == "no_path" }
+    assert_not_nil(no_path_cookie, "No path cookie is nil")
+    assert_equal("no_path", no_path_cookie.value)
+    assert_equal("/many_cookies_as_string", no_path_cookie.path)
+    assert_equal(true, DateTime.now < no_path_cookie.expires)
+  end
+
+  def test_many_cookies
+    agent = WWW::Mechanize.new { |a| a.log = Logger.new(nil) }
+    page = agent.get("http://localhost:#{@port}/many_cookies")
+    assert_equal(5, agent.cookies.length)
+
+    name_cookie = agent.cookies.find { |k| k.name == "name" }
+    assert_not_nil(name_cookie, "Name cookie was nil")
+    assert_equal("Aaron", name_cookie.value)
+    assert_equal("/", name_cookie.path)
+    assert_equal(true, DateTime.now < name_cookie.expires)
+
+    expired_cookie = agent.cookies.find { |k| k.name == "expired" }
+    assert_not_nil(expired_cookie, "Expired cookie was nil")
+    assert_equal("doh", expired_cookie.value)
+    assert_equal("/", expired_cookie.path)
+    assert_equal(true, DateTime.now > expired_cookie.expires)
+
+    no_exp_cookie = agent.cookies.find { |k| k.name == "no_expires" }
+    assert_not_nil(no_exp_cookie, "No expires cookie is nil")
+    assert_equal("nope", no_exp_cookie.value)
+    assert_equal("/", no_exp_cookie.path)
+    assert_nil(no_exp_cookie.expires)
+
+    path_cookie = agent.cookies.find { |k| k.name == "a_path" }
+    assert_not_nil(path_cookie, "Path cookie is nil")
+    assert_equal("some_path", path_cookie.value)
+    assert_equal(true, DateTime.now < path_cookie.expires)
+
+    no_path_cookie = agent.cookies.find { |k| k.name == "no_path" }
+    assert_not_nil(no_path_cookie, "No path cookie is nil")
+    assert_equal("no_path", no_path_cookie.value)
+    assert_equal("/many_cookies", no_path_cookie.path)
+    assert_equal(true, DateTime.now < no_path_cookie.expires)
+  end
+
+  def test_get_cookie
+    agent = WWW::Mechanize.new { |a| a.log = Logger.new(nil) }
+    assert_equal(true,
+      agent.cookie_jar.empty?(
+      URI::parse("http://localhost:#{@port}/one_cookie")))
+
+    assert_equal(0, agent.cookies.length)
+
+    page = agent.get("http://localhost:#{@port}/one_cookie")
+    assert_equal(1, agent.cookies.length)
+
+    cookie = agent.cookies.first
+    assert_equal("foo", cookie.name)
+    assert_equal("bar", cookie.value)
+    assert_equal("/", cookie.path)
+    assert_equal("localhost", cookie.domain)
+
+    assert_equal(false,
+      agent.cookie_jar.empty?(
+      URI::parse("http://localhost:#{@port}/one_cookie")))
+    page = agent.get("http://localhost:#{@port}/one_cookie")
+
+    assert_equal(1, agent.cookies.length)
+
+    cookie = agent.cookies.first
+    assert_equal("foo", cookie.name)
+    assert_equal("bar", cookie.value)
+    assert_equal("/", cookie.path)
+    assert_equal("localhost", cookie.domain)
+  end
+end

data/test/tc_forms.rb

@@ -0,0 +1,339 @@
+$:.unshift File.join(File.dirname(__FILE__), "..", "lib")
+
+require 'webrick'
+require 'test/unit'
+require 'rubygems'
+require 'mechanize'
+require 'servlets'
+
+class FormsMechTest < Test::Unit::TestCase
+  def setup
+    @port = 2000
+  end
+
+  def test_post
+    agent = WWW::Mechanize.new { |a| a.log = Logger.new(nil) }
+    page = agent.get("http://localhost:#{@port}/form_test.html")
+    post_form = page.forms.find { |f| f.name == "post_form1" }
+    assert_not_nil(post_form, "Post form is null")
+    assert_equal("post", post_form.method.downcase)
+    assert_equal("/form_post", post_form.action)
+    assert_equal(1, post_form.fields.size)
+    assert_equal(1, post_form.buttons.size)
+    assert_equal(2, post_form.radiobuttons.size)
+    assert_equal(2, post_form.checkboxes.size)
+    assert_not_nil(post_form.fields.find { |f| f.name == "first_name" },
+      "First name field was nil"
+    )
+    assert_not_nil(
+    post_form.radiobuttons.find { |f| f.name == "gender" && f.value == "male"},
+      "Gender male button was nil"
+    )
+
+    assert_not_nil(
+    post_form.radiobuttons.find {|f| f.name == "gender" && f.value == "female"},
+      "Gender female button was nil"
+    )
+
+    assert_not_nil(post_form.checkboxes.find { |f| f.name == "cool person" },
+      "couldn't find cool person checkbox")
+    assert_not_nil(post_form.checkboxes.find { |f| f.name == "likes ham" },
+      "couldn't find likes ham checkbox")
+
+    # Now set all the fields
+    post_form.fields.find { |f| f.name == "first_name" }.value = "Aaron"
+    post_form.radiobuttons.find { |f| 
+      f.name == "gender" && f.value == "male" 
+    }.checked = true
+    post_form.checkboxes.find { |f| f.name == "likes ham" }.checked = true
+    page = agent.submit(post_form, post_form.buttons.first)
+
+    # Check that the submitted fields exist
+    assert_equal(3, page.links.size, "Not enough links")
+    assert_not_nil(
+      page.links.find { |l| l.text == "likes ham:on" },
+      "likes ham check box missing"
+    )
+    assert_not_nil(
+      page.links.find { |l| l.text == "first_name:Aaron" },
+      "first_name field missing"
+    )
+    assert_not_nil(
+      page.links.find { |l| l.text == "gender:male" },
+      "gender field missing"
+    )
+  end
+
+  def test_get
+    agent = WWW::Mechanize.new { |a| a.log = Logger.new(nil) }
+    page = agent.get("http://localhost:#{@port}/form_test.html")
+    get_form = page.forms.find { |f| f.name == "get_form1" }
+    assert_not_nil(get_form, "Get form is null")
+    assert_equal("get", get_form.method.downcase)
+    assert_equal("/form_post", get_form.action)
+    assert_equal(1, get_form.fields.size)
+    assert_equal(1, get_form.buttons.size)
+    assert_equal(2, get_form.radiobuttons.size)
+    assert_equal(2, get_form.checkboxes.size)
+    assert_not_nil(get_form.fields.find { |f| f.name == "first_name" },
+      "First name field was nil"
+    )
+    assert_not_nil(
+    get_form.radiobuttons.find { |f| f.name == "gender" && f.value == "male"},
+      "Gender male button was nil"
+    )
+
+    assert_not_nil(
+    get_form.radiobuttons.find {|f| f.name == "gender" && f.value == "female"},
+      "Gender female button was nil"
+    )
+
+    assert_not_nil(get_form.checkboxes.find { |f| f.name == "cool person" },
+      "couldn't find cool person checkbox")
+    assert_not_nil(get_form.checkboxes.find { |f| f.name == "likes ham" },
+      "couldn't find likes ham checkbox")
+
+    # Now set all the fields
+    get_form.fields.find { |f| f.name == "first_name" }.value = "Aaron"
+    get_form.radiobuttons.find { |f| 
+      f.name == "gender" && f.value == "male" 
+    }.checked = true
+    get_form.checkboxes.find { |f| f.name == "likes ham" }.checked = true
+    page = agent.submit(get_form, get_form.buttons.first)
+
+    # Check that the submitted fields exist
+    assert_equal(3, page.links.size, "Not enough links")
+    assert_not_nil(
+      page.links.find { |l| l.text == "likes ham:on" },
+      "likes ham check box missing"
+    )
+    assert_not_nil(
+      page.links.find { |l| l.text == "first_name:Aaron" },
+      "first_name field missing"
+    )
+    assert_not_nil(
+      page.links.find { |l| l.text == "gender:male" },
+      "gender field missing"
+    )
+  end
+
+  def test_post_with_space_in_action
+    agent = WWW::Mechanize.new { |a| a.log = Logger.new(nil) }
+    page = agent.get("http://localhost:#{@port}/form_test.html")
+    post_form = page.forms.find { |f| f.name == "post_form2" }
+    assert_not_nil(post_form, "Post form is null")
+    assert_equal("post", post_form.method.downcase)
+    assert_equal("/form post", post_form.action)
+    assert_equal(1, post_form.fields.size)
+    assert_equal(1, post_form.buttons.size)
+    assert_equal(2, post_form.radiobuttons.size)
+    assert_equal(2, post_form.checkboxes.size)
+    assert_not_nil(post_form.fields.find { |f| f.name == "first_name" },
+      "First name field was nil"
+    )
+    assert_not_nil(
+    post_form.radiobuttons.find { |f| f.name == "gender" && f.value == "male"},
+      "Gender male button was nil"
+    )
+
+    assert_not_nil(
+    post_form.radiobuttons.find {|f| f.name == "gender" && f.value == "female"},
+      "Gender female button was nil"
+    )
+
+    assert_not_nil(post_form.checkboxes.find { |f| f.name == "cool person" },
+      "couldn't find cool person checkbox")
+    assert_not_nil(post_form.checkboxes.find { |f| f.name == "likes ham" },
+      "couldn't find likes ham checkbox")
+
+    # Now set all the fields
+    post_form.fields.find { |f| f.name == "first_name" }.value = "Aaron"
+    post_form.radiobuttons.find { |f| 
+      f.name == "gender" && f.value == "male" 
+    }.checked = true
+    post_form.checkboxes.find { |f| f.name == "likes ham" }.checked = true
+    page = agent.submit(post_form, post_form.buttons.first)
+
+    # Check that the submitted fields exist
+    assert_equal(3, page.links.size, "Not enough links")
+    assert_not_nil(
+      page.links.find { |l| l.text == "likes ham:on" },
+      "likes ham check box missing"
+    )
+    assert_not_nil(
+      page.links.find { |l| l.text == "first_name:Aaron" },
+      "first_name field missing"
+    )
+    assert_not_nil(
+      page.links.find { |l| l.text == "gender:male" },
+      "gender field missing"
+    )
+  end
+
+  def test_get_with_space_in_action
+    agent = WWW::Mechanize.new { |a| a.log = Logger.new(nil) }
+    page = agent.get("http://localhost:#{@port}/form_test.html")
+    get_form = page.forms.find { |f| f.name == "get_form2" }
+    assert_not_nil(get_form, "Get form is null")
+    assert_equal("get", get_form.method.downcase)
+    assert_equal("/form post", get_form.action)
+    assert_equal(1, get_form.fields.size)
+    assert_equal(1, get_form.buttons.size)
+    assert_equal(2, get_form.radiobuttons.size)
+    assert_equal(2, get_form.checkboxes.size)
+    assert_not_nil(get_form.fields.find { |f| f.name == "first_name" },
+      "First name field was nil"
+    )
+    assert_not_nil(
+    get_form.radiobuttons.find { |f| f.name == "gender" && f.value == "male"},
+      "Gender male button was nil"
+    )
+
+    assert_not_nil(
+    get_form.radiobuttons.find {|f| f.name == "gender" && f.value == "female"},
+      "Gender female button was nil"
+    )
+
+    assert_not_nil(get_form.checkboxes.find { |f| f.name == "cool person" },
+      "couldn't find cool person checkbox")
+    assert_not_nil(get_form.checkboxes.find { |f| f.name == "likes ham" },
+      "couldn't find likes ham checkbox")
+
+    # Now set all the fields
+    get_form.fields.find { |f| f.name == "first_name" }.value = "Aaron"
+    get_form.radiobuttons.find { |f| 
+      f.name == "gender" && f.value == "male" 
+    }.checked = true
+    get_form.checkboxes.find { |f| f.name == "likes ham" }.checked = true
+    page = agent.submit(get_form, get_form.buttons.first)
+
+    # Check that the submitted fields exist
+    assert_equal(3, page.links.size, "Not enough links")
+    assert_not_nil(
+      page.links.find { |l| l.text == "likes ham:on" },
+      "likes ham check box missing"
+    )
+    assert_not_nil(
+      page.links.find { |l| l.text == "first_name:Aaron" },
+      "first_name field missing"
+    )
+    assert_not_nil(
+      page.links.find { |l| l.text == "gender:male" },
+      "gender field missing"
+    )
+  end
+
+  def test_post_with_param_in_action
+    agent = WWW::Mechanize.new { |a| a.log = Logger.new(nil) }
+    page = agent.get("http://localhost:#{@port}/form_test.html")
+    post_form = page.forms.find { |f| f.name == "post_form3" }
+    assert_not_nil(post_form, "Post form is null")
+    assert_equal("post", post_form.method.downcase)
+    assert_equal("/form_post?great day=yes&one=two", post_form.action)
+    assert_equal(1, post_form.fields.size)
+    assert_equal(1, post_form.buttons.size)
+    assert_equal(2, post_form.radiobuttons.size)
+    assert_equal(2, post_form.checkboxes.size)
+    assert_not_nil(post_form.fields.find { |f| f.name == "first_name" },
+      "First name field was nil"
+    )
+    assert_not_nil(
+    post_form.radiobuttons.find { |f| f.name == "gender" && f.value == "male"},
+      "Gender male button was nil"
+    )
+
+    assert_not_nil(
+    post_form.radiobuttons.find {|f| f.name == "gender" && f.value == "female"},
+      "Gender female button was nil"
+    )
+
+    assert_not_nil(post_form.checkboxes.find { |f| f.name == "cool person" },
+      "couldn't find cool person checkbox")
+    assert_not_nil(post_form.checkboxes.find { |f| f.name == "likes ham" },
+      "couldn't find likes ham checkbox")
+
+    # Now set all the fields
+    post_form.fields.find { |f| f.name == "first_name" }.value = "Aaron"
+    post_form.radiobuttons.find { |f| 
+      f.name == "gender" && f.value == "male" 
+    }.checked = true
+    post_form.checkboxes.find { |f| f.name == "likes ham" }.checked = true
+    page = agent.submit(post_form, post_form.buttons.first)
+
+    # Check that the submitted fields exist
+    assert_equal(3, page.links.size, "Not enough links")
+    assert_not_nil(
+      page.links.find { |l| l.text == "likes ham:on" },
+      "likes ham check box missing"
+    )
+    assert_not_nil(
+      page.links.find { |l| l.text == "first_name:Aaron" },
+      "first_name field missing"
+    )
+    assert_not_nil(
+      page.links.find { |l| l.text == "gender:male" },
+      "gender field missing"
+    )
+  end
+
+  def test_get_with_param_in_action
+    agent = WWW::Mechanize.new { |a| a.log = Logger.new(nil) }
+    page = agent.get("http://localhost:#{@port}/form_test.html")
+    get_form = page.forms.find { |f| f.name == "get_form3" }
+    assert_not_nil(get_form, "Get form is null")
+    assert_equal("get", get_form.method.downcase)
+    assert_equal("/form_post?great day=yes&one=two", get_form.action)
+    assert_equal(1, get_form.fields.size)
+    assert_equal(1, get_form.buttons.size)
+    assert_equal(2, get_form.radiobuttons.size)
+    assert_equal(2, get_form.checkboxes.size)
+    assert_not_nil(get_form.fields.find { |f| f.name == "first_name" },
+      "First name field was nil"
+    )
+    assert_not_nil(
+    get_form.radiobuttons.find { |f| f.name == "gender" && f.value == "male"},
+      "Gender male button was nil"
+    )
+
+    assert_not_nil(
+    get_form.radiobuttons.find {|f| f.name == "gender" && f.value == "female"},
+      "Gender female button was nil"
+    )
+
+    assert_not_nil(get_form.checkboxes.find { |f| f.name == "cool person" },
+      "couldn't find cool person checkbox")
+    assert_not_nil(get_form.checkboxes.find { |f| f.name == "likes ham" },
+      "couldn't find likes ham checkbox")
+
+    # Now set all the fields
+    get_form.fields.find { |f| f.name == "first_name" }.value = "Aaron"
+    get_form.radiobuttons.find { |f| 
+      f.name == "gender" && f.value == "male" 
+    }.checked = true
+    get_form.checkboxes.find { |f| f.name == "likes ham" }.checked = true
+    page = agent.submit(get_form, get_form.buttons.first)
+
+    # Check that the submitted fields exist
+    assert_equal(5, page.links.size, "Not enough links")
+    assert_not_nil(
+      page.links.find { |l| l.text == "likes ham:on" },
+      "likes ham check box missing"
+    )
+    assert_not_nil(
+      page.links.find { |l| l.text == "first_name:Aaron" },
+      "first_name field missing"
+    )
+    assert_not_nil(
+      page.links.find { |l| l.text == "gender:male" },
+      "gender field missing"
+    )
+    assert_not_nil(
+      page.links.find { |l| l.text == "great day:yes" },
+      "great day field missing"
+    )
+    assert_not_nil(
+      page.links.find { |l| l.text == "one:two" },
+      "one field missing"
+    )
+  end
+end