Handle bang operator

Author: JohanLarsson

ReflectionAnalyzers.Tests/REFL001CastReturnValueTests/CodeFix.MethodInfoInvoke.cs

@@ -11,33 +11,69 @@ public static class MethodInfoInvoke
             private static readonly CastReturnValueFix Fix = new();
             private static readonly ExpectedDiagnostic ExpectedDiagnostic = ExpectedDiagnostic.Create(Descriptors.REFL001CastReturnValue);
 
-            [Test]
-            public static void AssigningLocal()
+            [TestCase("typeof(C).GetMethod(nameof(M)).Invoke(null, null)")]
+            [TestCase("typeof(C).GetMethod(nameof(M))!.Invoke(null, null)")]
+            [TestCase("typeof(C).GetMethod(nameof(M))?.Invoke(null, null)")]
+            [TestCase("typeof(C).GetMethod(nameof(M))?.Invoke(null, null) ?? throw new Exception()")]
+            public static void AssigningLocal(string expression)
             {
                 var before = @"
+#pragma warning disable CS8019, CS8602, CS8605
 namespace N
 {
+    using System;
+
     public class C
     {
         public C()
         {
-            var value = ↓typeof(C).GetMethod(nameof(M)).Invoke(null, null);
+            var x = ↓typeof(C).GetMethod(nameof(M)).Invoke(null, null);
         }
 
         public static int M() => 0;
     }
-}";
+}".AssertReplace("typeof(C).GetMethod(nameof(M)).Invoke(null, null)", expression);
 
                 var after = @"
+#pragma warning disable CS8019, CS8602, CS8605
 namespace N
 {
+    using System;
+
     public class C
     {
         public C()
         {
-            var value = (int)typeof(C).GetMethod(nameof(M)).Invoke(null, null);
+            var x = (int)typeof(C).GetMethod(nameof(M)).Invoke(null, null);
         }
 
+        public static int M() => 0;
+    }
+}".AssertReplace("typeof(C).GetMethod(nameof(M)).Invoke(null, null)", expression);
+                RoslynAssert.CodeFix(Analyzer, Fix, ExpectedDiagnostic, before, after);
+            }
+
+            [Test]
+            public static void Returning()
+            {
+                var before = @"
+namespace N
+{
+    public class C
+    {
+        public object? Get() => ↓typeof(C).GetMethod(nameof(M)).Invoke(null, null);
+
+        public static int M() => 0;
+    }
+}";
+
+                var after = @"
+namespace N
+{
+    public class C
+    {
+        public object? Get() => (int)typeof(C).GetMethod(nameof(M)).Invoke(null, null);
+
         public static int M() => 0;
     }
 }";

ReflectionAnalyzers/Helpers/Reflection/GetX.cs

@@ -2,7 +2,9 @@
 {
     using System.Diagnostics.CodeAnalysis;
     using System.Threading;
+
     using Gu.Roslyn.AnalyzerExtensions;
+
     using Microsoft.CodeAnalysis;
     using Microsoft.CodeAnalysis.CSharp;
     using Microsoft.CodeAnalysis.CSharp.Syntax;
@@ -179,6 +181,10 @@ private static bool TryFindInvocation(MemberAccessExpressionSyntax memberAccess,
         {
             switch (memberAccess.Expression)
             {
+                case PostfixUnaryExpressionSyntax { RawKind: (int)SyntaxKind.SuppressNullableWarningExpression, Operand: InvocationExpressionSyntax candidate }
+                    when candidate.TryGetTarget(expected, semanticModel, cancellationToken, out _):
+                    invocation = candidate;
+                    return true;
                 case InvocationExpressionSyntax candidate
                     when candidate.TryGetTarget(expected, semanticModel, cancellationToken, out _):
                     invocation = candidate;