eric-mechanize 0.9.3.20090623142847

data/test/test_redirect_verb_handling.rb

@@ -0,0 +1,45 @@
+require File.expand_path(File.join(File.dirname(__FILE__), "helper"))
+
+class TestRedirectNotGetOrHead < Test::Unit::TestCase
+  include WWW
+
+  def setup
+    @agent = WWW::Mechanize.new
+  end
+
+  def test_to_s
+    page = MechTestHelper.fake_page(@agent)
+    error = Mechanize::RedirectNotGetOrHeadError.new(page, :put)
+    assert_match(/put/, error.to_s)
+  end
+
+  def test_head_redirect_results_in_head_request
+    page = @agent.head('http://localhost/redirect')
+    assert_equal(page.uri.to_s, 'http://localhost/verb')
+    assert_equal(page.body, "method: HEAD")
+  end
+
+  def test_get_redirect_results_in_get_request
+    page = @agent.get('http://localhost/redirect')
+    assert_equal(page.uri.to_s, 'http://localhost/verb')
+    assert_equal(page.body, "method: GET")
+  end
+
+  def test_post_redirect_results_in_get_request
+    page = @agent.post('http://localhost/redirect')
+    assert_equal(page.uri.to_s, 'http://localhost/verb')
+    assert_equal(page.body, "method: GET")
+  end
+
+  def test_put_redirect_results_in_get_request
+    page = @agent.put('http://localhost/redirect', 'foo')
+    assert_equal(page.uri.to_s, 'http://localhost/verb')
+    assert_equal(page.body, "method: GET")
+  end
+
+  def test_delete_redirect_results_in_get_request
+    page = @agent.delete('http://localhost/redirect')
+    assert_equal(page.uri.to_s, 'http://localhost/verb')
+    assert_equal(page.body, "method: GET")
+  end
+end

data/test/test_referer.rb

@@ -0,0 +1,39 @@
+require File.expand_path(File.join(File.dirname(__FILE__), "helper"))
+
+class RefererTest < Test::Unit::TestCase
+  def setup
+    @agent = WWW::Mechanize.new
+  end
+
+  def test_no_referer
+    page = @agent.get("http://localhost/referer")
+    assert_equal('', page.body)
+  end
+
+  def test_send_referer
+    page = @agent.get("http://localhost/tc_referer.html")
+    page = @agent.click page.links.first
+    assert_equal("http://localhost/tc_referer.html", page.body)
+  end
+
+  def test_fetch_two
+    page1 = @agent.get("http://localhost/tc_referer.html")
+    page2 = @agent.get("http://localhost/tc_pretty_print.html")
+    page = @agent.click page1.links.first
+    assert_equal("http://localhost/tc_referer.html", page.body)
+  end
+
+  def test_fetch_two_first
+    page1 = @agent.get("http://localhost/tc_referer.html")
+    page2 = @agent.get("http://localhost/tc_pretty_print.html")
+    page = @agent.click page1.links.first
+    assert_equal("http://localhost/tc_referer.html", page.body)
+  end
+
+  def test_post_form
+    page1 = @agent.get("http://localhost/tc_referer.html")
+    page2 = @agent.get("http://localhost/tc_pretty_print.html")
+    page = @agent.submit page1.forms.first
+    assert_equal("http://localhost/tc_referer.html", page.body)
+  end
+end

data/test/test_relative_links.rb

