require 'test/unit'
require File.dirname(__FILE__) + '/../lib/controller_helper'

class ControllerMock
  include RHNH::Magellan::ControllerHelper
  
  attr_accessor :session
  
  def initialize
    @session = {}
  end
end

class ControllerHelperTest < Test::Unit::TestCase
  def test_set_nav_state
    c = ControllerMock.new
    c.set_nav_state :key, 1
    assert_equal 1, c.session[:rhnh_navigator_state][:key]
  end
end