Added initial TestModuleWithStubsCommand to Code Explorer

Author: rkapka

RetailCoder.VBE/Navigation/CodeExplorer/CodeExplorerViewModel.cs

@@ -64,6 +64,8 @@ public CodeExplorerViewModel(FolderHelper folderHelper, RubberduckParserState st
             OpenDesignerCommand = commands.OfType<OpenDesignerCommand>().SingleOrDefault();
 
             AddTestModuleCommand = commands.OfType<UI.CodeExplorer.Commands.AddTestModuleCommand>().SingleOrDefault();
+            AddTestModuleWithStubsCommand = commands.OfType<AddTestModuleWithStubsCommand>().SingleOrDefault();
+
             AddStdModuleCommand = commands.OfType<AddStdModuleCommand>().SingleOrDefault();
             AddClassModuleCommand = commands.OfType<AddClassModuleCommand>().SingleOrDefault();
             AddUserFormCommand = commands.OfType<AddUserFormCommand>().SingleOrDefault();
@@ -491,6 +493,7 @@ private void SwitchNodeState(CodeExplorerItemViewModel node, bool expandedState)
         public CommandBase OpenCommand { get; }
 
         public CommandBase AddTestModuleCommand { get; }
+        public CommandBase AddTestModuleWithStubsCommand { get; }
         public CommandBase AddStdModuleCommand { get; }
         public CommandBase AddClassModuleCommand { get; }
         public CommandBase AddUserFormCommand { get; }

RetailCoder.VBE/Rubberduck.csproj

@@ -395,6 +395,7 @@
     <Compile Include="UI\About\AboutDialog.Designer.cs">
       <DependentUpon>AboutDialog.cs</DependentUpon>
     </Compile>
+    <Compile Include="UI\CodeExplorer\Commands\AddTestModuleWithStubsCommand.cs" />
     <Compile Include="UI\CodeExplorer\Commands\CodeExplorerCommandAttribute.cs" />
     <Compile Include="UI\CodeExplorer\Commands\CommitCommand.cs" />
     <Compile Include="UI\CodeExplorer\Commands\AddUserFormCommand.cs" />

RetailCoder.VBE/UI/CodeExplorer/CodeExplorerControl.xaml

@@ -25,6 +25,7 @@
             <BitmapImage x:Key="UndoImage" UriSource="../../Resources/arrow-circle-left.png" />
             <BitmapImage x:Key="PrintImage" UriSource="../../Resources/printer.png" />
             <BitmapImage x:Key="AddTestModuleImage" UriSource="../../Resources/Custom/PNG/ListOfTests.png" />
+            <BitmapImage x:Key="AddTestModuleWithStubsImage" UriSource="../../Resources/Custom/PNG/ListOfTests.png" />
             <BitmapImage x:Key="AddStdModuleImage" UriSource="../../Resources/Custom/PNG/AddModule.png" />
             <BitmapImage x:Key="AddClassModuleImage" UriSource="../../Resources/Custom/PNG/AddClass.png" />
             <BitmapImage x:Key="AddUserFormImage" UriSource="../../Resources/Custom/PNG/AddForm.png" />
@@ -161,6 +162,13 @@
                                         <Image Height="16" Source="{StaticResource AddTestModuleImage}" />
                                     </MenuItem.Icon>
                                 </MenuItem>
+                                <MenuItem Header="{Resx ResxName=Rubberduck.UI.RubberduckUI, Key=CodeExplorer_AddTestModuleWithStubsText}"
+                                  Command="{Binding AddTestModuleWithStubsCommand}"
+                                  CommandParameter="{Binding SelectedItem}">
+                                    <MenuItem.Icon>
+                                        <Image Height="16" Source="{StaticResource AddTestModuleWithStubsImage}" />
+                                    </MenuItem.Icon>
+                                </MenuItem>
                                 <Separator />
                                 <MenuItem Header="{Resx ResxName=Rubberduck.UI.RubberduckUI, Key=CodeExplorer_AddStdModuleText}"
                                   Command="{Binding AddStdModuleCommand}"

RetailCoder.VBE/UI/CodeExplorer/Commands/AddTestModuleWithStubsCommand.cs

@@ -0,0 +1,57 @@
+using System.Runtime.InteropServices;
+using NLog;
+using Rubberduck.Navigation.CodeExplorer;
+using Rubberduck.Parsing.Symbols;
+using Rubberduck.UI.Command;
+using Rubberduck.VBEditor.SafeComWrappers.Abstract;
+
+namespace Rubberduck.UI.CodeExplorer.Commands
+{
+    [CodeExplorerCommand]
+    public class AddTestModuleWithStubsCommand : CommandBase
+    {
+        private readonly IVBE _vbe;
+        private readonly Command.AddTestModuleCommand _newUnitTestModuleCommand;
+
+        public AddTestModuleWithStubsCommand(IVBE vbe, Command.AddTestModuleCommand newUnitTestModuleCommand) : base(LogManager.GetCurrentClassLogger())
+        {
+            _vbe = vbe;
+            _newUnitTestModuleCommand = newUnitTestModuleCommand;
+        }
+
+        protected override bool EvaluateCanExecute(object parameter)
+        {
+            try
+            {
+                return GetDeclaration(parameter) != null || _vbe.VBProjects.Count == 1;
+            }
+            catch (COMException)
+            {
+                return false;
+            }
+        }
+
+        protected override void OnExecute(object parameter)
+        {
+            if (parameter != null)
+            {
+                _newUnitTestModuleCommand.Execute(GetDeclaration(parameter));
+            }
+            else
+            {
+                _newUnitTestModuleCommand.Execute(_vbe.ActiveVBProject);
+            }
+        }
+
+        private Declaration GetDeclaration(object parameter)
+        {
+            var node = parameter as CodeExplorerItemViewModel;
+            while (node != null && !(node is ICodeExplorerDeclarationViewModel))
+            {
+                node = node.Parent;
+            }
+
+            return ((ICodeExplorerDeclarationViewModel)node)?.Declaration;
+        }
+    }
+}

RetailCoder.VBE/UI/RubberduckUI.Designer.cs

@@ -2100,4 +2100,8 @@ the projects in the VBE.</value>
   <data name="EmptyUIRefreshMessage_Title" xml:space="preserve">
     <value>Rubberduck doesn't see anything yet.</value>
   </data>
+  <data name="CodeExplorer_AddTestModuleWithStubsText" xml:space="preserve">
+    <value>Test module with stubs</value>
+    <comment>TC</comment>
+  </data>
 </root>

RetailCoder.VBE/UI/RubberduckUI.resx

@@ -158,6 +158,40 @@ public void AddTestModule()
             components.Verify(c => c.Add(ComponentType.StandardModule), Times.Once);
         }
 
