From 2f45fa7f5c698d7ac6d73d146bf840d9902a5932 Mon Sep 17 00:00:00 2001
From: James Couball <jcouball@yahoo.com>
Date: Tue, 17 May 2022 17:41:10 -0700
Subject: [PATCH] Support the --all option for git fetch

Signed-off-by: James Couball <jcouball@yahoo.com>
---
 README.md                   |  1 +
 lib/git/base.rb             |  6 +++-
 lib/git/lib.rb              |  5 +--
 tests/test_helper.rb        | 63 +++++++++++++++++++++++++++++++++++++
 tests/units/test_remotes.rb | 32 +++++++++++++++++--
 5 files changed, 101 insertions(+), 6 deletions(-)

diff --git a/README.md b/README.md
index d015e3cc..d4b68c55 100644
--- a/README.md
+++ b/README.md
@@ -288,6 +288,7 @@ g.remote(name).merge(branch)
 g.fetch
 g.fetch(g.remotes.first)
 g.fetch('origin', {:ref => 'some/ref/head'} )
+g.fetch(all: true, force: true, depth: 2)
 
 g.pull
 g.pull(Git::Repo, Git::Branch) # fetch and a merge
diff --git a/lib/git/base.rb b/lib/git/base.rb
index 541cc554..2d931cf3 100644
--- a/lib/git/base.rb
+++ b/lib/git/base.rb
@@ -336,7 +336,11 @@ def checkout_file(version, file)
 
     # fetches changes from a remote branch - this does not modify the working directory,
     # it just gets the changes from the remote if there are any
-    def fetch(remote = 'origin', opts={})
+    def fetch(remote = 'origin', opts = {})
+      if remote.is_a?(Hash)
+        opts = remote
+        remote = nil
+      end
       self.lib.fetch(remote, opts)
     end
 
diff --git a/lib/git/lib.rb b/lib/git/lib.rb
index 5bf2e455..cb408246 100644
--- a/lib/git/lib.rb
+++ b/lib/git/lib.rb
@@ -877,14 +877,15 @@ def tag(name, *opts)
 
     def fetch(remote, opts)
       arr_opts = []
+      arr_opts << '--all' if opts[:all]
       arr_opts << '--tags' if opts[:t] || opts[:tags]
       arr_opts << '--prune' if opts[:p] || opts[:prune]
       arr_opts << '--prune-tags' if opts[:P] || opts[:'prune-tags']
       arr_opts << '--force' if opts[:f] || opts[:force]
       arr_opts << '--unshallow' if opts[:unshallow]
       arr_opts << '--depth' << opts[:depth] if opts[:depth]
-      arr_opts << '--'
-      arr_opts << remote
+      arr_opts << '--' if remote || opts[:ref]
+      arr_opts << remote if remote
       arr_opts << opts[:ref] if opts[:ref]
 
       command('fetch', arr_opts)
diff --git a/tests/test_helper.rb b/tests/test_helper.rb
index b04f3f4d..31ed8477 100644
--- a/tests/test_helper.rb
+++ b/tests/test_helper.rb
@@ -97,4 +97,67 @@ def with_custom_env_variables(&block)
       Git::Lib::ENV_VARIABLE_NAMES.each { |k| ENV[k] = saved_env[k] }
     end
   end
+
+  # Assert that the expected command line args are generated for a given Git::Lib method
+  #
+  # This assertion generates an empty git repository and then runs calls
+  # Git::Base method named by `git_cmd` passing that method `git_cmd_args`.
+  #
+  # Before calling `git_cmd`, this method stubs the `Git::Lib#command` method to
+  # capture the args sent to it by `git_cmd`. These args are captured into
+  # `actual_command_line`.
+  #
+  # assert_equal is called comparing the given `expected_command_line` to
+  # `actual_command_line`.
+  #
+  # @example Fetch with no args
+  #   expected_command_line = ['fetch', '--', 'origin']
+  #   git_cmd = :fetch
+  #   git_cmd_args = []
+  #   assert_command_line(expected_command_line, git_cmd, git_cmd_args)
+  #
+  # @example Fetch with some args
+  #   expected_command_line = ['fetch', '--depth', '2', '--', 'origin', 'master']
+  #   git_cmd = :fetch
+  #   git_cmd_args = ['origin', ref: 'master', depth: '2']
+  #   assert_command_line(expected_command_line, git_cmd, git_cmd_args)
+  #
+  # @example Fetch all
+  #   expected_command_line = ['fetch', '--all']
+  #   git_cmd = :fetch
+  #   git_cmd_args = [all: true]
+  #   assert_command_line(expected_command_line, git_cmd, git_cmd_args)
+  #
+  # @param expected_command_line [Array<String>] The expected arguments to be sent to Git::Lib#command
+  # @param git_cmd [Symbol] the method to be called on the Git::Base object
+  # @param git_cmd_args [Array<Object>] The arguments to be sent to the git_cmd method
+  #
+  # @yield [git] An initialization block
+  #   The initialization block is called after a test project is created with Git.init.
+  #   The current working directory is set to the root of the test project's working tree.
+  # @yieldparam git [Git::Base] The Git::Base object resulting from initializing the test project
+  # @yieldreturn [void] the return value of the block is ignored
+  #
+  # @return [void]
+  #
+  def assert_command_line(expected_command_line, git_cmd, git_cmd_args)
+    actual_command_line = nil
+
+    in_temp_dir do |path|
+      git = Git.init('test_project')
+
+      Dir.chdir 'test_project' do
+        yield(git) if block_given?
+
+        # Mock the Git::Lib#command method to capture the actual command line args
+        git.lib.define_singleton_method(:command) do |cmd, *opts, &block|
+          actual_command_line = [cmd, *opts.flatten]
+        end
+
+        git.send(git_cmd, *git_cmd_args)
+      end
+    end
+
+    assert_equal(expected_command_line, actual_command_line)
+  end
 end
