macros.rb

require 'json'

module Chicago
  module Riot
    module Macros
      def asserts_response_status(expected)
        asserts("response status is #{expected}") { last_response.status }.equals(expected)
      end
      
      def asserts_content_type(expected)
        asserts("content type is #{expected}") { last_response.headers['Content-type'] }.equals(expected)
      end

      def asserts_response_body(expected)
        asserts("response body matches #{expected.inspect}") { last_response.body }.matches(expected)
      end

      def asserts_location(expected_path)
        asserts("location matches #{expected_path}") do
          last_response.headers["Location"]
        end.matches(expected_path)
      end

      def asserts_json_response(json)
        asserts_content_type 'application/json'

        json = json.to_json unless json.instance_of?(String)
        json

        asserts("response body has JSON") do
          last_response.body
        end.equals(json)
        # Calling situation is kind of yucky, but maybe not. The maybe not is because of how explicit it is
        # to say "situation" (gus)
      end

      # Usage:
      #   assert_redirected_to '/foo/bar'
      #   assert_redirected_to %r[foo/bar]
      def asserts_redirected_to(expected_path)
        asserts_response_status 302
        asserts_location expected_path
      end
    end # Macros
  end # Riot
end # Chicago

Riot::Context.instance_eval { include Chicago::Riot::Macros }