From 11aa5157355d06ddbc9cad8fd0aa43a75ac8431e Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jos=C3=A9=20Valim?= <jose.valim@gmail.com>
Date: Sun, 10 Oct 2010 13:43:32 +0200
Subject: Add expire! and rerender to the template API. This will be used by
 SASS template handler.

---
 actionpack/lib/action_view/template.rb    | 23 +++++++++++++--
 actionpack/test/template/template_test.rb | 49 +++++++++++++++++++++++++++++--
 2 files changed, 67 insertions(+), 5 deletions(-)

(limited to 'actionpack')

diff --git a/actionpack/lib/action_view/template.rb b/actionpack/lib/action_view/template.rb
index 074daa5d28..5c3e0e899b 100644
--- a/actionpack/lib/action_view/template.rb
+++ b/actionpack/lib/action_view/template.rb
@@ -154,13 +154,30 @@ module ActionView
     # Notice this method raises an error if the template to be refreshed does not have a
     # virtual path set (true just for inline templates).
     def refresh(view)
-      raise "A template need to have a virtual path in order to be refreshed" unless @virtual_path
+      raise "A template needs to have a virtual path in order to be refreshed" unless @virtual_path
       lookup  = view.lookup_context
       pieces  = @virtual_path.split("/")
       name    = pieces.pop
-      partial = name.sub!(/^_/, "")
+      partial = !!name.sub!(/^_/, "")
       lookup.disable_cache do
-        lookup.find_template(name, pieces.join, partial || false, @locals)
+        lookup.find_template(name, pieces.join, partial, @locals)
+      end
+    end
+
+    # Expires this template by setting his updated_at date to Jan 1st, 1970.
+    def expire!
+      @updated_at = Time.utc(1970)
+    end
+
+    # Receives a view context and renders a template exactly like self by using
+    # the @virtual_path. It raises an error if no @virtual_path was given.
+    def rerender(view)
+      raise "A template needs to have a virtual path in order to be rerendered" unless @virtual_path
+      name = @virtual_path.dup
+      if name.sub!(/(^|\/)_([^\/]*)$/, '\1\2')
+        view.render :partial => name
+      else
+        view.render :template => @virtual_path
       end
     end
 
diff --git a/actionpack/test/template/template_test.rb b/actionpack/test/template/template_test.rb
index 34679962b2..22401dd145 100644
--- a/actionpack/test/template/template_test.rb
+++ b/actionpack/test/template/template_test.rb
@@ -90,20 +90,65 @@ class TestERBTemplate < ActiveSupport::TestCase
     assert_equal "hellopartialhello", render
   end
 
-  def test_refresh
+  def test_refresh_with_templates
     @template = new_template("Hello", :virtual_path => "test/foo")
     @template.locals = [:key]
     @context.lookup_context.expects(:find_template).with("foo", "test", false, [:key]).returns("template")
     assert_equal "template", @template.refresh(@context)
   end
 
+  def test_refresh_with_partials
+    @template = new_template("Hello", :virtual_path => "test/_foo")
+    @template.locals = [:key]
+    @context.lookup_context.expects(:find_template).with("foo", "test", true, [:key]).returns("partial")
+    assert_equal "partial", @template.refresh(@context)
+  end
+
   def test_refresh_raises_an_error_without_virtual_path
     @template = new_template("Hello", :virtual_path => nil)
-    assert_raise RuntimeError, /OMG/ do
+    assert_raise RuntimeError do
       @template.refresh(@context)
     end
   end
 
+  def test_template_expire_sets_the_timestamp_to_1970
+    @template = new_template("Hello", :updated_at => Time.utc(2010))
+    assert_equal Time.utc(2010), @template.updated_at
+    @template.expire!
+    assert_equal Time.utc(1970), @template.updated_at
+  end
+
+  def test_template_rerender_renders_a_template_like_self
+    @template = new_template("Hello", :virtual_path => "test/foo_bar")
+    @context.expects(:render).with(:template => "test/foo_bar").returns("template")
+    assert_equal "template", @template.rerender(@context)
+  end
+
+  def test_template_rerender_renders_a_root_template_like_self
+    @template = new_template("Hello", :virtual_path => "foo_bar")
+    @context.expects(:render).with(:template => "foo_bar").returns("template")
+    assert_equal "template", @template.rerender(@context)
+  end
+
+  def test_template_rerender_renders_a_partial_like_self
+    @template = new_template("Hello", :virtual_path => "test/_foo_bar")
+    @context.expects(:render).with(:partial => "test/foo_bar").returns("partial")
+    assert_equal "partial", @template.rerender(@context)
+  end
+
+  def test_template_rerender_renders_a_root_partial_like_self
+    @template = new_template("Hello", :virtual_path => "_foo_bar")
+    @context.expects(:render).with(:partial => "foo_bar").returns("partial")
+    assert_equal "partial", @template.rerender(@context)
+  end
+
+  def test_rerender_raises_an_error_without_virtual_path
+    @template = new_template("Hello", :virtual_path => nil)
+    assert_raise RuntimeError do
+      @template.rerender(@context)
+    end
+  end
+
   if "ruby".encoding_aware?
     def test_resulting_string_is_utf8
       @template = new_template
-- 
cgit v1.2.3