More unit tests

Author: Hosch250

RubberduckTests/SourceControl/BranchesViewModelTests.cs

@@ -761,5 +761,39 @@ public void OnCreateBranch_WhenCreateFails_ActionFailedEventIsRaised()
             //assert
             Assert.IsTrue(wasRaised, "ActionFailedEvent was not raised.");
         }
+
+        [TestMethod]
+        public void PublishPublishesBranch()
+        {
+            //arrange
+            var branch = "dev";
+            var vm = new BranchesViewViewModel
+            {
+                Provider = _provider.Object
+            };
+
+            //act
+            vm.PublishBranchToolbarButtonCommand.Execute(branch);
+
+            //Assert
+            _provider.Verify(git => git.Publish(branch));
+        }
+
+        [TestMethod]
+        public void UnpublishUnpublishesBranch()
+        {
+            //arrange
+            var branch = "master";
+            var vm = new BranchesViewViewModel
+            {
+                Provider = _provider.Object
+            };
+
+            //act
+            vm.UnpublishBranchToolbarButtonCommand.Execute(branch);
+
+            //Assert
+            _provider.Verify(git => git.Unpublish(branch));
+        }
     }
 }

RubberduckTests/SourceControl/ChangesViewModelTests.cs

@@ -260,8 +260,6 @@ public void ExcludeFileExcludesFile()
             Assert.AreEqual(1, vm.ExcludedChanges.Count);
         }
 
-        // I need to figure out how to make this throw.
-        [Ignore]
         [TestMethod]
         public void ChangesPresenter_WhenCommitFails_ActionFailedEventIsRaised()
         {
@@ -278,6 +276,12 @@ public void ChangesPresenter_WhenCommitFails_ActionFailedEventIsRaised()
                     }
             };
 
+            _providerMock.Setup(p => p.Commit(It.IsAny<string>()))
+                .Throws(
+                    new SourceControlException("A source control exception was thrown.",
+                        new LibGit2Sharp.LibGit2SharpException("With an inner libgit2sharp exception"))
+                    );
+
             var wasRaised = false;
 
             vm.ErrorThrown += (e, sender) => wasRaised = true;
@@ -288,5 +292,35 @@ public void ChangesPresenter_WhenCommitFails_ActionFailedEventIsRaised()
             //assert
             Assert.IsTrue(wasRaised, "ActionFailedEvent was not raised.");
         }
+
+        [TestMethod]
+        public void UndoUndoesChanges()
+        {
+            //arrange
+            var fileStatusEntries = new List<FileStatusEntry>
+                    {
+                        new FileStatusEntry(@"C:\path\to\module.bas", FileStatus.Modified),
+                        new FileStatusEntry(@"C:\path\to\class.cls", FileStatus.Unaltered),
+                        new FileStatusEntry(@"C:\path\to\added.bas", FileStatus.Added | FileStatus.Modified),
+                        new FileStatusEntry(@"C:\path\to\addedUnmodified.bas", FileStatus.Added),
+                        new FileStatusEntry(@"C:\path\to\untracked.frx", FileStatus.Untracked)
+                    };
+
+            var vm = new ChangesViewViewModel
+            {
+                Provider = _providerMock.Object
+            };
+
+            var localLocation = "C:\\users\\desktop\\git\\";
+
+            _providerMock.Setup(git => git.Status()).Returns(fileStatusEntries);
+            _providerMock.SetupGet(git => git.CurrentRepository).Returns(new Repository{LocalLocation = localLocation});
+
+            //act
+            vm.UndoChangesToolbarButtonCommand.Execute(fileStatusEntries[0]);
+
+            //Assert
+            _providerMock.Verify(git => git.Undo(localLocation + fileStatusEntries[0].FilePath));
+        }
     }
 }