require 'hpricot/forms'

module HpricotFormsTestHelpers
    def forms(condition = nil)
        elements("//form#{condition.nil? ? '' : '[' + condition + ']'}", Hpricot::TestForm)
    end
    
    def links(condition = nil)
        elements("//a#{condition.nil? ? '' : '[' + condition + ']'}")
    end
    
    def elements(hsearch, klass = Hpricot::TestField)
        return Hpricot(body).search(hsearch).collect { |x| klass.new(x) }
    end
    
    def as_partial_form
        Hpricot::TestForm.new(Hpricot(body))
    end
end

module ActionController
    class TestResponse < AbstractResponse
        include HpricotFormsTestHelpers
    end
    
    class CgiResponse
        include HpricotFormsTestHelpers
    end
end