tenderlove-mechanize 0.9.3.20090617085936

data/test/test_cookies.rb

@@ -0,0 +1,123 @@
+require File.expand_path(File.join(File.dirname(__FILE__), "helper"))
+
+class CookiesMechTest < Test::Unit::TestCase
+  def setup
+    @agent = WWW::Mechanize.new
+  end
+
+  def test_meta_tag_cookies
+    cookies = @agent.cookies.length
+    page = @agent.get("http://localhost/meta_cookie.html")
+    assert_equal(cookies + 1, @agent.cookies.length)
+  end
+
+  def test_send_cookies
+    page = @agent.get("http://localhost/many_cookies")
+    page = @agent.get("http://localhost/send_cookies")
+    assert_equal(3, 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_no_space_cookies
+    page = @agent.get("http://localhost/one_cookie_no_space")
+    assert_equal(1, @agent.cookies.length)
+    foo_cookie = @agent.cookies.find { |k| k.name == 'foo' }
+    assert_not_nil(foo_cookie, 'Foo cookie was nil')
+    assert_equal('bar', foo_cookie.value)
+    assert_equal('/', foo_cookie.path)
+    assert_equal(true, Time.now < foo_cookie.expires)
+  end
+
+  def test_many_cookies_as_string
+    page = @agent.get("http://localhost/many_cookies_as_string")
+    assert_equal(4, @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, Time.now < name_cookie.expires)
+
+    expired_cookie = @agent.cookies.find { |k| k.name == "expired" }
+    assert_nil(expired_cookie, "Expired cookie was not nil")
+
+    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, Time.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("/", no_path_cookie.path)
+    assert_equal(true, Time.now < no_path_cookie.expires)
+  end
+
+  def test_many_cookies
+    page = @agent.get("http://localhost/many_cookies")
+    assert_equal(4, @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, Time.now < name_cookie.expires)
+
+    expired_cookie = @agent.cookies.find { |k| k.name == "expired" }
+    assert_nil(expired_cookie, "Expired cookie was not nil")
+
+    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, Time.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("/", no_path_cookie.path)
+    assert_equal(true, Time.now < no_path_cookie.expires)
+  end
+
+  def test_get_cookie
+    assert_equal(true,
+      @agent.cookie_jar.empty?(
+      URI::parse("http://localhost/one_cookie")))
+
+    assert_equal(0, @agent.cookies.length)
+
+    page = @agent.get("http://localhost/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/one_cookie")))
+    page = @agent.get("http://localhost/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/test_encoded_links.rb

@@ -0,0 +1,20 @@
+require File.expand_path(File.join(File.dirname(__FILE__), "helper"))
+
+class TestEncodedLinks < Test::Unit::TestCase
+  def setup
+    @agent = WWW::Mechanize.new
+    @page = @agent.get("http://localhost/tc_encoded_links.html")
+  end
+
+  def test_click_link
+    link = @page.links.first
+    assert_equal('/form_post?a=b&b=c', link.href)
+    page = @agent.click(link)
+    assert_equal("http://localhost/form_post?a=b&b=c", page.uri.to_s)
+  end
+
+  def test_hpricot_link
+    page = @agent.click(@page.search('a').first)
+    assert_equal("http://localhost/form_post?a=b&b=c", page.uri.to_s)
+  end
+end

data/test/test_errors.rb

@@ -0,0 +1,49 @@
+require File.expand_path(File.join(File.dirname(__FILE__), "helper"))
+
+class MechErrorsTest < Test::Unit::TestCase
+  def setup
+    @agent = WWW::Mechanize.new
+  end
+
+  def test_bad_form_method
+    page = @agent.get("http://localhost/bad_form_test.html")
+    assert_raise(RuntimeError) {
+      @agent.submit(page.forms.first)
+    }
+  end
+
+  def test_non_exist
+    begin
+      page = @agent.get("http://localhost/bad_form_test.html")
+    rescue RuntimeError => ex
+      assert_equal("404", ex.inspect)
+    end
+  end
+
+  def test_too_many_radio
+    page = @agent.get("http://localhost/form_test.html")
+    form = page.form_with(:name => 'post_form1')
+    form.radiobuttons.each { |r| r.checked = true }
+    assert_raise(RuntimeError) {
+      @agent.submit(form)
+    }
+  end
+
+  def test_unknown_agent
+    assert_raise(RuntimeError) {
+      @agent.user_agent_alias = "Aaron's Browser"
+    }
+  end
+
+  def test_bad_url
+    assert_raise(RuntimeError) {
+      @agent.get('/foo.html')
+    }
+  end
+
+  def test_unsupported_scheme
+    assert_raise(WWW::Mechanize::UnsupportedSchemeError) {
+      @agent.get('ftp://server.com/foo.html')
+    }
+  end
+end

data/test/test_follow_meta.rb

@@ -0,0 +1,108 @@
+require File.expand_path(File.join(File.dirname(__FILE__), 'helper'))
+
+class FollowMetaTest < Test::Unit::TestCase
+  def setup
+    @agent = WWW::Mechanize.new
+  end
+
+  def test_dont_follow_meta_by_default
+    page = @agent.get('http://localhost/tc_follow_meta.html')
+    assert_equal('http://localhost/tc_follow_meta.html', page.uri.to_s)
+    assert_equal(1, page.meta.length)
+  end
+
+  def test_meta_refresh_does_not_send_referer
+    @agent.follow_meta_refresh = true
+    requests = []
+    @agent.pre_connect_hooks << lambda { |params|
+      requests << params[:request]
+    }
+
+    page = @agent.get('http://localhost/tc_follow_meta.html')
+    assert_nil requests[1]['referer']
+  end
+
+  def test_follow_meta_if_set
+    @agent.follow_meta_refresh = true
+
+    page = @agent.get('http://localhost/tc_follow_meta.html')
+
+    assert_equal(2, @agent.history.length)
+    assert_equal('http://localhost/tc_follow_meta.html',
+                 @agent.history[0].uri.to_s)
+    assert_equal('http://localhost/index.html', page.uri.to_s)
+    assert_equal('http://localhost/index.html', @agent.history.last.uri.to_s)
+  end
+  
+  def test_follow_meta_with_empty_url
+    @agent.follow_meta_refresh = true
+  
+    page = @agent.get('http://localhost/refresh_with_empty_url')
+  
+    assert_equal(3, @agent.history.length)
+    assert_equal('http://localhost/refresh_with_empty_url',
+                 @agent.history[0].uri.to_s)
+    assert_equal('http://localhost/refresh_with_empty_url',
+                 @agent.history[1].uri.to_s)
+    assert_equal('http://localhost/index.html', page.uri.to_s)
+    assert_equal('http://localhost/index.html', @agent.history.last.uri.to_s)
+  end
+  
+  def test_follow_meta_without_url
+    @agent.follow_meta_refresh = true
+  
+    page = @agent.get('http://localhost/refresh_without_url')
+  
+    assert_equal(3, @agent.history.length)
+    assert_equal('http://localhost/refresh_without_url',
+                 @agent.history[0].uri.to_s)
+    assert_equal('http://localhost/refresh_without_url',
+                 @agent.history[1].uri.to_s)
+    assert_equal('http://localhost/index.html', page.uri.to_s)
+    assert_equal('http://localhost/index.html', @agent.history.last.uri.to_s)
+  end
+
+  def test_always_follow_302
+    @agent.follow_meta_refresh = false
+    page = @agent.get('http://localhost/response_code?code=302&ct=test/xml')
+    assert_equal('http://localhost/index.html', page.uri.to_s)
+    assert_equal(2, @agent.history.length)
+  end
+
+  def test_infinite_refresh_throws_exception
+    @agent.follow_meta_refresh = true
+    assert_raises(WWW::Mechanize::RedirectLimitReachedError) {
+      begin
+        @agent.get('http://localhost/infinite_refresh')
+      rescue WWW::Mechanize::RedirectLimitReachedError => ex
+        raise ex
+      end
+    }
+  end
+  
+  def test_dont_honor_http_refresh_by_default
+    page = @agent.get('http://localhost/http_refresh?refresh_time=0')
+    assert_equal('http://localhost/http_refresh?refresh_time=0', page.uri.to_s)
+  end
+  
+  def test_honor_http_refresh_if_set
+    @agent.follow_meta_refresh = true
+    page = @agent.get('http://localhost/http_refresh?refresh_time=0')
+    assert_equal('http://localhost/index.html', page.uri.to_s)
+    assert_equal(2, @agent.history.length)
+  end
+
+  def test_honor_http_refresh_delay_if_set
+    @agent.follow_meta_refresh = true
+    class << @agent
+      attr_accessor :slept
+      def sleep *args
+        @slept = args
+      end
+    end
+
+    page = @agent.get('http://localhost/http_refresh?refresh_time=1')
+    assert_equal [1], @agent.slept
+  end
+
+end

data/test/test_form_action.rb

@@ -0,0 +1,44 @@
+require File.expand_path(File.join(File.dirname(__FILE__), "helper"))
+
+class TestFormAction < Test::Unit::TestCase
+  def setup
+    @agent = WWW::Mechanize.new
+    @page  = @agent.get("http://localhost/tc_form_action.html")
+  end
+
+  def test_post_encoded_action
+    form = @page.form(:name => 'post_form1') { |f|
+      f.first_name = "Aaron"
+    }
+    assert_equal('/form_post?a=b&b=c', form.action)
+    page = form.submit
+    assert_equal("http://localhost/form_post?a=b&b=c", page.uri.to_s)
+  end
+
+  def test_get_encoded_action
+    form = @page.form('post_form2') { |f|
+      f.first_name = "Aaron"
+    }
+    assert_equal('/form_post?a=b&b=c', form.action)
+    page = form.submit
+    assert_equal("http://localhost/form_post?first_name=Aaron", page.uri.to_s)
+  end
+
+  def test_post_nonencoded_action
+    form = @page.form('post_form3') { |f|
+      f.first_name = "Aaron"
+    }
+    assert_equal('/form_post?a=b&b=c', form.action)
+    page = form.submit
+    assert_equal("http://localhost/form_post?a=b&b=c", page.uri.to_s)
+  end
+
+  def test_post_pound_sign
+    form = @page.form('post_form4') { |f|
+      f.first_name = "Aaron"
+    }
+    assert_equal('/form_post#1', form.action)
+    page = form.submit
+    assert_equal("http://localhost/form_post#1", page.uri.to_s)
+  end
+end

data/test/test_form_as_hash.rb

@@ -0,0 +1,61 @@
+require File.expand_path(File.join(File.dirname(__FILE__), "helper"))
+
+class TestFormHash < Test::Unit::TestCase
+  def setup
+    @agent = WWW::Mechanize.new
+    @page  = @agent.get('http://localhost/form_multival.html')
+  end
+
+  def test_form_hash
+    form = @page.form_with(:name => 'post_form')
+
+    assert_not_nil(form)
+    field_length = form.fields.length
+    assert_nil(form['intarweb'])
+    form['intarweb'] = 'Aaron'
+
+    assert_not_nil(form['intarweb'])
+    assert_equal(field_length + 1, form.fields.length)
+  end
+
+  def test_add_field_via_hash
+    form = @page.form_with(:name => 'post_form')
+
+    assert_not_nil(form)
+    field_length = form.fields.length
+    assert_nil(form['intarweb'])
+    form['intarweb'] = 'Aaron'
+
+    assert_not_nil(form['intarweb'])
+    assert_equal(field_length + 1, form.fields.length)
+  end
+
+  def test_fields_as_hash
+    form = @page.form_with(:name => 'post_form')
+
+    assert_not_nil(form)
+    assert_equal(2, form.fields_with(:name => 'first').length)
+
+    form['first'] = 'Aaron'
+    assert_equal('Aaron', form['first'])
+    assert_equal('Aaron', form.field_with(:name => 'first').value)
+  end
+
+  def test_keys
+    @page = @agent.get('http://localhost/empty_form.html')
+    form = @page.forms.first
+
+    assert_not_nil(form)
+    assert_equal(false, form.has_field?('name'))
+    assert_equal(false, form.has_value?('Aaron'))
+    assert_equal(0, form.keys.length)
+    assert_equal(0, form.values.length)
+    form['name'] = 'Aaron'
+    assert_equal(true, form.has_field?('name'))
+    assert_equal(true, form.has_value?('Aaron'))
+    assert_equal(1, form.keys.length)
+    assert_equal(['name'], form.keys)
+    assert_equal(1, form.values.length)
+    assert_equal(['Aaron'], form.values)
+  end
+end

data/test/test_form_button.rb

@@ -0,0 +1,38 @@
+require File.expand_path(File.join(File.dirname(__FILE__), "helper"))
+
+class TestFormButtons < Test::Unit::TestCase
+  def setup
+    @agent = WWW::Mechanize.new
+  end
+
+  def test_submit_input_tag
+    assert_form_contains_button('<input type="submit" value="submit" />')
+  end
+
+  def test_button_input_tag
+    assert_form_contains_button('<input type="button" value="submit" />')
+  end
+
+  def test_submit_button_tag
+    assert_form_contains_button('<button type="submit" value="submit"/>')
+  end
+
+  def test_button_button_tag
+    assert_form_contains_button('<button type="button" value="submit"/>')
+  end
+
+  def assert_form_contains_button(button)
+    page = WWW::Mechanize::Page.new(nil, html_response, html(button), 200, @agent)
+    assert_equal(1, page.forms.length)
+    assert_equal(1, page.forms.first.buttons.length)
+  end
+
+  def html(input)
+    "<html><body><form>#{input}</form></body></html>"
+  end
+
+  def html_response
+    { 'content-type' => 'text/html' }
+  end
+end
+

data/test/test_form_no_inputname.rb

@@ -0,0 +1,15 @@
+require File.expand_path(File.join(File.dirname(__FILE__), "helper"))
+
+class FormNoInputNameTest < Test::Unit::TestCase
+  def setup
+    @agent = WWW::Mechanize.new
+    @page = @agent.get('http://localhost/form_no_input_name.html')
+  end
+
+  def test_no_input_name
+    form = @page.forms.first
+    assert_equal(0, form.fields.length)
+    assert_equal(0, form.radiobuttons.length)
+    assert_equal(0, form.checkboxes.length)
+  end
+end