Support merge-base

Evgenii Pecherkin · Evgenii Pecherkin · commit 5fb245915da5 · 2018-06-14T19:20:41.000-04:00
diff --git a/README.md b/README.md
@@ -215,6 +215,8 @@ And here are the operations that will need to write to your git repository.
      g.merge(g.branch('master'))
      g.merge([branch1, branch2])
 
+     g.merge_base('branch1', 'branch2')
+
      r = g.add_remote(name, uri)  # Git::Remote
      r = g.add_remote(name, Git::Base)  # Git::Remote
 
diff --git a/lib/git/base/factory.rb b/lib/git/base/factory.rb
@@ -68,6 +68,14 @@ def tag(tag_name)
         Git::Object.new(self, tag_name, 'tag', true)
       end
 
+      # Find as good common ancestors as possible for a merge
+      # example: g.merge_base('master', 'some_branch', 'some_sha', octopus: true)
+      # returns Array<Git::Object::Commit>
+      def merge_base(*args)
+        shas = self.lib.merge_base(*args)
+        shas.map { |sha| gcommit(sha) }
+      end
+
     end
 
   end
diff --git a/lib/git/lib.rb b/lib/git/lib.rb
@@ -660,6 +660,21 @@ def merge(branch, message = nil)
       command('merge', arr_opts)
     end
 
+    def merge_base(*args)
+      opts = args.last.is_a?(Hash) ? args.pop : {}
+
+      arg_opts = []
+
+      arg_opts << '--octopus' if opts[:octopus]
+      arg_opts << '--independent' if opts[:independent]
+      arg_opts << '--fork-point' if opts[:fork_point]
+      arg_opts << '--all' if opts[:all]
+
+      arg_opts += args
+
+      command('merge-base', arg_opts).lines.map(&:strip)
+    end
+
     def unmerged
       unmerged = []
       command_lines('diff', ["--cached"]).each do |line|
diff --git a/tests/units/test_merge_base.rb b/tests/units/test_merge_base.rb
@@ -0,0 +1,172 @@
+#!/usr/bin/env ruby
+
+require File.dirname(__FILE__) + '/../test_helper'
+
+class TestMergeBase < Test::Unit::TestCase
+  def setup
+    set_file_paths
+  end
+
+  def test_branch_and_master_merge_base
+    in_temp_dir do |path|
+      g = Git.clone(@wbare, 'branch_merge_test')
+      Dir.chdir('branch_merge_test') do
+
+        true_ancestor_sha = g.gcommit('master').sha
+
+        g.branch('new_branch').in_branch('test commit 1') do
+          new_file('new_file_1', 'hello')
+          g.add
+          true
+        end
+
+        g.branch('master').in_branch('test commit 2') do
+          new_file('new_file_2', 'hello')
+          g.add
+          true
+        end
+
+        ancestors = g.merge_base('master', 'new_branch')
+        assert_equal(ancestors.size, 1) # there is only one true ancestor
+        assert_equal(ancestors.first.sha, true_ancestor_sha) # proper common ancestor
+      end
+    end
+  end
+
+  def test_branch_and_master_independent_merge_base
+    in_temp_dir do |path|
+      g = Git.clone(@wbare, 'branch_merge_test')
+      Dir.chdir('branch_merge_test') do
+
+        true_ancestor_sha = g.gcommit('master').sha
+
+        g.branch('new_branch').in_branch('test commit 1') do
+          new_file('new_file_1', 'hello')
+          g.add
+          true
+        end
+
+        g.branch('master').in_branch('test commit 2') do
+          new_file('new_file_2', 'hello')
+          g.add
+          true
+        end
+
+        independent_commits = g.merge_base(true_ancestor_sha, 'master', 'new_branch', independent: true)
+        assert_equal(independent_commits.size, 2) # both new master and a branch are unreachable from each other
+        true_independent_commits_shas = [g.gcommit('master').sha, g.gcommit('new_branch').sha]
+        assert_equal(independent_commits.map(&:sha).sort, true_independent_commits_shas.sort)
+      end
+    end
+  end
+
+  def test_branch_and_master_fork_point_merge_base
+    in_temp_dir do |path|
+      g = Git.clone(@wbare, 'branch_merge_test')
+      Dir.chdir('branch_merge_test') do
+
+        g.branch('master').in_branch('test commit 1') do
+          new_file('new_file_1', 'hello')
+          g.add
+          true
+        end
+
+        true_ancestor_sha = g.gcommit('master').sha
+
+        g.branch('new_branch').in_branch('test commit 2') do
+          new_file('new_file_2', 'hello')
+          g.add
+          true
+        end
+
+        g.branch('master').in_branch('test commit 3') do
+          g.reset_hard(g.gcommit('HEAD^'))
+
+          new_file('new_file_3', 'hello')
+          g.add
+          true
+        end
+
+        ancestors = g.merge_base('master', 'new_branch', fork_point: true)
+        assert_equal(ancestors.size, 1) # there is only one true ancestor
+        assert_equal(ancestors.first.sha, true_ancestor_sha) # proper common ancestor
+      end
+    end
+  end
+
+  def test_branches_and_master_merge_base
+    in_temp_dir do |path|
+      g = Git.clone(@wbare, 'branch_merge_test')
+      Dir.chdir('branch_merge_test') do
+
+        g.branch('new_branch_1').in_branch('test commit 1') do
+          new_file('new_file_1', 'hello')
+          g.add
+          true
+        end
+
+        g.branch('master').in_branch('test commit 2') do
+          new_file('new_file_2', 'hello')
+          g.add
+          true
+        end
+
+        non_octopus_ancestor_sha = g.gcommit('master').sha
+
+        g.branch('new_branch_2').in_branch('test commit 3') do
+          new_file('new_file_3', 'hello')
+          g.add
+          true
+        end
+
+        g.branch('master').in_branch('test commit 4') do
+          new_file('new_file_4', 'hello')
+          g.add
+          true
+        end
+
+        ancestors = g.merge_base('master', 'new_branch_1', 'new_branch_2')
+        assert_equal(ancestors.size, 1) # there is only one true ancestor
+        assert_equal(ancestors.first.sha, non_octopus_ancestor_sha) # proper common ancestor
+      end
+    end
+  end
+
+  def test_branches_and_master_octopus_merge_base
+    in_temp_dir do |path|
+      g = Git.clone(@wbare, 'branch_merge_test')
+      Dir.chdir('branch_merge_test') do
+
+        true_ancestor_sha = g.gcommit('master').sha
+
+        g.branch('new_branch_1').in_branch('test commit 1') do
+          new_file('new_file_1', 'hello')
+          g.add
+          true
+        end
+
+        g.branch('master').in_branch('test commit 2') do
+          new_file('new_file_2', 'hello')
+          g.add
+          true
+        end
+
+        g.branch('new_branch_2').in_branch('test commit 3') do
+          new_file('new_file_3', 'hello')
+          g.add
+          true
+        end
+
+        g.branch('master').in_branch('test commit 4') do
+          new_file('new_file_4', 'hello')
+          g.add
+          true
+        end
+
+        ancestors = g.merge_base('master', 'new_branch_1', 'new_branch_2', octopus: true)
+        assert_equal(ancestors.size, 1) # there is only one true ancestor
+        assert_equal(ancestors.first.sha, true_ancestor_sha) # proper common ancestor
+      end
+    end
+  end
+end