@@ -0,0 +1,40 @@
+require File.expand_path(File.join(File.dirname(__FILE__), "helper"))
+
+class TestRelativeLinks < Test::Unit::TestCase
+  def setup
+    @agent = WWW::Mechanize.new
+  end
+
+  def test_dot_dot_slash
+    @page = @agent.get("http://localhost/relative/tc_relative_links.html")
+    page = @page.links.first.click
+    assert_equal('http://localhost/tc_relative_links.html', @agent.current_page.uri.to_s)
+  end
+
+  def test_too_many_dots
+    @page = @agent.get("http://localhost/relative/tc_relative_links.html")
+    page = @page.link_with(:text => 'too many dots').click
+    assert_not_nil(page)
+    assert_equal('http://localhost/tc_relative_links.html', page.uri.to_s)
+  end
+
+  def test_go_forward
+    @page = @agent.get("http://localhost/tc_relative_links.html")
+    @page = @page.links.first.click
+    assert_equal('http://localhost/relative/tc_relative_links.html', @agent.current_page.uri.to_s)
+  end
+
+  def test_frame_dot_dot_slash
+    @page = @agent.get("http://localhost/relative/tc_relative_links.html")
+    page = @agent.click(@page.frame_with(:text => 'frame1'))
+    assert_equal('http://localhost/tc_relative_links.html', @agent.current_page.uri.to_s)
+  end
+
+  def test_frame_forward_back_forward
+    @page = @agent.get("http://localhost/tc_relative_links.html")
+    page1 = @agent.click @page.frame_with(:name => 'frame1')
+    assert_equal('http://localhost/relative/tc_relative_links.html', @agent.current_page.uri.to_s)
+    page2 = @agent.click @page.frame_with(:name => 'frame2')
+    assert_equal('http://localhost/relative/tc_relative_links.html', @agent.current_page.uri.to_s)
+  end
+end

data/test/test_request.rb

@@ -0,0 +1,13 @@
+require File.expand_path(File.join(File.dirname(__FILE__), "helper"))
+
+class RequestTest < Test::Unit::TestCase
+  def setup
+    @agent = WWW::Mechanize.new
+  end
+
+  def test_uri_is_not_polluted
+    uri = URI.parse('http://localhost/')
+    @agent.get(uri, {'q' => 'Ruby'})
+    assert_equal 'http://localhost/', uri.to_s
+  end
+end

data/test/test_response_code.rb

@@ -0,0 +1,52 @@
+require File.expand_path(File.join(File.dirname(__FILE__), "helper"))
+
+class ResponseCodeMechTest < Test::Unit::TestCase
+  def setup
+    @agent = WWW::Mechanize.new
+  end
+
+  def test_eof_error_loop
+    assert_raises(EOFError) {
+      @agent.get("http://localhost/http_headers?Content-Length=300")
+    }
+  end
+
+  def test_redirect
+    @agent.get("http://localhost/response_code?code=300")
+    assert_equal("http://localhost/index.html",
+      @agent.current_page.uri.to_s)
+
+    @agent.get("http://localhost/response_code?code=301")
+    assert_equal("http://localhost/index.html",
+      @agent.current_page.uri.to_s)
+
+    @agent.get("http://localhost/response_code?code=302")
+    assert_equal("http://localhost/index.html",
+      @agent.current_page.uri.to_s)
+
+    @agent.get("http://localhost/response_code?code=303")
+    assert_equal("http://localhost/index.html",
+      @agent.current_page.uri.to_s)
+
+    @agent.get("http://localhost/response_code?code=307")
+    assert_equal("http://localhost/index.html",
+      @agent.current_page.uri.to_s)
+  end
+
+  def test_do_not_follow_redirect
+    @agent.redirect_ok = false
+
+    @agent.get("http://localhost/response_code?code=302")
+    assert_equal("http://localhost/response_code?code=302",
+      @agent.current_page.uri.to_s)
+  end
+
+  def test_error
+    @agent = WWW::Mechanize.new
+    begin
+      @agent.get("http://localhost/response_code?code=500")
+    rescue WWW::Mechanize::ResponseCodeError => err
+      assert_equal("500", err.response_code)
+    end
+  end
+end

data/test/test_save_file.rb

