Expose list of components mocks. Export test.

Author: Hosch250

RubberduckTests/CodeExplorer/CodeExplorerTests.cs

@@ -155,7 +155,7 @@ public void ImportModule()
             vbComponents.Verify(c => c.Import("C:\\Users\\Rubberduck\\Desktop\\StdModule1.bas"), Times.Once);
         }
 
-        /*[TestMethod]
+        [TestMethod]
         public void ExportModule()
         {
             var builder = new MockVbeBuilder();
@@ -164,7 +164,7 @@ public void ExportModule()
 
             var project = projectMock.Build();
             var vbe = builder.AddProject(project).Build();
-            var component = projectMock.Components.First();
+            var component = projectMock.MockComponents.First();
 
             var saveFileDialog = new Mock<ISaveFileDialog>();
             saveFileDialog.Setup(o => o.OverwritePrompt);
@@ -187,7 +187,7 @@ public void ExportModule()
             vm.ExportCommand.Execute(vm.SelectedItem);
 
             component.Verify(c => c.Export("C:\\Users\\Rubberduck\\Desktop\\StdModule1.bas"), Times.Once);
-        }*/
+        }
 
         [TestMethod]
         public void RemoveModule_Cancel()

RubberduckTests/Mocks/MockProjectBuilder.cs

@@ -20,14 +20,25 @@ public class MockProjectBuilder
         private readonly Mock<VBComponents> _vbComponents;
         private readonly Mock<References> _vbReferences;
 
-        private readonly List<VBComponent> _components = new List<VBComponent>();
+        private readonly List<Mock<VBComponent>> _componentsMock = new List<Mock<VBComponent>>();
+        //private readonly List<VBComponent> _components = new List<VBComponent>();
         private readonly List<Reference> _references = new List<Reference>();
 
         public Mock<VBComponents> MockVBComponents
         {
             get { return _vbComponents; }
         }
 
+        public List<Mock<VBComponent>> MockComponents
+        {
+            get { return _componentsMock; }
+        }
+
+        private List<VBComponent> Components
+        {
+            get { return _componentsMock.Select(m => m.Object).ToList(); }
+        }
+
         public MockProjectBuilder(string name, string filename, vbext_ProjectProtection protection, Func<VBE> getVbe, MockVbeBuilder mockVbeBuilder)
         {
             _getVbe = getVbe;
@@ -67,7 +78,7 @@ public MockProjectBuilder(string name, string filename, vbext_ProjectProtection
         /// <returns>Returns the <see cref="MockProjectBuilder"/> instance.</returns>
         public MockProjectBuilder AddComponent(Mock<VBComponent> component)
         {
-            _components.Add(component.Object);
+            _componentsMock.Add(component);
             _getVbe().ActiveCodePane = component.Object.CodeModule.CodePane;
             return this;
         }
@@ -136,21 +147,21 @@ private Mock<VBComponents> CreateComponentsMock()
             result.SetupGet(m => m.Parent).Returns(() => _project.Object);
             result.SetupGet(m => m.VBE).Returns(_getVbe);
 
-            result.Setup(c => c.GetEnumerator()).Returns(() => _components.GetEnumerator());
-            result.As<IEnumerable>().Setup(c => c.GetEnumerator()).Returns(() => _components.GetEnumerator());
+            result.Setup(c => c.GetEnumerator()).Returns(() => Components.GetEnumerator());
+            result.As<IEnumerable>().Setup(c => c.GetEnumerator()).Returns(() => Components.GetEnumerator());
 
-            result.Setup(m => m.Item(It.IsAny<int>())).Returns<int>(index => _components.ElementAt(index));
-            result.Setup(m => m.Item(It.IsAny<string>())).Returns<string>(name => _components.Single(item => item.Name == name));
-            result.SetupGet(m => m.Count).Returns(_components.Count);
+            result.Setup(m => m.Item(It.IsAny<int>())).Returns<int>(index => Components.ElementAt(index));
+            result.Setup(m => m.Item(It.IsAny<string>())).Returns<string>(name => Components.Single(item => item.Name == name));
+            result.SetupGet(m => m.Count).Returns(Components.Count);
 
             result.Setup(m => m.Add(It.IsAny<vbext_ComponentType>())).Callback((vbext_ComponentType c) =>
             {
-                _components.Add(CreateComponentMock("test", c, string.Empty, new Selection()).Object);
+                _componentsMock.Add(CreateComponentMock("test", c, string.Empty, new Selection()));
             });
 
             result.Setup(m => m.Remove(It.IsAny<VBComponent>())).Callback((VBComponent c) =>
             {
-                _components.Remove(c);
+                _componentsMock.Remove(_componentsMock.First(m => m.Object == c));
             });
 
             result.Setup(m => m.Import(It.IsAny<string>())).Callback((string s) =>
@@ -166,7 +177,7 @@ private Mock<VBComponents> CreateComponentsMock()
                 vbext_ComponentType type;
                 types.TryGetValue(parts.Last(), out type);
 
-                _components.Add(CreateComponentMock(s.Split('\\').Last(), type, string.Empty, new Selection()).Object);
+                _componentsMock.Add(CreateComponentMock(s.Split('\\').Last(), type, string.Empty, new Selection()));
             });
 
             return result;