class Shoulda::ActionController::Matchers::SetTheFlashMatcher

def description
  description = "set the flash"
  description << " to #{@value.inspect}" unless @value.nil?
  description
end

def expectation
  expectation = "the flash to be set"
  expectation << " to #{@value.inspect}" unless @value.nil?
  expectation << ", but #{flash_description}"
  expectation
end

def failure_message
  "Expected #{expectation}"
end

def flash
  @controller.response.session['flash']
end

def flash_description
  if flash.blank?
    "no flash was set"
  else
    "was #{flash.inspect}"
  end
end

def matches?(controller)
  @controller = controller
  sets_the_flash? && string_value_matches? && regexp_value_matches?
end

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

def regexp_value_matches?
  return true unless Regexp === @value
  flash.values.any? {|value| value =~ @value }
end

def sets_the_flash?
  !flash.blank?
end

def string_value_matches?
  return true unless String === @value
  flash.values.any? {|value| value == @value }
end

def to(value)
  @value = value
  self
end