Update #each methods to work when no block provided

ar-dm · Jan 24, 2013 · 422176a · 422176a
1 parent d610a56
commit 422176a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/lib/dm-core/support/ordered_set.rb b/lib/dm-core/support/ordered_set.rb
@@ -316,6 +316,7 @@ def clear
  #
  # @api private
  def each
+ return to_enum unless block_given?
  entries.each { |entry| yield(entry) }
  self
  end

diff --git a/lib/dm-core/support/subject_set.rb b/lib/dm-core/support/subject_set.rb
@@ -207,6 +207,7 @@ def [](name)
  #
  # @api private
  def each
+ return to_enum unless block_given?
  entries.each { |entry| yield(entry) }
  self
  end