summaryrefslogtreecommitdiff
path: root/spec/lib/gitlab/git/diff_collection_spec.rb
diff options
context:
space:
mode:
authorAhmad Sherif <me@ahmadsherif.com>2017-07-14 00:22:09 +0200
committerAhmad Sherif <me@ahmadsherif.com>2017-07-19 18:05:22 +0200
commitef2b81adb442f739216e9785dd890de952a12d23 (patch)
treeddda5928b8db242ff04717ab54624ac15f22aad5 /spec/lib/gitlab/git/diff_collection_spec.rb
parenta03d21ef3905e2755bba06684f8f37f706652229 (diff)
downloadgitlab-ce-ef2b81adb442f739216e9785dd890de952a12d23.tar.gz
Migrate DiffCollection limiting logic to Gitalyfeature/send-diff-limits-to-gitaly
Diffstat (limited to 'spec/lib/gitlab/git/diff_collection_spec.rb')
-rw-r--r--spec/lib/gitlab/git/diff_collection_spec.rb2
1 files changed, 2 insertions, 0 deletions
diff --git a/spec/lib/gitlab/git/diff_collection_spec.rb b/spec/lib/gitlab/git/diff_collection_spec.rb
index d20298fa139..0cfb210e390 100644
--- a/spec/lib/gitlab/git/diff_collection_spec.rb
+++ b/spec/lib/gitlab/git/diff_collection_spec.rb
@@ -484,6 +484,8 @@ describe Gitlab::Git::DiffCollection, seed_helper: true do
end
def each
+ return enum_for(:each) unless block_given?
+
loop do
break if @count.zero?
# It is critical to decrement before yielding. We may never reach the lines after 'yield'.