test/template_cache_test.rb in sinatra-assetpack-0.2.2 vs test/template_cache_test.rb in sinatra-assetpack-0.2.3

@@ -1,17 +1,17 @@
 require File.expand_path('../test_helper', __FILE__)
 
 class TemplateCacheTest < UnitTest
-  class App < UnitTest::App
-    register Sinatra::AssetPack
-
+  class App < Main
     set :reload_templates, false
     assets do |a|
       a.css :a, '/a.css', %w[/css/screen.css]
     end
   end
 
-  def app() App; end
+  def app
+    App
+  end
 
   test "cache dynamic files when reload templates is false" do
     App.any_instance.expects(:asset_filter_css).times(1).returns "OK"
     10.times { get '/css/screen.css' }
   end