diff --git a/tests/units/test_remotes.rb b/tests/units/test_remotes.rb
index cc547f8b..ab8f6f85 100644
--- a/tests/units/test_remotes.rb
+++ b/tests/units/test_remotes.rb
@@ -1,6 +1,6 @@
 #!/usr/bin/env ruby
 
-require File.dirname(__FILE__) + '/../test_helper'
+require_relative '../test_helper'
 
 class TestRemotes < Test::Unit::TestCase
   def setup
@@ -123,6 +123,34 @@ def test_fetch
     end
   end
 
+  def test_fetch_cmd_with_no_args
+    expected_command_line = ['fetch', '--', 'origin']
+    git_cmd = :fetch
+    git_cmd_args = []
+    assert_command_line(expected_command_line, git_cmd, git_cmd_args)
+  end
+
+  def test_fetch_cmd_with_origin_and_branch
+    expected_command_line = ['fetch', '--depth', '2', '--', 'origin', 'master']
+    git_cmd = :fetch
+    git_cmd_args = ['origin', ref: 'master', depth: '2']
+    assert_command_line(expected_command_line, git_cmd, git_cmd_args)
+  end
+
+  def test_fetch_cmd_with_all
+    expected_command_line = ['fetch', '--all']
+    git_cmd = :fetch
+    git_cmd_args = [all: true]
+    assert_command_line(expected_command_line, git_cmd, git_cmd_args)
+  end
+
+  def test_fetch_cmd_with_all_with_other_args
+    expected_command_line = ['fetch', '--all', '--force', '--depth', '2']
+    git_cmd = :fetch
+    git_cmd_args = [all: true, force: true, depth: '2']
+    assert_command_line(expected_command_line, git_cmd, git_cmd_args)
+  end
+
   def test_fetch_command_injection
     test_file = 'VULNERABILITY_EXISTS'
     vulnerability_exists = false
@@ -208,6 +236,4 @@ def test_push
       assert(rem.tag('test-tag'))
     end
   end
-
-
 end

<!DOCTYPE html PUBLIC '-//W3C//DTD XHTML 1.0 Transitional//EN' 'http://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd'>
<html xmlns='http://www.w3.org/1999/xhtml'>
<head>
<title>pFad - Phonifier reborn</title>
<meta http-equiv='Content-Type' content='text/html; charset=utf-8' />
</head>
<body>
<h1>Pfad - The Proxy pFad of &#169; 2024 Garber Painting. All rights reserved.</h1>


<!-- Disclaimer -->
<p>Note: This service is not intended for secure transactions such as banking, social media, email, or purchasing. Use at your own risk. We assume no liability whatsoever for broken pages.</p>
<br>
<p>Alternative Proxies:</p><p><a href="http://rainy.clevelandohioweatherforecast.com/php-proxy/index.php?q=https://patch-diff.githubusercontent.com/raw/ruby-git/ruby-git/pull/583.patch" target="_blank">Alternative Proxy</a></p><p><a href="http://rainy.clevelandohioweatherforecast.com/pFad/index.php?u=https://patch-diff.githubusercontent.com/raw/ruby-git/ruby-git/pull/583.patch" target="_blank">pFad Proxy</a></p><p><a href="http://rainy.clevelandohioweatherforecast.com/pFad/v3index.php?u=https://patch-diff.githubusercontent.com/raw/ruby-git/ruby-git/pull/583.patch" target="_blank">pFad v3 Proxy</a></p><p><a href="http://rainy.clevelandohioweatherforecast.com/pFad/v4index.php?u=https://patch-diff.githubusercontent.com/raw/ruby-git/ruby-git/pull/583.patch" target="_blank">pFad v4 Proxy</a></p></body>
</html>