Made fragment caching in views work for rjs and builder as well (closes #6642) [zsombor]

git-svn-id: http://svn-commit.rubyonrails.org/rails/trunk@8542 5ecf4fe2-1ee6-0310-87b1-e25e094e27de

actionpack/CHANGELOG

@@ -1,5 +1,7 @@
 *SVN*
 
+* Made fragment caching in views work for rjs and builder as well #6642 [zsombor]
+
 * Fixed rendering of partials with layout when done from site layout #9209 [antramm]
 
 * Fix atom_feed_helper to comply with the atom spec.  Closes #10672 [xaviershay]

actionpack/lib/action_controller/caching.rb

@@ -371,11 +371,10 @@ module ActionController #:nodoc:
         name.is_a?(Hash) ? url_for(name).split("://").last : name
       end
 
-      # Called by CacheHelper#cache
-      def cache_erb_fragment(block, name = {}, options = nil)
+      def fragment_for(block, name = {}, options = nil) #:nodoc:
         unless perform_caching then block.call; return end
 
-        buffer = eval(ActionView::Base.erb_variable, block.binding)
+        buffer = yield
 
         if cache = read_fragment(name, options)
           buffer.concat(cache)
@@ -386,6 +385,33 @@ module ActionController #:nodoc:
         end
       end
 
+      # Called by CacheHelper#cache
+      def cache_rxml_fragment(block, name = {}, options = nil) #:nodoc:
+        fragment_for(block, name, options) do
+          eval('xml.target!', block.binding)
+        end
+      end
+      
+      # Called by CacheHelper#cache
+      def cache_rjs_fragment(block, name = {}, options = nil) #:nodoc:
+        fragment_for(block, name, options) do
+          begin
+            debug_mode, ActionView::Base.debug_rjs = ActionView::Base.debug_rjs, false
+            eval('page.to_s', block.binding)
+          ensure
+            ActionView::Base.debug_rjs = debug_mode
+          end
+        end
+      end
+      
+      # Called by CacheHelper#cache
+      def cache_erb_fragment(block, name = {}, options = nil) #:nodoc:
+        fragment_for(block, name, options) do
+          eval(ActionView::Base.erb_variable, block.binding)
+        end
+      end
+
+
       # Writes <tt>content</tt> to the location signified by <tt>name</tt> (see <tt>expire_fragment</tt> for acceptable formats)
       def write_fragment(name, content, options = nil)
         return unless perform_caching

actionpack/lib/action_view/helpers/cache_helper.rb

@@ -32,7 +32,21 @@ module ActionView
       #      <i>Topics listed alphabetically</i>
       #    <% end %>
       def cache(name = {}, &block)
-        @controller.cache_erb_fragment(block, name)
+        template_extension = first_render[/\.(\w+)$/, 1].to_sym
+        case template_extension
+        when :erb, :rhtml
+          @controller.cache_erb_fragment(block, name)
+        when :rjs
+          @controller.cache_rjs_fragment(block, name)
+        when :builder, :rxml
+          @controller.cache_rxml_fragment(block, name)
+        else
+          # do a last ditch effort for those brave souls using
+          # different template engines. This should give plugin
+          # writters a simple hook.
+          raise "fragment caching not supported for #{template_extension} files." unless @controller.respond_to?("cache_#{template_extension}_fragment")
+          @controller.send "cache_#{template_extension}_fragment", block, name
+        end
       end
     end
   end

actionpack/test/controller/caching_test.rb

@@ -347,3 +347,135 @@ class ActionCacheTest < Test::Unit::TestCase
       assert File.exist?(full_path), "#{full_path.inspect} does not exist."
     end
 end
+
+class FragmentCachingTestController < ActionController::Base
+  def some_action; end;
+end
+
+class FragmentCachingTest < Test::Unit::TestCase
+  def setup
+    ActionController::Base.perform_caching = true
+    @store = ActionController::Caching::Fragments::UnthreadedMemoryStore.new
+    ActionController::Base.fragment_cache_store = @store
+    @controller = FragmentCachingTestController.new
+    @params = {:controller => 'posts', :action => 'index'}
+    @request = ActionController::TestRequest.new
+    @response = ActionController::TestResponse.new
+    @controller.params = @params
+    @controller.request = @request
+    @controller.response = @response
+    @controller.send(:initialize_current_url)
+  end
+
+  def test_fragement_cache_key
+    assert_equal 'what a key', @controller.fragment_cache_key('what a key')
+    assert_equal( "test.host/fragment_caching_test/some_action",
+                  @controller.fragment_cache_key(:controller => 'fragment_caching_test',:action => 'some_action'))
+  end
+
+  def test_read_fragment__with_caching_enabled
+    @store.write('name', 'value')
+    assert_equal 'value', @controller.read_fragment('name')
+  end
+
+  def test_read_fragment__with_caching_disabled
+    ActionController::Base.perform_caching = false
+    @store.write('name', 'value')
+    assert_nil @controller.read_fragment('name')
+  end
+
+  def test_write_fragment__with_caching_enabled
+    assert_nil @store.read('name')
+    assert_equal 'value', @controller.write_fragment('name', 'value')
+    assert_equal 'value', @store.read('name')
+  end
+
+  def test_write_fragment__with_caching_disabled
+    assert_nil @store.read('name')
+    ActionController::Base.perform_caching = false
+    assert_equal nil, @controller.write_fragment('name', 'value')
+    assert_nil @store.read('name')
+  end
+
+  def test_expire_fragment__with_simple_key
+    @store.write('name', 'value')
+    @controller.expire_fragment 'name'
+    assert_nil @store.read('name')
+  end
+
+  def test_expire_fragment__with__regexp
+    @store.write('name', 'value')
+    @store.write('another_name', 'another_value')
+    @store.write('primalgrasp', 'will not expire ;-)')
+
+    @controller.expire_fragment /name/
+
+    assert_nil @store.read('name')
+    assert_nil @store.read('another_name')
+    assert_equal 'will not expire ;-)', @store.read('primalgrasp')
+  end
+
+  def test_fragment_for__with_disabled_caching
+    ActionController::Base.perform_caching = false
+
+    @store.write('expensive', 'fragment content')
+    fragment_computed = false
+
+    buffer = 'generated till now -> '
+    @controller.fragment_for(Proc.new { fragment_computed = true }, 'expensive') { buffer }
+
+    assert fragment_computed
+    assert_equal 'generated till now -> ', buffer
+  end
+
+
+  def test_fragment_for
+    @store.write('expensive', 'fragment content')
+    fragment_computed = false
+
+    buffer = 'generated till now -> '
+    @controller.fragment_for(Proc.new { fragment_computed = true }, 'expensive') { buffer}
+
+    assert !fragment_computed
+    assert_equal 'generated till now -> fragment content', buffer
+  end
+
+  def test_cache_erb_fragment
+    @store.write('expensive', 'fragment content')
+    _erbout = 'generated till now -> '
+
+    assert_equal( 'generated till now -> fragment content',
+                  @controller.cache_erb_fragment(Proc.new{ }, 'expensive'))
+  end
+
+  def test_cache_rxml_fragment
+    @store.write('expensive', 'fragment content')
+    xml = 'generated till now -> '
+    class << xml; def target!; to_s; end; end
+
+    assert_equal( 'generated till now -> fragment content',
+                  @controller.cache_rxml_fragment(Proc.new{ }, 'expensive'))
+  end
+
+  def test_cache_rjs_fragment
+    @store.write('expensive', 'fragment content')
+    page = 'generated till now -> '
+
+    assert_equal( 'generated till now -> fragment content',
+                  @controller.cache_rjs_fragment(Proc.new{ }, 'expensive'))
+  end
+
+  def test_cache_rjs_fragment_debug_mode_does_not_interfere
+    @store.write('expensive', 'fragment content')
+    page = 'generated till now -> '
+
+    begin
+      debug_mode, ActionView::Base.debug_rjs = ActionView::Base.debug_rjs, true
+      assert_equal( 'generated till now -> fragment content',
+                    @controller.cache_rjs_fragment(Proc.new{ }, 'expensive'))
+      assert ActionView::Base.debug_rjs
+    ensure
+      ActionView::Base.debug_rjs = debug_mode
+    end
+  end
+end