test/unit/container_test.rb in apotomo-0.1.4 vs test/unit/container_test.rb in apotomo-1.0.0.beta1

@@ -1,11 +1,12 @@
 require 'test_helper'
 
 class ContainerTest < Test::Unit::TestCase
+  include Apotomo::TestCaseMethods::TestController
+  
   context "Rendering a container" do
     setup do
       @family = container('family')
-      @family.controller = @controller
     end
     
     should "return an empty view if childless" do
       assert_equal "<div id=\"family\"></div>", @family.invoke
     end