Merge pull request #88 from ddfreyne/argument-list-each-enumerator

Let ArgumentList#each return enumerator unless block given
denisdefreyne · Jan 19, 2019 · 8b74c86 · 8b74c86
2 parents ea6b55e + 2298e0a
commit 8b74c86
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/lib/cri/argument_list.rb b/lib/cri/argument_list.rb
@@ -37,6 +37,8 @@ def [](key)
     end
 
     def each
+      return to_enum(__method__) unless block_given?
+
       @arguments_array.each { |e| yield(e) }
       self
     end

diff --git a/test/test_argument_list.rb b/test/test_argument_list.rb
@@ -33,6 +33,12 @@ def test_enum
       assert_equal(%w[A B C], args.map(&:upcase))
     end
 
+    def test_enum_without_block
+      args = Cri::ArgumentList.new(%w[a b c], false, [])
+
+      assert_equal(%w[A B C], args.each.map(&:upcase))
+    end
+
     def test_no_method_error
       args = Cri::ArgumentList.new(%w[a b c], false, [])