@@ -165,7 +165,7 @@ public async Task FindsFunctionDefinition()
165
165
{
166
166
SymbolReference symbol = await GetDefinition ( FindsFunctionDefinitionData . SourceDetails ) . ConfigureAwait ( true ) ;
167
167
Assert . Equal ( "My-Function" , symbol . SymbolName ) ;
168
- Assert . Equal ( "function My-Function($myInput)" , symbol . DisplayString ) ;
168
+ Assert . Equal ( "function My-Function ($myInput)" , symbol . DisplayString ) ;
169
169
Assert . Equal ( SymbolType . Function , symbol . SymbolType ) ;
170
170
AssertIsRegion ( symbol . NameRegion , 1 , 10 , 1 , 21 ) ;
171
171
AssertIsRegion ( symbol . ScriptRegion , 1 , 1 , 4 , 2 ) ;
@@ -182,7 +182,7 @@ await psesHost.ExecutePSCommandAsync(
182
182
CancellationToken . None ) . ConfigureAwait ( true ) ;
183
183
184
184
SymbolReference symbol = await GetDefinition ( FindsFunctionDefinitionOfAliasData . SourceDetails ) . ConfigureAwait ( true ) ;
185
- Assert . Equal ( "function My-Function($myInput)" , symbol . DisplayString ) ;
185
+ Assert . Equal ( "function My-Function ($myInput)" , symbol . DisplayString ) ;
186
186
Assert . Equal ( SymbolType . Function , symbol . SymbolType ) ;
187
187
AssertIsRegion ( symbol . NameRegion , 1 , 10 , 1 , 21 ) ;
188
188
AssertIsRegion ( symbol . ScriptRegion , 1 , 1 , 4 , 2 ) ;
@@ -193,10 +193,17 @@ await psesHost.ExecutePSCommandAsync(
193
193
public async Task FindsReferencesOnFunction ( )
194
194
{
195
195
IEnumerable < SymbolReference > symbols = await GetReferences ( FindsReferencesOnFunctionData . SourceDetails ) . ConfigureAwait ( true ) ;
196
- Assert . Collection ( symbols ,
197
- ( i ) => AssertIsRegion ( i . NameRegion , 1 , 10 , 1 , 21 ) ,
198
- ( i ) => AssertIsRegion ( i . NameRegion , 3 , 5 , 3 , 16 ) ,
199
- ( i ) => AssertIsRegion ( i . NameRegion , 10 , 1 , 10 , 12 ) ) ;
196
+ Assert . Equal ( 8 , symbols . Count ( ) ) ;
197
+ Assert . All ( symbols , ( i ) =>
198
+ {
199
+ Assert . Equal ( "My-Function" , i . SymbolName ) ;
200
+ Assert . Equal ( SymbolType . Function , i . SymbolType ) ;
201
+ if ( i . IsDeclaration )
202
+ {
203
+ Assert . Equal ( "function My-Function ($myInput)" , i . DisplayString ) ;
204
+ }
205
+ } ) ;
206
+ Assert . Distinct ( symbols ) ;
200
207
}
201
208
202
209
[ Fact ]
0 commit comments