Add another Code Explorer test.

Author: Hosch250

RubberduckTests/CodeExplorer/CodeExplorerTests.cs

@@ -8,6 +8,7 @@
 using Rubberduck.Parsing.VBA;
 using Rubberduck.Settings;
 using Rubberduck.UI.CodeExplorer.Commands;
+using Rubberduck.UnitTesting;
 using Rubberduck.VBEditor.VBEHost;
 using RubberduckTests.Mocks;
 
@@ -75,6 +76,31 @@ public void AddUserForm()
                 vbComponents.Count(c => c.Type == vbext_ComponentType.vbext_ct_MSForm) == 1);
         }
 
+        [TestMethod]
+        public void AddTestModule()
+        {
+            var builder = new MockVbeBuilder();
+            VBComponent component;
+            var vbe = builder.BuildFromSingleStandardModule("", out component);
+            var mockHost = new Mock<IHostApplication>();
+            mockHost.SetupAllProperties();
+
+            var configLoader = new Mock<ConfigurationLoader>(null, null);
+            configLoader.Setup(c => c.LoadConfiguration()).Returns(GetDefaultUnitTestConfig());
+
+            var commands = new List<ICommand>
+            {
+                new CodeExplorer_AddTestModuleCommand(vbe.Object, new NewUnitTestModuleCommand(vbe.Object, configLoader.Object))
+            };
+
+            var vm = new CodeExplorerViewModel(new RubberduckParserState(), commands);
+            vm.AddTestModuleCommand.Execute(null);
+
+            var vbComponents = vbe.Object.VBProjects.Item(0).VBComponents.Cast<VBComponent>();
+
+            Assert.IsTrue(vbComponents.Count(c => c.Type == vbext_ComponentType.vbext_ct_StdModule) == 2);
+        }
+
         #region
         private Configuration GetDefaultUnitTestConfig()
         {

RubberduckTests/Mocks/MockProjectBuilder.cs

@@ -155,6 +155,7 @@ private Mock<References> CreateReferencesMock()
             result.As<IEnumerable>().Setup(m => m.GetEnumerator()).Returns(() => _references.GetEnumerator());
             result.Setup(m => m.Item(It.IsAny<int>())).Returns<int>(index => _references.ElementAt(index - 1));
             result.SetupGet(m => m.Count).Returns(() => _references.Count);
+            result.Setup(m => m.AddFromFile(It.IsAny<string>()));
             return result;
         }
 
@@ -202,6 +203,9 @@ private Mock<CodeModule> CreateCodeModuleMock(string name, string content, Selec
             result.SetupProperty(m => m.Name, name);
 
             codePane.SetupGet(m => m.CodeModule).Returns(() => result.Object);
+
+            result.Setup(m => m.AddFromFile(It.IsAny<string>()));
+            result.Setup(m => m.AddFromString(It.IsAny<string>()));
             return result;
         }