@@ -26,12 +26,14 @@ module Trample end def login + pages_before = @pages.size if block_given? yield - @login = pages.pop + pages_added = @pages.size - pages_before + @login = pages.slice! pages_before, pages_added end - @login + @login ||= [] end def ==(other) lib/trample/configuration.rb @@ -12,7 +12,7 @@ module Trample end def trample - hit @config.login unless @config.login.nil? + @config.login.each {|page| hit page} unless @config.login.empty? @config.iterations.times do @config.pages.each do |p| hit p lib/trample/session.rb @@ -13,6 +13,7 @@ class ConfigurationTest < Test::Unit::TestCase post "http://google.com/", {:q => "something"} post "http://google.com/", &@params_proc login do + get "http://google.com" post "http://google.com/login" do { :q => "whatever" } end @@ -46,9 +47,12 @@ class ConfigurationTest < Test::Unit::TestCase assert_equal expected, @config.pages[2] end + should "support a login parameter which contains pages to hit for authentication" do + assert_equal ["http://google.com", "http://google.com/login"], @config.login.map{|page| page.url} + end + should "support a login parameter which contains a page to hit with params" do - assert_equal "http://google.com/login", @config.login.url - assert_equal({:q => "whatever"}, @config.login.parameters) + assert_equal({:q => "whatever"}, @config.login.last.parameters) end should "be equal if all the objects are the same" do test/configuration_test.rb 3ee9a8f07dc5ef26dc5adc66c6aea7b22a8f6194 Wes Gibbs wes@hashrocket.com http://github.com/wgibbs/trample/commit/ce0ea9f15dac042df3a10352bfeeb7e6f805c48a ce0ea9f15dac042df3a10352bfeeb7e6f805c48a 2009-10-03T11:46:56-07:00 2009-10-03T11:46:56-07:00 login supports multiple page requests Simulating a login on many apps may require more than a single POST. For instance, a GET might be required first to return a login form containing a hidden authenticity token that must be returned in the POST parameters for login to succeed. ef5580cc34dae8111e81e328aa2755e90ba18a9f Wes Gibbs wes@hashrocket.com