GH-213 - do not report NS1004 when arg matcher is used directly in re…

…turn statement
nsubstitute · Sep 23, 2023 · e996107 · e996107
1 parent 8e35c28
commit e996107
Show file tree

Hide file tree

Showing 6 changed files with 67 additions and 1 deletion.
diff --git a/...yzers.Shared/DiagnosticAnalyzers/AbstractNonSubstitutableMemberArgumentMatcherAnalyzer.cs b/...yzers.Shared/DiagnosticAnalyzers/AbstractNonSubstitutableMemberArgumentMatcherAnalyzer.cs
@@ -20,7 +20,7 @@ internal abstract class AbstractNonSubstitutableMemberArgumentMatcherAnalyzer :
         OperationKind.SimpleAssignment);
 
     private static readonly ImmutableHashSet<OperationKind> IgnoredAncestors =
-        ImmutableHashSet.Create(OperationKind.VariableDeclarator, OperationKind.VariableDeclaration);
+        ImmutableHashSet.Create(OperationKind.VariableDeclarator, OperationKind.VariableDeclaration, OperationKind.Return);
 
     private static readonly ImmutableHashSet<OperationKind> DynamicOperations =
         ImmutableHashSet.Create(OperationKind.DynamicInvocation, OperationKind.DynamicIndexerAccess, OperationKind.DynamicMemberReference);

diff --git a/...erArgumentMatcherAnalyzerTests/NonSubstitutableMemberArgumentMatcherDiagnosticVerifier.cs b/...erArgumentMatcherAnalyzerTests/NonSubstitutableMemberArgumentMatcherDiagnosticVerifier.cs
@@ -135,6 +135,10 @@ public abstract class NonSubstitutableMemberArgumentMatcherDiagnosticVerifier :
     [MemberData(nameof(CorrectlyUsedArgTestCasesWithoutDelegates))]
     public abstract Task ReportsDiagnostics_WhenAssigningInvalidArgMatchersToMemberPrecededByWithAnyArgsLikeMethod(string receivedMethod, string arg);
 
+    [CombinatoryTheory]
+    [MemberData(nameof(CorrectlyUsedArgTestCasesWithoutDelegates))]
+    public abstract Task ReportsNoDiagnostics_WhenUsedDirectlyWithReturnStatement(string arg);
+
     public static IEnumerable<object[]> MisusedArgTestCases
     {
         get

diff --git a/...titutableMemberArgumentMatcherAnalyzerTests/NonSubstitutableMemberArgumentMatcherTests.cs b/...titutableMemberArgumentMatcherAnalyzerTests/NonSubstitutableMemberArgumentMatcherTests.cs
@@ -1000,6 +1000,36 @@ public void Test()
             {receivedMethod}[1] = {arg};
         }}
     }}
+}}";
+        await VerifyNoDiagnostic(source);
+    }
+
+    public override async Task ReportsNoDiagnostics_WhenUsedDirectlyWithReturnStatement(string arg)
+    {
+        var source = $@"using System;
+using NSubstitute;
+using NSubstitute.ReceivedExtensions;
+
+namespace MyNamespace
+{{
+    public interface IFoo
+    {{
+        int Bar(int? x);
+    }}
+
+    public class FooTests
+    {{
+        public void Test()
+        {{
+            var substitute = Substitute.For<IFoo>();
+            substitute.Received(1).Bar(MatchesArg());
+        }}
+
+        private int? MatchesArg()
+        {{
+            return {arg};
+        }}
+    }}
 }}";
         await VerifyNoDiagnostic(source);
     }

diff --git a/...ts.Shared/DiagnosticAnalyzers/INonSubstitutableMemberArgumentMatcherDiagnosticVerifier.cs b/...ts.Shared/DiagnosticAnalyzers/INonSubstitutableMemberArgumentMatcherDiagnosticVerifier.cs
@@ -59,4 +59,6 @@ public interface INonSubstitutableMemberArgumentMatcherDiagnosticVerifier
     Task ReportsDiagnostics_WhenAssigningArgMatchersToNonSubstitutableMember_InWhenLikeMethod(string whenMethod, string arg);
 
     Task ReportsNoDiagnostics_WhenAssigningArgMatchersToSubstitutableMember_InWhenLikeMethod(string whenMethod, string arg);
+
+    Task ReportsNoDiagnostics_WhenUsedDirectlyWithReturnStatement(string arg);
 }
diff --git a/...erArgumentMatcherAnalyzerTests/NonSubstitutableMemberArgumentMatcherDiagnosticVerifier.cs b/...erArgumentMatcherAnalyzerTests/NonSubstitutableMemberArgumentMatcherDiagnosticVerifier.cs
@@ -139,6 +139,10 @@ public abstract class NonSubstitutableMemberArgumentMatcherDiagnosticVerifier :
     [Fact]
     public abstract Task ReportsNoDiagnostic_WhenOverloadCannotBeInferred();
 
+    [CombinatoryTheory]
+    [MemberData(nameof(CorrectlyUsedArgTestCasesWithoutDelegates))]
+    public abstract Task ReportsNoDiagnostics_WhenUsedDirectlyWithReturnStatement(string arg);
+
     public static IEnumerable<object[]> MisusedArgTestCases
     {
         get { return MisusedArgs.Select(argArray => argArray.Select<string, object>(arg => arg).ToArray()); }

diff --git a/...titutableMemberArgumentMatcherAnalyzerTests/NonSubstitutableMemberArgumentMatcherTests.cs b/...titutableMemberArgumentMatcherAnalyzerTests/NonSubstitutableMemberArgumentMatcherTests.cs
@@ -1017,4 +1017,30 @@ End Namespace
 
         await VerifyNoDiagnostic(source);
     }
+
+    public override async Task ReportsNoDiagnostics_WhenUsedDirectlyWithReturnStatement(string arg)
+    {
+        var source = @"Imports System
+Imports NSubstitute
+Imports NSubstitute.ReceivedExtensions
+
+Namespace MyNamespace
+    Interface IFoo
+        Function Bar(ByVal x As Integer?) As Integer
+    End Interface
+
+    Public Class FooTests
+        Public Sub Test()
+            Dim substitute = NSubstitute.Substitute.[For](Of IFoo)()
+            substitute.Received(1).Bar(MatchesArg())
+        End Sub
+
+        Private Function MatchesArg() As Integer?
+            Return Arg.Any(Of Integer)()
+        End Function
+    End Class
+End Namespace
+";
+        await VerifyNoDiagnostic(source);
+    }
 }