Add Repository.prototype.fetchheadForeach and tests

John Haley · John Haley · commit b346a97cc59a · 2015-03-26T13:17:52.000-07:00
diff --git a/generate/input/callbacks.json b/generate/input/callbacks.json
@@ -371,9 +371,9 @@
     ],
     "return": {
       "type": "int",
-      "noResults": 1,
+      "noResults": 0,
       "success": 0,
-      "error": -1
+      "error": 1
     }
   },
   "git_repository_mergehead_foreach_cb": {
@@ -389,9 +389,9 @@
     ],
     "return": {
       "type": "int",
-      "noResults": 1,
+      "noResults": 0,
       "success": 0,
-      "error": -1
+      "error": 1
     }
   },
   "git_revwalk_hide_cb": {
diff --git a/generate/input/descriptor.json b/generate/input/descriptor.json
@@ -1456,7 +1456,10 @@
           "ignore": true
         },
         "git_repository_fetchhead_foreach": {
-          "ignore": true
+          "isAsync": true,
+          "return": {
+            "isErrorCode": true
+          }
         },
         "git_repository_hashfile": {
           "ignore": true
diff --git a/lib/repository.js b/lib/repository.js
@@ -843,4 +843,14 @@ Repository.prototype.checkoutBranch = function(branch, opts) {
   });
 };
 
+var fetchheadForeach = Repository.prototype.fetchheadForeach;
+/**
+ * @async
+ * @param {FetchheadForeachCb} callback The callback function to be called on
+ * each entry
+ */
+Repository.prototype.fetchheadForeach = function(callback) {
+  return fetchheadForeach.call(this, callback, null);
+};
+
 module.exports = Repository;
diff --git a/test/runner.js b/test/runner.js
@@ -55,7 +55,9 @@ beforeEach(function() {
 
 afterEach(function(done) {
   process.nextTick(function() {
-    global.gc();
+    if (global.gc) {
+      global.gc();
+    }
     done();
   });
 });
diff --git a/test/tests/repository.js b/test/tests/repository.js
@@ -137,4 +137,36 @@ describe("Repository", function() {
           });
       });
   });
+
+  it("gets fetch-heads", function() {
+    var repo = this.repository;
+    var foundMaster;
+
+    return repo.fetch("origin", {
+      credentials: function(url, userName) {
+        return NodeGit.Cred.sshKeyFromAgent(userName);
+      },
+      certificateCheck: function() {
+        return 1;
+      }
+    })
+    .then(function() {
+      return repo.fetchheadForeach(function(refname, remoteUrl, oid, isMerge) {
+        if (refname == "refs/heads/master") {
+          foundMaster = true;
+          assert.equal(refname, "refs/heads/master");
+          assert.equal(remoteUrl, "https://github.com/nodegit/test");
+          assert.equal(
+            oid.toString(),
+            "32789a79e71fbc9e04d3eff7425e1771eb595150");
+          assert.equal(isMerge, 1);
+        }
+      });
+    })
+    .then(function() {
+      if (!foundMaster) {
+        throw new Error("Couldn't find master in iteration of fetch heads");
+      }
+    });
+  });
 });
