Add test for sm_branch_only

1 files changed, 29 insertions, 1 deletions

diff --git a/LibGit2Sharp.Tests/SubmoduleFixture.cs b/LibGit2Sharp.Tests/SubmoduleFixture.cs
index d60e61f3..58c8a830 100644
--- a/LibGit2Sharp.Tests/SubmoduleFixture.cs
+++ b/LibGit2Sharp.Tests/SubmoduleFixture.cs
@@ -20,6 +20,26 @@ namespace LibGit2Sharp.Tests
             }
         }
 
+        [Fact]
+        public void RetrievingSubmoduleInBranchShouldWork()
+        {
+            var path = SandboxSubmoduleTestRepo();
+            using (var repo = new Repository(path))
+            {
+                var submodule = repo.Submodules["sm_branch_only"];
+                Assert.Null(submodule);
+
+                repo.Checkout("dev", new CheckoutOptions { CheckoutModifiers = CheckoutModifiers.Force });
+                submodule = repo.Submodules["sm_branch_only"];
+                Assert.NotNull(submodule);
+                Assert.NotEqual(SubmoduleStatus.Unmodified, submodule.RetrieveStatus());
+
+                repo.Checkout("master", new CheckoutOptions { CheckoutModifiers = CheckoutModifiers.Force });
+                submodule = repo.Submodules["sm_branch_only"];
+                Assert.Null(submodule);
+            }
+        }
+
         [Theory]
         [InlineData("sm_added_and_uncommited", SubmoduleStatus.InConfig | SubmoduleStatus.InIndex | SubmoduleStatus.InWorkDir | SubmoduleStatus.IndexAdded)]
         [InlineData("sm_changed_file", SubmoduleStatus.InConfig | SubmoduleStatus.InHead | SubmoduleStatus.InIndex | SubmoduleStatus.InWorkDir | SubmoduleStatus.WorkDirFilesModified)]
@@ -29,12 +49,20 @@ namespace LibGit2Sharp.Tests
         [InlineData("sm_gitmodules_only", SubmoduleStatus.InConfig)]
         [InlineData("sm_missing_commits", SubmoduleStatus.InConfig | SubmoduleStatus.InHead | SubmoduleStatus.InIndex | SubmoduleStatus.InWorkDir | SubmoduleStatus.WorkDirModified)]
         [InlineData("sm_unchanged", SubmoduleStatus.InConfig | SubmoduleStatus.InHead | SubmoduleStatus.InIndex | SubmoduleStatus.InWorkDir)]
-        public void CanRetrieveTheStatusOfASubmodule(string name, SubmoduleStatus expectedStatus)
+        [InlineData("sm_branch_only", null)]
+        public void CanRetrieveTheStatusOfASubmodule(string name, SubmoduleStatus? expectedStatus)
         {
             var path = SandboxSubmoduleTestRepo();
             using (var repo = new Repository(path))
             {
                 var submodule = repo.Submodules[name];
+
+                if (expectedStatus == null)
+                {
+                    Assert.Null(submodule);
+                    return;
+                }
+
                 Assert.NotNull(submodule);
                 Assert.Equal(name, submodule.Name);
                 Assert.Equal(name, submodule.Path);