Fix a crash in UnnecessaryBoxedVariable

cushon · Error Prone Team · commit ed35fdafea8d · 2022-04-15T11:17:07.000-07:00
Don't assume that return statements are enclosed by a method with a return type, to handle statement lambdas inside constructors. Fixes #3115 PiperOrigin-RevId: 442056569
diff --git a/core/src/main/java/com/google/errorprone/bugpatterns/UnnecessaryBoxedVariable.java b/core/src/main/java/com/google/errorprone/bugpatterns/UnnecessaryBoxedVariable.java
@@ -21,6 +21,7 @@
 import static com.google.errorprone.matchers.Matchers.anyOf;
 import static com.google.errorprone.matchers.method.MethodMatchers.instanceMethod;
 import static com.google.errorprone.matchers.method.MethodMatchers.staticMethod;
+import static com.google.errorprone.util.ASTHelpers.findEnclosingMethod;
 import static com.google.errorprone.util.ASTHelpers.getSymbol;
 
 import com.google.common.collect.ArrayListMultimap;
@@ -429,11 +430,12 @@ public Void visitReturn(ReturnTree node, Void unused) {
       // Don't count a return value as a boxed usage, except if we are returning a parameter, and
       // the method's return type is boxed.
       if (nodeSymbol.getKind() == ElementKind.PARAMETER) {
-        MethodTree enclosingMethod =
-            ASTHelpers.findEnclosingNode(getCurrentPath(), MethodTree.class);
-        Type returnType = ASTHelpers.getType(enclosingMethod.getReturnType());
-        if (!returnType.isPrimitive()) {
-          boxedUsageFound.add((VarSymbol) nodeSymbol);
+        MethodTree enclosingMethod = findEnclosingMethod(state.withPath(getCurrentPath()));
+        if (enclosingMethod != null) {
+          Type returnType = ASTHelpers.getType(enclosingMethod.getReturnType());
+          if (!returnType.isPrimitive()) {
+            boxedUsageFound.add((VarSymbol) nodeSymbol);
+          }
         }
       }
       return null;
diff --git a/core/src/test/java/com/google/errorprone/bugpatterns/UnnecessaryBoxedVariableTest.java b/core/src/test/java/com/google/errorprone/bugpatterns/UnnecessaryBoxedVariableTest.java
@@ -69,4 +69,22 @@ public void lambdas() {
             "}")
         .doTest();
   }
+
+  @Test
+  public void lambdaReturn() {
+    compilationTestHelper
+        .addSourceLines(
+            "Test.java",
+            "class Test {",
+            "  interface F {",
+            "    int f(Integer i);",
+            "  }",
+            "  Test() {",
+            "    F f = (Integer i) -> {",
+            "      return i;",
+            "    };",
+            "  }",
+            "}")
+        .doTest();
+  }
 }
