class Shoulda::Matchers::ActionController::SetTheFlashMatcher

def flash
  if @flash
    @flash
  else
    @flash = @controller.flash.dup
    used = @controller.flash.instance_variable_get(:@used).dup
    @flash.instance_variable_set(:@used, used)
    sweep_flash_if_necessary
    @flash
  end
end