class Shoulda::Matchers::ActionController::SetFlashMatcher

def [](key)
  @key = key
  underlying_matcher[key]
  self
end

def in_context(context)
  underlying_matcher.in_context(context)
  self
end

def initialize
  store = FlashStore.future
  @underlying_matcher = SetSessionOrFlashMatcher.new(store)
end

def now
  if key || expected_value
    raise QualifierOrderError
  end
  store = FlashStore.now
  @underlying_matcher = SetSessionOrFlashMatcher.new(store)
  self
end

def to(expected_value = nil, &block)
  @expected_value = expected_value
  underlying_matcher.to(expected_value, &block)
  self
end