protected override ExpressionSyntax GetNewExpression(ExpressionSyntax expression, FluentAssertionsDiagnosticProperties properties)
{
if (properties.VisitorName == nameof(CollectionShouldHaveElementAtAnalyzer.ElementAtIndexShouldBeSyntaxVisitor))
{
var remove = NodeReplacement.RemoveAndExtractArguments("ElementAt");
var newExpression = GetNewExpression(expression, remove);
return(GetNewExpression(newExpression, NodeReplacement.RenameAndPrependArguments("Be", "HaveElementAt", remove.Arguments)));
}
else if (properties.VisitorName == nameof(CollectionShouldHaveElementAtAnalyzer.IndexerShouldBeSyntaxVisitor))
{
var remove = NodeReplacement.RemoveAndRetrieveIndexerArguments("Should");
var newExpression = GetNewExpression(expression, remove);
return(GetNewExpression(newExpression, NodeReplacement.RenameAndPrependArguments("Be", "HaveElementAt", remove.Arguments)));
}
else if (properties.VisitorName == nameof(CollectionShouldHaveElementAtAnalyzer.SkipFirstShouldBeSyntaxVisitor))
{
var remove = NodeReplacement.RemoveAndExtractArguments("Skip");
var newExpression = GetNewExpression(expression, remove, NodeReplacement.Remove("First"));
return(GetNewExpression(newExpression, NodeReplacement.RenameAndPrependArguments("Be", "HaveElementAt", remove.Arguments)));
}
throw new System.InvalidOperationException($"Invalid visitor name - {properties.VisitorName}");
}
