Skip to content

Commit fc4db39

Browse files
committed
Merge branch 'next' of https://github.com/rubberduck-vba/Rubberduck into next
2 parents 44176be + d028b9c commit fc4db39

File tree

1 file changed

+9
-6
lines changed

1 file changed

+9
-6
lines changed

RubberduckTests/Mocks/MockProjectBuilder.cs

Lines changed: 9 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -250,11 +250,10 @@ private Mock<ICodeModule> CreateCodeModuleMock(string name, string content, Sele
250250
var codePane = CreateCodePaneMock(name, selection, component);
251251
codePane.SetupGet(m => m.VBE).Returns(_getVbe);
252252

253-
var result = CreateCodeModuleMock(content);
253+
var result = CreateCodeModuleMock(content, name);
254254
result.SetupGet(m => m.VBE).Returns(_getVbe);
255255
result.SetupGet(m => m.Parent).Returns(() => component.Object);
256256
result.SetupGet(m => m.CodePane).Returns(() => codePane.Object);
257-
result.SetupProperty(m => m.Name, name);
258257

259258
codePane.SetupGet(m => m.CodeModule).Returns(() => result.Object);
260259

@@ -267,7 +266,7 @@ private Mock<ICodeModule> CreateCodeModuleMock(string name, string content, Sele
267266
Tokens.Sub + ' ', Tokens.Function + ' ', Tokens.Property + ' '
268267
};
269268

270-
private Mock<ICodeModule> CreateCodeModuleMock(string content)
269+
private Mock<ICodeModule> CreateCodeModuleMock(string content, string name)
271270
{
272271
var lines = content.Split(new[] { Environment.NewLine }, StringSplitOptions.None).ToList();
273272

@@ -312,8 +311,10 @@ private Mock<ICodeModule> CreateCodeModuleMock(string content)
312311
lines.AddRange(newLine.Split(new[] { Environment.NewLine }, StringSplitOptions.None));
313312
});
314313

314+
codeModule.SetupProperty(m => m.Name, name);
315+
315316
codeModule.Setup(m => m.Equals(It.IsAny<ICodeModule>()))
316-
.Returns((ICodeModule other) => ReferenceEquals(codeModule.Object.Target, other.Target));
317+
.Returns((ICodeModule other) => name.Equals(other.Name) && content.Equals(other.Content()));
317318
codeModule.Setup(m => m.GetHashCode()).Returns(() => codeModule.Object.Target.GetHashCode());
318319

319320
return codeModule;
@@ -331,8 +332,10 @@ private Mock<ICodePane> CreateCodePaneMock(string name, Selection selection, Moc
331332
var window = windows.CreateWindow(name);
332333
windows.Add(window);
333334

334-
codePane.Setup(p => p.GetQualifiedSelection()).Returns(() =>
335-
new QualifiedSelection(new QualifiedModuleName(component.Object), selection));
335+
codePane.Setup(p => p.GetQualifiedSelection()).Returns(() => {
336+
if (selection.IsEmpty()) { return null; }
337+
return new QualifiedSelection(new QualifiedModuleName(component.Object), selection);
338+
});
336339
codePane.SetupProperty(p => p.Selection, selection);
337340
codePane.Setup(p => p.Show());
338341

0 commit comments

Comments
 (0)