$:.unshift File.join(File.dirname(__FILE__), '..', '..', 'lib')

require 'test/unit'
require 'ostruct'

require 'nitro'
require 'nitro/controller'
require 'nitro/flash'

class TC_Flash < Test::Unit::TestCase # :nodoc: all
  include Nitro

  class MyController < Controller
    attr_accessor :flag
    
    def action1
      flash[:msg] = 'Hello world!'
    end
    
    def action2
      @flag = flash[:msg]
    end
  end

  def setup
    @conf = OpenStruct.new
  end

  def teardown
    @conf = nil
  end
  
  def test_all
    ctx = Context.new(@conf)
    ctx.headers = {}
    ctx.params = {}
    ctx.instance_eval '@session = {}'
    c = MyController.new(ctx)  
    c.action1
    c.action2
    assert_equal 'Hello world!', c.flag
    c.action2
    assert_equal 'Hello world!', c.flag
  end  
end