From 2e64976d8714eb9a35ed218ecdb798617a3cb86d Mon Sep 17 00:00:00 2001
From: nmilojevic1 <nmilojevic@gitlab.com>
Date: Wed, 29 Jan 2020 11:26:31 +0100
Subject: [PATCH] Use wrap_original for stubing delete_all_refs
---
spec/services/projects/after_import_service_spec.rb | 3 +--
1 file changed, 1 insertion(+), 2 deletions(-)
diff --git a/spec/services/projects/after_import_service_spec.rb b/spec/services/projects/after_import_service_spec.rb
index 7289edd1e9185..4f74a779cd520 100644
--- a/spec/services/projects/after_import_service_spec.rb
+++ b/spec/services/projects/after_import_service_spec.rb
@@ -77,9 +77,8 @@
before do
call_count = 0
- original_method = repository.method(:delete_all_refs_except)
- allow(repository).to receive(:delete_all_refs_except) do |*args|
+ allow(repository).to receive(:delete_all_refs_except).and_wrap_original do |original_method, *args|
call_count += 1
call_count > 1 ? original_method.call(*args) : raise(exception)
end
--
GitLab