mechanize-ntlm 0.9.1

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

data/test/test_set_fields.rb

@@ -0,0 +1,44 @@
+require File.expand_path(File.join(File.dirname(__FILE__), "helper"))
+
+class TestSetFields < Test::Unit::TestCase
+  def setup
+    @agent = WWW::Mechanize.new
+    @page = @agent.get("http://localhost/form_set_fields.html")
+    @form = @page.forms.first
+  end
+
+  def test_set_no_fields
+    before = {}
+    @form.fields.each { |f| before[f.name] = f.value }
+    @form.set_fields
+    before.each { |k,v| assert_equal(v, @form.send(k)) }
+  end
+
+  def test_set_one_field
+    @form.set_fields( :gender => 'male' )
+    assert_equal('male', @form.gender)
+  end
+
+  def test_set_many_fields
+    @form.set_fields( :gender       => 'male',
+                      :first_name   => 'Aaron',
+                      'green[eggs]' => 'Ham'
+                    )
+    assert_equal('male', @form.gender)
+    assert_equal('Aaron', @form.first_name)
+    assert_equal('Ham', @form.fields_with(:name => 'green[eggs]').first.value)
+  end
+
+  def test_set_multiple_duplicate_fields
+    page = @agent.get("http://localhost/form_multival.html")
+    form = page.form('post_form')
+    form.set_fields( :first => { 0 => 'a', 1 => 'b' } )
+    assert_equal('a', form.fields_with(:name => 'first')[0].value)
+    assert_equal('b', form.fields_with(:name => 'first')[1].value)
+  end
+
+  def test_set_second_field
+    @form.set_fields( :first_name => ['Aaron', 1] )
+    assert_equal('Aaron', @form.fields_with(:name => 'first_name')[1].value)
+  end
+end

data/test/test_ssl_server.rb

@@ -0,0 +1,20 @@
+require File.expand_path(File.join(File.dirname(__FILE__), "helper"))
+
+class SSLServerTest < Test::Unit::TestCase
+  def setup
+    @agent = WWW::Mechanize.new
+  end
+
+  def test_ssl_request
+    non_ssl_page = @agent.get("http://localhost/form_test.html")
+    ssl_page = @agent.get("https://localhost/form_test.html")
+    assert_equal(non_ssl_page.body.length, ssl_page.body.length)
+  end
+
+  def test_ssl_request_verify
+    non_ssl_page = @agent.get("http://localhost/form_test.html")
+    @agent.ca_file = 'data/server.crt'
+    ssl_page = @agent.get("https://localhost/form_test.html")
+    assert_equal(non_ssl_page.body.length, ssl_page.body.length)
+  end
+end

data/test/test_subclass.rb

@@ -0,0 +1,14 @@
+require File.expand_path(File.join(File.dirname(__FILE__), "helper"))
+
+class TestSubclass < Test::Unit::TestCase
+  def setup
+    @agent = WWW::Mechanize.new
+  end
+
+  def test_send_cookie
+    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

data/test/test_textarea.rb

@@ -0,0 +1,45 @@
+require File.expand_path(File.join(File.dirname(__FILE__), "helper"))
+
+class TestTextArea < Test::Unit::TestCase
+  def setup
+    @agent = WWW::Mechanize.new
+    @page  = @agent.get("http://localhost/tc_textarea.html")
+  end
+
+  def test_empty_text_area
+    form = @page.forms_with(:name => 'form1').first
+    assert_equal('', form.field_with(:name => 'text1').value)
+    form.text1 = 'Hello World'
+    assert_equal('Hello World', form.field_with(:name => 'text1').value)
+    page = @agent.submit(form)
+    assert_equal(1, page.links.length)
+    assert_equal('text1:Hello World', page.links[0].text)
+  end
+
+  def test_non_empty_textfield
+    form = @page.forms_with(:name => 'form2').first
+    assert_equal('sample text', form.field_with(:name => 'text1').value)
+    page = @agent.submit(form)
+    assert_equal(1, page.links.length)
+    assert_equal('text1:sample text', page.links[0].text)
+  end
+
+  def test_multi_textfield
+    form = @page.form_with(:name => 'form3')
+    assert_equal(2, form.fields_with(:name => 'text1').length)
+    assert_equal('', form.fields_with(:name => 'text1')[0].value)
+    assert_equal('sample text', form.fields_with(:name => 'text1')[1].value)
+    form.text1 = 'Hello World'
+    assert_equal('Hello World', form.fields_with(:name => 'text1')[0].value)
+    assert_equal('sample text', form.fields_with(:name => 'text1')[1].value)
+    page = @agent.submit(form)
+    assert_equal(2, page.links.length)
+    link = page.links_with(:text => 'text1:sample text')
+    assert_not_nil(link)
+    assert_equal(1, link.length)
+
+    link = page.links_with(:text => 'text1:Hello World')
+    assert_not_nil(link)
+    assert_equal(1, link.length)
+  end
+end