@@ -47,21 +47,21 @@ public class CollectionShouldContainPropertyCodeFix : FluentAssertionsCodeFixPro
47
47
{
48
48
public override ImmutableArray < string > FixableDiagnosticIds => ImmutableArray . Create ( CollectionShouldContainPropertyAnalyzer . DiagnosticId ) ;
49
49
50
- protected override StatementSyntax GetNewStatement ( ExpressionStatementSyntax statement , FluentAssertionsDiagnosticProperties properties )
50
+ protected override ExpressionSyntax GetNewExpression ( ExpressionSyntax expression , FluentAssertionsDiagnosticProperties properties )
51
51
{
52
52
if ( properties . VisitorName == nameof ( CollectionShouldContainPropertyAnalyzer . AnyShouldBeTrueSyntaxVisitor ) )
53
53
{
54
54
var remove = new NodeReplacement . RemoveAndExtractArgumentsNodeReplacement ( "Any" ) ;
55
- var newStatement = GetNewStatement ( statement , remove ) ;
55
+ var newStatement = GetNewExpression ( expression , remove ) ;
56
56
57
- return GetNewStatement ( newStatement , new NodeReplacement . RenameAndPrependArgumentsNodeReplacement ( "BeTrue" , "Contain" , remove . Arguments ) ) ;
57
+ return GetNewExpression ( newStatement , new NodeReplacement . RenameAndPrependArgumentsNodeReplacement ( "BeTrue" , "Contain" , remove . Arguments ) ) ;
58
58
}
59
59
else if ( properties . VisitorName == nameof ( CollectionShouldContainPropertyAnalyzer . WhereShouldNotBeEmptySyntaxVisitor ) )
60
60
{
61
61
var remove = new NodeReplacement . RemoveAndExtractArgumentsNodeReplacement ( "Where" ) ;
62
- var newStatement = GetNewStatement ( statement , remove ) ;
62
+ var newStatement = GetNewExpression ( expression , remove ) ;
63
63
64
- return GetNewStatement ( newStatement , new NodeReplacement . RenameAndPrependArgumentsNodeReplacement ( "NotBeEmpty" , "Contain" , remove . Arguments ) ) ;
64
+ return GetNewExpression ( newStatement , new NodeReplacement . RenameAndPrependArgumentsNodeReplacement ( "NotBeEmpty" , "Contain" , remove . Arguments ) ) ;
65
65
}
66
66
throw new System . InvalidOperationException ( $ "Invalid visitor name - { properties . VisitorName } ") ;
67
67
}
0 commit comments