Fix crash with switch expression as a lambda body (#543)

Fixes #524 

Fixed by bumping Checker Framework version

2 files changed, 17 insertions, 1 deletions

diff --git a/gradle/dependencies.gradle b/gradle/dependencies.gradle
index 3b4a4e5..c882a84 100755
--- a/gradle/dependencies.gradle
+++ b/gradle/dependencies.gradle
@@ -32,7 +32,7 @@ if (project.hasProperty("epApiVersion")) {
 }
 def versions = [
     asm                    : "7.1",
-    checkerFramework       : "3.21.0",
+    checkerFramework       : "3.21.1",
     // The version of Error Prone used to check NullAway's code
     errorProne             : "2.10.0",
     // The version of Error Prone that NullAway is compiled and tested against
diff --git a/jdk17-unit-tests/src/test/java/com/uber/nullaway/jdk17/NullAwaySwitchTests.java b/jdk17-unit-tests/src/test/java/com/uber/nullaway/jdk17/NullAwaySwitchTests.java
index 1b47d2f..8cd43c0 100644
--- a/jdk17-unit-tests/src/test/java/com/uber/nullaway/jdk17/NullAwaySwitchTests.java
+++ b/jdk17-unit-tests/src/test/java/com/uber/nullaway/jdk17/NullAwaySwitchTests.java
@@ -156,4 +156,20 @@ public class NullAwaySwitchTests {
             "}")
         .doTest();
   }
+
+  @Test
+  public void testSwitchExprLambda() {
+    defaultCompilationHelper
+        .addSourceLines(
+            "SwitchExpr.java",
+            "package com.uber;",
+            "import java.util.function.Function;",
+            "class SwitchExpr {",
+            "  int i;",
+            "  public void testSwitchExprLambda() {",
+            "    Function<SwitchExpr,Object> f = (s) -> switch (s.i) { case 3, 4, 5 -> new Object(); default -> \"hello\"; };",
+            "  }",
+            "}")
+        .doTest();
+  }
 }