Merge pull request #1658 from rcjsuen/getTree-fix

implausible · web-flow · commit 0b3cdc12c5ae · 2019-04-17T12:19:50.000-07:00
Ensures that commits from parent(*) has a repository
diff --git a/lib/commit.js b/lib/commit.js
@@ -5,6 +5,7 @@ var Commit = NodeGit.Commit;
 var LookupWrapper = NodeGit.Utils.lookupWrapper;
 
 var _amend = Commit.prototype.amend;
+var _parent = Commit.prototype.parent;
 
 /**
  * Retrieves the commit pointed to by the oid
@@ -394,6 +395,21 @@ Commit.prototype.history = function() {
   return event;
 };
 
+/**
+ * Get the specified parent of the commit.
+ * 
+ * @param {number} the position of the parent, starting from 0
+ * @async
+ * @return {Commit} the parent commit at the specified position
+ */
+Commit.prototype.parent = function (id) {
+  var repository = this.repo;
+  return _parent.call(this, id).then(function(parent) {
+    parent.repo = repository;
+    return parent;
+  });
+};
+
 /**
  * Retrieve the commit's parent shas.
  *
diff --git a/test/tests/commit.js b/test/tests/commit.js
@@ -128,6 +128,18 @@ describe("Commit", function() {
     assert.equal(this.commit.timeOffset(), 780);
   });
 
+  it("can call getTree on a parent commit", function() {
+    return this.commit.parent(0)
+    .then(function(parent) {
+      return parent.getTree();
+    })
+    .then(function(tree) {
+      assert.equal(
+        tree.id().toString(), "327ff68e59f94f0c25d2c62fb0938efa01e8a107"
+      );
+    });
+  });
+
   it("can create a commit", function() {
     var test = this;
     var expectedCommitId = "315e77328ef596f3bc065d8ac6dd2c72c09de8a5";
