Add spec for remote_branches with env

1 files changed, 24 insertions, 1 deletions

diff --git a/ruby/spec/lib/gitlab/git/repository_mirroring_spec.rb b/ruby/spec/lib/gitlab/git/repository_mirroring_spec.rb
index a0efc83f8..dca981717 100644
--- a/ruby/spec/lib/gitlab/git/repository_mirroring_spec.rb
+++ b/ruby/spec/lib/gitlab/git/repository_mirroring_spec.rb
@@ -6,8 +6,11 @@ describe Gitlab::Git::RepositoryMirroring do
   class FakeRepository
     include Gitlab::Git::RepositoryMirroring
 
-    def initialize(projects_stub)
+    attr_reader :rugged
+
+    def initialize(projects_stub, rugged_instance = nil)
       @gitlab_projects = projects_stub
+      @rugged = rugged_instance
     end
 
     def gitlab_projects_error
@@ -15,6 +18,26 @@ describe Gitlab::Git::RepositoryMirroring do
     end
   end
 
+  describe '#remote_branches' do
+    let(:projects_stub) { double.as_null_object }
+    let(:rugged_stub) { double.as_null_object }
+
+    subject(:repository) { FakeRepository.new(projects_stub, rugged_stub) }
+
+    it 'passes environment to `ls-remote`' do
+      env = { option_a: true, option_b: false }
+
+      allow(repository).to receive(:feature_enabled?)
+        .with(:remote_branches_ls_remote)
+        .and_return(true)
+      expect(repository).to receive(:list_remote_refs)
+        .with('remote_a', env: env)
+        .and_return([])
+
+      repository.remote_branches('remote_a', env: env)
+    end
+  end
+
   describe '#push_remote_branches' do
     let(:projects_stub) { double.as_null_object }