Exposed function mergeheadForeach

Kyle Smith · Kyle Smith · commit 07ab50da39fa · 2016-03-04T08:32:28.000-07:00
diff --git a/generate/input/descriptor.json b/generate/input/descriptor.json
@@ -1759,7 +1759,10 @@
           "ignore": true
         },
         "git_repository_mergehead_foreach": {
-          "ignore": true
+          "isAsync": true,
+          "return": {
+            "isErrorCode": true
+          }
         },
         "git_repository_message": {
           "ignore": true
diff --git a/lib/repository.js b/lib/repository.js
@@ -1200,6 +1200,16 @@ Repository.prototype.fetchheadForeach = function(callback) {
   return fetchheadForeach.call(this, callback, null);
 };
 
+var mergeheadForeach = Repository.prototype.mergeheadForeach;
+/**
+ * @async
+ * @param {MergeheadForeachCb} callback The callback function to be called on
+ * each entry
+ */
+Repository.prototype.mergeheadForeach = function(callback) {
+  return mergeheadForeach.call(this, callback, null);
+};
+
 /**
  * Stages or unstages line selection of a specified file
  *
diff --git a/test/tests/repository.js b/test/tests/repository.js
@@ -3,6 +3,8 @@ var path = require("path");
 var promisify = require("promisify-node");
 var fse = promisify(require("fs-extra"));
 var local = path.join.bind(path, __dirname);
+var IndexUtils = require("../utils/index_setup");
+var RepoUtils = require("../utils/repository_setup");
 
 describe("Repository", function() {
   var NodeGit = require("../../");
@@ -260,4 +262,39 @@ describe("Repository", function() {
             });
       });
   });
+
+  it("can get all merge heads in a repo with mergeheadForeach", function() {
+    var repo;
+    var repoPath = path.join(__dirname, "..", "repos", "merge-head");
+    var ourBranchName = "ours";
+    var theirBranchName = "theirs";
+    var fileName = "testFile.txt";
+    var numMergeHeads = 0;
+    var assertBranchTargetIs = function (theirBranch, mergeHead) {
+      assert.equal(theirBranch.target(), mergeHead.toString());
+      numMergeHeads++;
+    };
+
+    return RepoUtils.createRepository(repoPath)
+      .then(function(_repo) {
+        repo = _repo;
+        return IndexUtils.createConflict(
+          repo,
+          ourBranchName,
+          theirBranchName,
+          fileName
+        );
+      })
+      .then(function() {
+        return repo.getBranch(theirBranchName);
+      })
+      .then(function(theirBranch) {
+        return repo.mergeheadForeach(
+          assertBranchTargetIs.bind(this, theirBranch)
+        );
+      })
+      .then(function() {
+        assert.equal(numMergeHeads, 1);
+      });
+  });
 });