@@ -0,0 +1,48 @@
+require File.expand_path(File.join(File.dirname(__FILE__), "helper"))
+
+class TestSaveFile < Test::Unit::TestCase
+  def setup
+    @agent = WWW::Mechanize.new
+  end
+
+  def test_save_file
+    page = @agent.get('http://localhost:2000/form_no_action.html')
+    length = page.response['Content-Length']
+    page.save_as("test.html")
+    file_length = nil
+    File.open("test.html", "r") { |f| file_length = f.read.length }
+    FileUtils.rm("test.html")
+    assert_equal(length.to_i, file_length)
+  end
+
+  def test_save_file_default
+    page = WWW::Mechanize::File.new(
+                                    URI.parse('http://localhost/test.html'),
+                                    {},
+                                    "hello"
+                                   )
+    page.save
+    assert(File.exists?('test.html'))
+    page.save
+    assert(File.exists?('test.html.1'))
+    page.save
+    assert(File.exists?('test.html.2'))
+    FileUtils.rm("test.html")
+    FileUtils.rm("test.html.1")
+    FileUtils.rm("test.html.2")
+  end
+
+  def test_save_file_default_with_dots
+    page = WWW::Mechanize::File.new(
+                                    URI.parse('http://localhost/../test.html'),
+                                    {},
+                                    "hello"
+                                   )
+    page.save
+    assert(File.exists?('test.html'))
+    page.save
+    assert(File.exists?('test.html.1'))
+    FileUtils.rm("test.html")
+    FileUtils.rm("test.html.1")
+  end
+end

data/test/test_scheme.rb

@@ -0,0 +1,48 @@
+require File.expand_path(File.join(File.dirname(__FILE__), "helper"))
+
+class SchemeTest < Test::Unit::TestCase
+  def setup
+    @agent = WWW::Mechanize.new
+    @agent.log = Class.new(Object) do
+      def method_missing(*args)
+      end
+    end.new
+  end
+
+  def test_file_scheme
+    f = File.expand_path(__FILE__)
+    page = @agent.get("file://#{f}")
+    assert_equal(File.read(f), page.body)
+  end
+
+  def test_file_scheme_parses_html
+    f = File.expand_path(
+               File.join(File.dirname(__FILE__), "htdocs", 'google.html'))
+    page = @agent.get("file://#{f}")
+    assert_equal(File.read(f), page.body)
+    assert_kind_of(WWW::Mechanize::Page, page)
+  end
+
+  def test_file_scheme_supports_directories
+    f = File.expand_path(File.join(File.dirname(__FILE__), "htdocs"))
+    page = @agent.get("file://#{f}")
+    assert_equal(Dir[File.join(f, '*')].length, page.links.length)
+    assert_kind_of(WWW::Mechanize::Page, page)
+  end
+
+  def test_click_file_link
+    f = File.expand_path(File.join(File.dirname(__FILE__), "htdocs"))
+    page = @agent.get("file://#{f}")
+    link = page.links.find { |l| l.text =~ /tc_follow_meta/ }
+    assert_not_nil(link)
+    path = URI.parse(link.href).path
+    
+    page = link.click
+    assert_equal(File.read(path), page.body)
+
+    link = page.meta.first
+    assert_not_nil(link)
+    page = @agent.click(link)
+    assert_equal("http://localhost/index.html", @agent.history.last.uri.to_s)
+  end
+end

data/test/test_select.rb

