Incorporate RefsService.FindAllBranches Gitaly RPCgitaly-all-branches

1 files changed, 7 insertions, 2 deletions

diff --git a/spec/lib/gitlab/git/repository_spec.rb b/spec/lib/gitlab/git/repository_spec.rb
index d9d95e22e3b..50736d353ad 100644
--- a/spec/lib/gitlab/git/repository_spec.rb
+++ b/spec/lib/gitlab/git/repository_spec.rb
@@ -939,16 +939,21 @@ describe Gitlab::Git::Repository, seed_helper: true do
     context 'with deleted branch with Gitaly disabled' do
       before do
         allow(Gitlab::GitalyClient).to receive(:feature_enabled?).and_return(false)
+      end
+
+      it 'returns no results' do
         ref = double()
         allow(ref).to receive(:name) { 'bad-branch' }
         allow(ref).to receive(:target) { raise Rugged::ReferenceError }
         branches = double()
         allow(branches).to receive(:each) { [ref].each }
         allow(repository.rugged).to receive(:branches) { branches }
-      end
 
-      it { is_expected.to eq([]) }
+        expect(subject).to be_empty
+      end
     end
+
+    it_behaves_like 'wrapping gRPC errors', Gitlab::GitalyClient::RefService, :branches
   end
 
   describe '#branch_count' do