Merge branch 'next' into Issue2135

Author: Hosch250

RetailCoder.VBE/UI/Command/RunAllTestsCommand.cs

@@ -37,6 +37,8 @@ protected override bool CanExecuteImpl(object parameter)
 
         protected override void ExecuteImpl(object parameter)
         {
+            EnsureRubberduckIsReferencedForEarlyBoundTests();
+
             if (!_state.IsDirty())
             {
                 RunTests();
@@ -48,6 +50,18 @@ protected override void ExecuteImpl(object parameter)
             }
         }
 
+        private void EnsureRubberduckIsReferencedForEarlyBoundTests()
+        {
+            foreach (var member in _state.AllUserDeclarations)
+            {
+                if (member.AsTypeName == "Rubberduck.PermissiveAssertClass" ||
+                    member.AsTypeName == "Rubberduck.AssertClass")
+                {
+                    member.Project.EnsureReferenceToAddInLibrary();
+                }
+            }
+        }
+
         private void TestsRefreshed(object sender, EventArgs e)
         {
             RunTests();

RetailCoder.VBE/UI/UnitTesting/TestExplorerViewModel.cs

@@ -228,8 +228,22 @@ private bool CanExecuteRefreshCommand(object parameter)
             return !Model.IsBusy && _state.IsDirty();
         }
 
+        private void EnsureRubberduckIsReferencedForEarlyBoundTests()
+        {
+            foreach (var member in _state.AllUserDeclarations)
+            {
+                if (member.AsTypeName == "Rubberduck.PermissiveAssertClass" ||
+                    member.AsTypeName == "Rubberduck.AssertClass")
+                {
+                    member.Project.EnsureReferenceToAddInLibrary();
+                }
+            }
+        }
+
         private void ExecuteRepeatLastRunCommand(object parameter)
         {
+            EnsureRubberduckIsReferencedForEarlyBoundTests();
+
             var tests = _model.LastRun.ToList();
             _model.ClearLastRun();
 
@@ -246,6 +260,8 @@ private void ExecuteRepeatLastRunCommand(object parameter)
 
         private void ExecuteRunNotExecutedTestsCommand(object parameter)
         {
+            EnsureRubberduckIsReferencedForEarlyBoundTests();
+
             _model.ClearLastRun();
 
             var stopwatch = new Stopwatch();
@@ -261,6 +277,8 @@ private void ExecuteRunNotExecutedTestsCommand(object parameter)
 
         private void ExecuteRunFailedTestsCommand(object parameter)
         {
+            EnsureRubberduckIsReferencedForEarlyBoundTests();
+
             _model.ClearLastRun();
 
             var stopwatch = new Stopwatch();
@@ -276,6 +294,8 @@ private void ExecuteRunFailedTestsCommand(object parameter)
 
         private void ExecuteRunPassedTestsCommand(object parameter)
         {
+            EnsureRubberduckIsReferencedForEarlyBoundTests();
+
             _model.ClearLastRun();
 
             var stopwatch = new Stopwatch();
@@ -301,6 +321,8 @@ private void ExecuteSelectedTestCommand(object obj)
                 return;
             }
 
+            EnsureRubberduckIsReferencedForEarlyBoundTests();
+
             _model.ClearLastRun();
 
             var stopwatch = new Stopwatch();