test/test_helper.rb in menu_builder-0.3.0 vs test/test_helper.rb in menu_builder-0.3.1

@@ -1,11 +1,24 @@
 require 'rubygems'
 require 'test/unit'
+
+gem "actionpack", "3.0.0.beta4"
+gem "activemodel", "3.0.0.beta4"
+
 require 'action_controller'
-require 'action_view/test_case'
+require 'action_view'
+require 'action_view/template'
+require 'active_support'
+require 'rails/railtie'
 
-$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
-$LOAD_PATH.unshift(File.dirname(__FILE__))
+$:.unshift File.expand_path(File.dirname(__FILE__) + '/../lib')
+require 'menu_builder'
 
 Dir["#{File.dirname(__FILE__)}/support/*.rb"].each { |f| require f }
 
-require 'menu_builder'
+MenuBuilder::Routes = ActionDispatch::Routing::RouteSet.new
+MenuBuilder::Routes.draw do |map|
+  match ':controller(/:action(/:id(.:format)))'
+end
+
+class ApplicationController < ActionController::Base; end
+ActionController::Base.send :include, MenuBuilder::Routes.url_helpers
\ No newline at end of file