# This is a wrapper of assert_select for rspec.

module Spec # :nodoc:
  module Rails
    module Matchers
      
      class AssertSelect #:nodoc:
        
        def initialize(*args, &block)
          @assertion = args.shift
          @spec_scope = args.shift
          @args = args
          @block = block
        end
        
        def matches?(response_or_text, &block)
          @args.unshift(HTML::Document.new(response_or_text).root) if String === response_or_text
          @block = block if block
          begin
            @spec_scope.send(@assertion, *@args, &@block)
          rescue Exception => @error
          end
          @error.nil?
        end
        
        def failure_message; @error.message; end
        def negative_failure_message; "should not #{description}, but did"; end
        
        def description
          {
            :assert_select => "have tag#{format_args(*@args)}",
            :assert_select_email => "send email#{format_args(*@args)}",
          }[@assertion]
        end
        
      private

        def format_args(*args)
          return "" if args.empty?
          return "(#{arg_list(*args)})"
        end

        def arg_list(*args)
          args.collect do |arg|
            arg.respond_to?(:description) ? arg.description : arg.inspect
          end.join(", ")
        end
        
      end
      
      # :call-seq:
      #   response.should have_tag(*args, &block)
      #   string.should have_tag(*args, &block)
      #
      # wrapper for assert_select with additional support for using
      # css selectors to set expectation on Strings. Use this in
      # helper specs, for example, to set expectations on the results
      # of helper methods.
      #
      # == Examples
      #
      #   # in a controller spec
      #   response.should have_tag("div", "some text")
      #
      #   # in a helper spec (person_address_tag is a method in the helper)
      #   person_address_tag.should have_tag("input#person_address")
      #
      # see documentation for assert_select at http://api.rubyonrails.org/
      def have_tag(*args, &block)
        args.unshift(self)
        args.unshift(:assert_select)
        AssertSelect.new(*args, &block)
      end
    
      # wrapper for a nested assert_select
      #
      #   response.should have_tag("div#form") do
      #     with_tag("input#person_name[name=?]", "person[name]")
      #   end
      #
      # see documentation for assert_select at http://api.rubyonrails.org/
      def with_tag(*args, &block)
        response.should have_tag(*args, &block)
      end
    
      # wrapper for a nested assert_select with false
      #
      #   response.should have_tag("div#1") do
      #     without_tag("span", "some text that shouldn't be there")
      #   end
      #
      # see documentation for assert_select at http://api.rubyonrails.org/
      def without_tag(*args, &block)
        should_not have_tag(*args, &block)
      end
    
      # :call-seq:
      #   response.should have_rjs(*args, &block)
      #
      # wrapper for assert_select_rjs
      #
      # see documentation for assert_select_rjs at http://api.rubyonrails.org/
      def have_rjs(*args, &block)
        args.unshift(self)
        args.unshift(:assert_select_rjs)
        AssertSelect.new(*args, &block)
      end
      
      # :call-seq:
      #   response.should send_email(*args, &block)
      #
      # wrapper for assert_select_email
      #
      # see documentation for assert_select_email at http://api.rubyonrails.org/
      def send_email(*args, &block)
        args.unshift(self)
        args.unshift(:assert_select_email)
        AssertSelect.new(*args, &block)
      end
      
      # wrapper for assert_select_encoded
      #
      # see documentation for assert_select_encoded at http://api.rubyonrails.org/
      def with_encoded(*args, &block)
        args.unshift(self)
        args.unshift(:assert_select_encoded)
        should AssertSelect.new(*args, &block)
      end
    end
  end
end