Add TreeDefinition.Add(Submodule)

dahlbyk · dahlbyk · commit d5c4a4461a15 · 2013-04-24T15:08:50.000-05:00
diff --git a/LibGit2Sharp.Tests/TreeDefinitionFixture.cs b/LibGit2Sharp.Tests/TreeDefinitionFixture.cs
@@ -157,6 +157,33 @@ public void CanAddAnExistingBlob(string blobSha, string targetPath)
             }
         }
 
+        [Fact]
+        public void CanAddAnExistingSubmodule()
+        {
+            const string submodulePath = "sm_unchanged";
+
+            using (var repo = new Repository(SubmoduleTestRepoWorkingDirPath))
+            {
+                var submodule = repo.Submodules[submodulePath];
+                Assert.NotNull(submodule);
+
+                TreeDefinition td = TreeDefinition.From(repo.Head.Tip.Tree);
+                Assert.NotNull(td[submodulePath]);
+
+                td.Remove(submodulePath);
+                Assert.Null(td[submodulePath]);
+
+                td.Add(submodule);
+
+                TreeEntryDefinition fetched = td[submodulePath];
+                Assert.NotNull(fetched);
+
+                Assert.Equal(submodule.HeadCommitId, fetched.TargetId);
+                Assert.Equal(GitObjectType.Commit, fetched.Type);
+                Assert.Equal(Mode.GitLink, fetched.Mode);
+            }
+        }
+
         [Fact]
         public void CanAddAnExistingTree()
         {
diff --git a/LibGit2Sharp/TreeDefinition.cs b/LibGit2Sharp/TreeDefinition.cs
@@ -174,6 +174,18 @@ public virtual TreeDefinition Add(string targetTreeEntryPath, Tree tree)
             return Add(targetTreeEntryPath, ted);
         }
 
+        /// <summary>
+        ///   Adds or replaces a gitlink <see cref="TreeEntryDefinition"/> equivalent to <paramref name="submodule"/>.
+        /// </summary>
+        /// <param name="submodule">The <see cref="Submodule"/> to be linked.</param>
+        /// <returns>The current <see cref="TreeDefinition"/>.</returns>
+        public virtual TreeDefinition Add(Submodule submodule)
+        {
+            Ensure.ArgumentNotNull(submodule, "submodule");
+
+            return AddGitLink(submodule.Path, submodule.HeadCommitId);
+        }
+
         /// <summary>
         ///   Adds or replaces a gitlink <see cref="TreeEntryDefinition"/>,
         ///   referencing the commit identified by <paramref name="objectId"/>,
