Add more tests for arrow and multi-case switch statements in MissingC…

…asesInEnumSwitchTest PiperOrigin-RevId: 655288858
google · Jul 23, 2024 · 7b58ec9 · 7b58ec9
1 parent 996d09c
commit 7b58ec9
Showing 1 changed file with 59 additions and 0 deletions.
diff --git a/core/src/test/java/com/google/errorprone/bugpatterns/MissingCasesInEnumSwitchTest.java b/core/src/test/java/com/google/errorprone/bugpatterns/MissingCasesInEnumSwitchTest.java
@@ -160,4 +160,63 @@ public void empty() {
             "}")
         .doTest();
   }
+
+  @Test
+  public void nonExhaustive_arrowStatement() {
+    assumeTrue(RuntimeVersion.isAtLeast14());
+    compilationHelper
+        .addSourceLines(
+            "Test.java",
+            "class Test {",
+            "  enum Case { ONE, TWO }",
+            "  void m(Case c) {",
+            "    // BUG: Diagnostic contains: TWO",
+            "    switch (c) {",
+            "      case ONE -> {",
+            "        System.err.println(\"found it!\");",
+            "      }",
+            "    }",
+            "  }",
+            "}")
+        .doTest();
+  }
+
+  @Test
+  public void nonExhaustive_multi() {
+    assumeTrue(RuntimeVersion.isAtLeast14());
+    compilationHelper
+        .addSourceLines(
+            "Test.java",
+            "class Test {",
+            "  enum Case { ONE, TWO, THREE }",
+            "  void m(Case c) {",
+            "    // BUG: Diagnostic contains: THREE",
+            "    switch (c) {",
+            "      case ONE, TWO:",
+            "        System.err.println(\"found it!\");",
+            "    }",
+            "  }",
+            "}")
+        .doTest();
+  }
+
+  @Test
+  public void nonExhaustive_multiArrow() {
+    assumeTrue(RuntimeVersion.isAtLeast14());
+    compilationHelper
+        .addSourceLines(
+            "Test.java",
+            "class Test {",
+            "  enum Case { ONE, TWO, THREE }",
+            "  void m(Case c) {",
+            "    // BUG: Diagnostic contains: THREE",
+            "    switch (c) {",
+            "      case ONE, TWO -> {",
+            "        System.err.println(\"found it!\");",
+            "      }",
+            "    }",
+            "  }",
+            "}")
+        .doTest();
+  }
 }