Make tag registry enumerable

dyspop · Apr 25, 2016 · 6adc431 · 6adc431
1 parent 23d2bee
commit 6adc431
Show file tree

Hide file tree

Showing 2 changed files with 12 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/template_unit_test.rb b/test/unit/template_unit_test.rb
@@ -67,4 +67,10 @@ 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"])
+  end
 end