class Shoulda::Matchers::ActionController::RespondWithMatcher

def correct_status_code?
  response_code == @status
end

def correct_status_code_range?
  @status.is_a?(Range) &&
    @status.include?(response_code)
end

def description
  "respond with #{@status}"
end

def expectation
  "response to be a #{@status}, but was #{response_code}"
end

def failure_message
  "Expected #{expectation}"
end

def failure_message_when_negated
  "Did not expect #{expectation}"
end

def initialize(status)
  @status = symbol_to_status_code(status)
end

def matches?(controller)
  @controller = controller
  correct_status_code? || correct_status_code_range?
end

def response_code
  @controller.response.response_code
end

def symbol_to_status_code(potential_symbol)
  case potential_symbol
  when :success  then 200
  when :redirect then 300..399
  when :missing  then 404
  when :error    then 500..599
  when Symbol
    if defined?(::Rack::Utils::SYMBOL_TO_STATUS_CODE)
      ::Rack::Utils::SYMBOL_TO_STATUS_CODE[potential_symbol]
    else
      ::ActionController::Base::SYMBOL_TO_STATUS_CODE[potential_symbol]
    end
  else
    potential_symbol
  end
end