Merge branch 'Issue1830' of https://github.com/Hosch250/Rubberduck

Author: Hosch250

RetailCoder.VBE/UI/Command/RunAllTestsCommand.cs

@@ -1,5 +1,7 @@
 using System;
 using System.Diagnostics;
+using System.Linq;
+using Microsoft.Vbe.Interop;
 using Rubberduck.Parsing.VBA;
 using Rubberduck.UI.UnitTesting;
 using Rubberduck.UnitTesting;
@@ -11,17 +13,24 @@ namespace Rubberduck.UI.Command
     /// </summary>
     public class RunAllTestsCommand : CommandBase
     {
+        private readonly VBE _vbe;
         private readonly ITestEngine _engine;
         private readonly TestExplorerModel _model;
         private readonly RubberduckParserState _state;
 
-        public RunAllTestsCommand(RubberduckParserState state, ITestEngine engine, TestExplorerModel model)
+        public RunAllTestsCommand(VBE vbe, RubberduckParserState state, ITestEngine engine, TestExplorerModel model)
         {
+            _vbe = vbe;
             _engine = engine;
             _model = model;
             _state = state;
         }
 
+        public override bool CanExecute(object parameter)
+        {
+            return _vbe.VBProjects.Cast<VBProject>().All(project => project.Mode == vbext_VBAMode.vbext_vm_Design);
+        }
+
         public override void Execute(object parameter)
         {
             if (!_state.IsDirty())

RetailCoder.VBE/UI/UnitTesting/TestExplorerViewModel.cs

@@ -18,6 +18,7 @@ namespace Rubberduck.UI.UnitTesting
 {
     public class TestExplorerViewModel : ViewModelBase, INavigateSelection, IDisposable
     {
+        private readonly VBE _vbe;
         private readonly RubberduckParserState _state;
         private readonly ITestEngine _testEngine;
         private readonly TestExplorerModel _model;
@@ -35,6 +36,7 @@ public TestExplorerViewModel(VBE vbe,
              IGeneralConfigService configService,
              IOperatingSystem operatingSystem)
         {
+            _vbe = vbe;
             _state = state;
             _testEngine = testEngine;
             _testEngine.TestCompleted += TestEngineTestCompleted;
@@ -45,7 +47,7 @@ public TestExplorerViewModel(VBE vbe,
 
             _navigateCommand = new NavigateCommand();
 
-            _runAllTestsCommand = new RunAllTestsCommand(state, testEngine, model);
+            _runAllTestsCommand = new RunAllTestsCommand(vbe, state, testEngine, model);
             _runAllTestsCommand.RunCompleted += RunCompleted;
 
             _addTestModuleCommand = new AddTestModuleCommand(vbe, state, newTestModuleCommand);
@@ -83,22 +85,26 @@ private void RunCompleted(object sender, TestRunEventArgs e)
 
         private bool CanExecuteRunPassedTestsCommand(object obj)
         {
-            return _model.Tests.Any(test => test.Result.Outcome == TestOutcome.Succeeded);
+            return _vbe.VBProjects.Cast<VBProject>().All(project => project.Mode == vbext_VBAMode.vbext_vm_Design) &&
+                   _model.Tests.Any(test => test.Result.Outcome == TestOutcome.Succeeded);
         }
 
         private bool CanExecuteRunFailedTestsCommand(object obj)
         {
-            return _model.Tests.Any(test => test.Result.Outcome == TestOutcome.Failed);
+            return _vbe.VBProjects.Cast<VBProject>().All(project => project.Mode == vbext_VBAMode.vbext_vm_Design) &&
+                   _model.Tests.Any(test => test.Result.Outcome == TestOutcome.Failed);
         }
 
         private bool CanExecuteRunNotExecutedTestsCommand(object obj)
         {
-            return _model.Tests.Any(test => test.Result.Outcome == TestOutcome.Unknown);
+            return _vbe.VBProjects.Cast<VBProject>().All(project => project.Mode == vbext_VBAMode.vbext_vm_Design) &&
+                   _model.Tests.Any(test => test.Result.Outcome == TestOutcome.Unknown);
         }
 
         private bool CanExecuteRepeatLastRunCommand(object obj)
         {
-            return _model.LastRun.Any();
+            return _vbe.VBProjects.Cast<VBProject>().All(project => project.Mode == vbext_VBAMode.vbext_vm_Design) &&
+                   _model.LastRun.Any();
         }
 
         public event EventHandler<EventArgs> TestCompleted;