@@ -250,11 +250,10 @@ private Mock<ICodeModule> CreateCodeModuleMock(string name, string content, Sele
250
250
var codePane = CreateCodePaneMock ( name , selection , component ) ;
251
251
codePane . SetupGet ( m => m . VBE ) . Returns ( _getVbe ) ;
252
252
253
- var result = CreateCodeModuleMock ( content ) ;
253
+ var result = CreateCodeModuleMock ( content , name ) ;
254
254
result . SetupGet ( m => m . VBE ) . Returns ( _getVbe ) ;
255
255
result . SetupGet ( m => m . Parent ) . Returns ( ( ) => component . Object ) ;
256
256
result . SetupGet ( m => m . CodePane ) . Returns ( ( ) => codePane . Object ) ;
257
- result . SetupProperty ( m => m . Name , name ) ;
258
257
259
258
codePane . SetupGet ( m => m . CodeModule ) . Returns ( ( ) => result . Object ) ;
260
259
@@ -267,7 +266,7 @@ private Mock<ICodeModule> CreateCodeModuleMock(string name, string content, Sele
267
266
Tokens . Sub + ' ' , Tokens . Function + ' ' , Tokens . Property + ' '
268
267
} ;
269
268
270
- private Mock < ICodeModule > CreateCodeModuleMock ( string content )
269
+ private Mock < ICodeModule > CreateCodeModuleMock ( string content , string name )
271
270
{
272
271
var lines = content . Split ( new [ ] { Environment . NewLine } , StringSplitOptions . None ) . ToList ( ) ;
273
272
@@ -312,8 +311,10 @@ private Mock<ICodeModule> CreateCodeModuleMock(string content)
312
311
lines . AddRange ( newLine . Split ( new [ ] { Environment . NewLine } , StringSplitOptions . None ) ) ;
313
312
} ) ;
314
313
314
+ codeModule . SetupProperty ( m => m . Name , name ) ;
315
+
315
316
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 ( ) ) ) ;
317
318
codeModule . Setup ( m => m . GetHashCode ( ) ) . Returns ( ( ) => codeModule . Object . Target . GetHashCode ( ) ) ;
318
319
319
320
return codeModule ;
@@ -331,8 +332,10 @@ private Mock<ICodePane> CreateCodePaneMock(string name, Selection selection, Moc
331
332
var window = windows . CreateWindow ( name ) ;
332
333
windows . Add ( window ) ;
333
334
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
+ } ) ;
336
339
codePane . SetupProperty ( p => p . Selection , selection ) ;
337
340
codePane . Setup ( p => p . Show ( ) ) ;
338
341
0 commit comments