@@ -0,0 +1,106 @@
+require File.expand_path(File.join(File.dirname(__FILE__), "helper"))
+
+class SelectTest < Test::Unit::TestCase
+  def setup
+    @agent = WWW::Mechanize.new
+    @page = @agent.get("http://localhost/form_select.html")
+    @form = @page.forms.first
+  end
+
+  def test_select_none
+    @form.fields_with(:name => 'list').first.select_none
+    assert_equal('1', @form.list)
+  end
+
+  def test_select_all
+    @form.fields_with(:name => 'list').first.select_all
+    assert_equal('6', @form.list)
+  end
+
+  def test_correct_class
+    assert_instance_of(WWW::Mechanize::Form::SelectList,
+      @form.field_with(:name => 'list'))
+  end
+
+  def test_click_all
+    @form.field_with(:name => 'list').options.each { |o| o.click }
+    option_list = @form.field_with(:name => 'list').options
+    assert_not_nil(option_list)
+    assert_equal(6, option_list.length)
+    assert_equal(option_list.last.value, @form.list)
+    page = @agent.submit(@form)
+    assert_equal(1, page.links.length)
+    assert_equal(1, page.links_with(:text => "list:#{option_list.last}").length)
+  end
+
+  def test_select_default
+    assert_equal("2", @form.list)
+    page = @agent.submit(@form)
+    assert_equal(1, page.links.length)
+    assert_equal(1, page.links_with(:text => 'list:2').length)
+  end
+
+  def test_select_one
+    @form.list = 'Aaron'
+    assert_equal('Aaron', @form.list)
+    page = @agent.submit(@form)
+    assert_equal(1, page.links.length)
+    assert_equal('list:Aaron', page.links.first.text)
+  end
+
+  def test_select_two
+    @form.list = ['1', 'Aaron']
+    assert_equal('1', @form.list)
+    page = @agent.submit(@form)
+    assert_equal(1, page.links.length)
+    assert_equal(1, page.links_with(:text => 'list:1').length)
+  end
+
+  def test_select_three
+    @form.list = ['1', '2', '3']
+    assert_equal('1', @form.list)
+    page = @agent.submit(@form)
+    assert_equal(1, page.links.length)
+    assert_equal(1, page.links_with(:text => 'list:1').length)
+  end
+
+  def test_select_three_twice
+    @form.list = ['1', '2', '3']
+    @form.list = ['1', '2', '3']
+    assert_equal('1', @form.list)
+    page = @agent.submit(@form)
+    assert_equal(1, page.links.length)
+    assert_equal(1, page.links_with(:text => 'list:1').length)
+  end
+
+  def test_select_with_empty_value
+    list = @form.field_with(:name => 'list')
+    list.options.last.instance_variable_set(:@value, '')
+    list.options.last.tick
+    page = @agent.submit(@form)
+    assert_equal(1, page.links.length)
+    assert_equal(1, page.links_with(:text => 'list:').length)
+  end
+
+  def test_select_with_click
+    @form.list = ['1', 'Aaron']
+    @form.field_with(:name => 'list').options[3].tick
+    assert_equal('4', @form.list)
+    page = @agent.submit(@form)
+    assert_equal(1, page.links.length)
+    assert_equal(1, page.links_with(:text => 'list:4').length)
+  end
+
+  def test_click_twice
+    list = @form.field_with(:name => 'list')
+    list.options[0].click
+    assert_equal('1', @form.list)
+    list.options[1].click
+    assert_equal('2', @form.list)
+    list.options.last.click
+    assert_equal('6', @form.list)
+    assert_equal(1, list.selected_options.length)
+    list.select_all
+    assert_equal(1, list.selected_options.length)
+  end
+end

data/test/test_select_all.rb

@@ -0,0 +1,15 @@
+require File.expand_path(File.join(File.dirname(__FILE__), "helper"))
+class SelectAllTest < Test::Unit::TestCase
+  def setup
+    @agent = WWW::Mechanize.new
+    @page = @agent.get("http://localhost/form_select_all.html")
+    @form = @page.forms.first
+  end
+
+  def test_select_default
+    assert_equal("6", @form.list)
+    page = @agent.submit(@form)
+    assert_equal(1, page.links.length)
+    assert_equal(1, page.links_with(:text => 'list:6').length)
+  end
+end

data/test/test_select_none.rb

@@ -0,0 +1,15 @@
+require File.expand_path(File.join(File.dirname(__FILE__), "helper"))
+class SelectNoneTest < Test::Unit::TestCase
+  def setup
+    @agent = WWW::Mechanize.new
+    @page = @agent.get("http://localhost/form_select_none.html")
+    @form = @page.forms.first
+  end
+
+  def test_select_default
+    assert_equal("1", @form.list)
+    page = @agent.submit(@form)
+    assert_equal(1, page.links.length)
+    assert_equal(1, page.links_with(:text => 'list:1').length)
+  end
+end

data/test/test_select_noopts.rb

@@ -0,0 +1,16 @@
+require File.expand_path(File.join(File.dirname(__FILE__), "helper"))
+
+class SelectNoOptionsTest < Test::Unit::TestCase
+  def setup
+    @agent = WWW::Mechanize.new
+    @page = @agent.get("http://localhost/form_select_noopts.html")
+    @form = @page.forms.first
+  end
+
+  def test_select_default
+    assert_not_nil(@form.field('list'))
+    assert_nil(@form.list)
+    page = @agent.submit(@form)
+    assert_equal(0, page.links.length)
+  end
+end