Merge pull request Shopify#748 from Shopify/expose-tags

Make Template.tags loop-able
polishgeeks · Apr 25, 2016 · e4cf55b · e4cf55b
2 parents 23d2bee + 5bb211d
commit e4cf55b
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 0 deletions.
diff --git a/lib/liquid/template.rb b/lib/liquid/template.rb
@@ -19,6 +19,8 @@ class Template
     @@file_system = BlankFileSystem.new
 
     class TagRegistry
+      include Enumerable
+
       def initialize
         @tags = {}
         @cache = {}
@@ -41,6 +43,10 @@ def delete(tag_name)
         @cache.delete(tag_name)
       end
 
+      def each(&block)
+        @tags.each(&block)
+      end
+
       private
 
       def lookup_class(name)

diff --git a/test/unit/block_unit_test.rb b/test/unit/block_unit_test.rb
@@ -46,6 +46,8 @@ def test_with_block
   def test_with_custom_tag
     Liquid::Template.register_tag("testtag", Block)
     assert Liquid::Template.parse("{% testtag %} {% endtesttag %}")
+  ensure
+    Liquid::Template.tags.delete('testtag')
   end
 
   private

diff --git a/test/unit/template_unit_test.rb b/test/unit/template_unit_test.rb
@@ -67,4 +67,12 @@ def test_tags_delete
     Template.tags.delete('fake')
     assert_nil Template.tags['fake']
   end
+
+  def test_tags_can_be_looped_over
+    Template.register_tag('fake', FakeTag)
+    result = Template.tags.map { |name, klass| [name, klass] }
+    assert result.include?(["fake", "TemplateUnitTest::FakeTag"])
+  ensure
+    Template.tags.delete('fake')
+  end
 end