+        [TestCategory("Code Explorer")]
+        [TestMethod]
+        public void AddTestModuleWithStubs()
+        {
+            var builder = new MockVbeBuilder();
+            var project = builder.ProjectBuilder("TestProject1", ProjectProtection.Unprotected)
+                .AddComponent("Module1", ComponentType.StandardModule, "");
+
+            var components = project.MockVBComponents;
+
+            var vbe = builder.AddProject(project.Build()).Build();
+
+            var configLoader = new Mock<ConfigurationLoader>(null, null, null, null, null, null, null);
+            configLoader.Setup(c => c.LoadConfiguration()).Returns(GetDefaultUnitTestConfig());
+
+            var state = new RubberduckParserState(vbe.Object, new DeclarationFinderFactory());
+            var vbeWrapper = vbe.Object;
+            var commands = new List<CommandBase>
+            {
+                new AddTestModuleWithStubsCommand(vbeWrapper, new Rubberduck.UI.Command.AddTestModuleCommand(vbeWrapper, state, configLoader.Object))
+            };
+            
+            var vm = new CodeExplorerViewModel(new FolderHelper(state), state, commands, _generalSettingsProvider.Object, _windowSettingsProvider.Object);
+
+            var parser = MockParser.Create(vbe.Object, state);
+            parser.Parse(new CancellationTokenSource());
+            if (parser.State.Status >= ParserState.Error) { Assert.Inconclusive("Parser Error"); }
+
+            vm.SelectedItem = vm.Projects.First().Items.First().Items.First();
+            vm.AddTestModuleWithStubsCommand.Execute(vm.SelectedItem);
+
+            components.Verify(c => c.Add(ComponentType.StandardModule), Times.Once);
+        }
+
         [TestCategory("Code Explorer")]
         [TestMethod]
         public void ImportModule()