Bring layouts with proc back alive.

1 files changed, 15 insertions, 1 deletions

diff --git a/actionpack/test/abstract/layouts_test.rb b/actionpack/test/abstract/layouts_test.rb
index df73d948f0..8cbb72dc83 100644
--- a/actionpack/test/abstract/layouts_test.rb
+++ b/actionpack/test/abstract/layouts_test.rb
@@ -67,7 +67,15 @@ module AbstractControllerTests
     
     class WithChildOfImplied < WithStringImpliedChild
     end
-    
+
+    class WithProc < Base
+      layout proc { |c| "omg" }
+
+      def index
+        render :_template => ActionView::Template::Text.new("Hello proc!")
+      end
+    end
+
     class WithSymbol < Base
       layout :hello
       
@@ -198,6 +206,12 @@ module AbstractControllerTests
         controller.process(:index)
         assert_equal "Hello nil!", controller.response_body
       end
+
+      test "when layout is specified as a proc, call it and use the layout returned" do
+        controller = WithProc.new
+        controller.process(:index)
+        assert_equal "OMGHI2U Hello proc!", controller.response_body
+      end
       
       test "when layout is specified as a symbol, call the requested method and use the layout returned" do
         controller = WithSymbol.new