Added tests for parsing switch expressions in return statements, and …

…got them passing.
neilccbrown · May 11, 2024 · bc42a60 · bc42a60
1 parent 3c275df
commit bc42a60
Show file tree

Hide file tree

Showing 2 changed files with 48 additions and 2 deletions.
diff --git a/bluej/src/main/java/bluej/parser/JavaParser.java b/bluej/src/main/java/bluej/parser/JavaParser.java
@@ -2899,8 +2899,7 @@ else if (tt1 == JavaTokenTypes.IDENT && tt2 == JavaTokenTypes.COMMA) {
             case 31: // LITERAL_switch
                 // Switch expression
                 parseSwitchExpression(token);
-                token = nextToken();
-                continue exprLoop;                
+                break;
             default:
                 tokenStream.pushBack(token);
                 error("Invalid expression token: " + token.getText());

diff --git a/bluej/src/test/java/bluej/parser/BasicParseTest.java b/bluej/src/test/java/bluej/parser/BasicParseTest.java
@@ -853,6 +853,53 @@ class Foo
         assertFalse(info.hadParseError());
     }
 
+    @Test
+    public void testSwitchExpression1()
+    {
+        String aSrc =
+                """
+                class Foo
+                {
+                    public int bar()
+                    {
+                        return switch ("hi") {
+                            case "bye" -> -1;
+                            default -> 0;
+                        };
+                    }
+                }
+                """;
+
+        ClassInfo info = InfoParser.parse(new StringReader(aSrc), new ClassLoaderResolver(getClass().getClassLoader()), null);
+        assertNotNull(info);
+        assertFalse(info.hadParseError());
+    }
+
+    @Test
+    public void testSwitchExpression2()
+    {
+        String aSrc =
+                """
+                class Foo
+                {
+                    public int bar()
+                    {
+                        return switch ("hi") {
+                            case "bye" -> -1;
+                            default -> 0;
+                        } + switch ("bye") {
+                            case "hi" -> -1;
+                            default -> 0;
+                        };
+                    }
+                }
+                """;
+
+        ClassInfo info = InfoParser.parse(new StringReader(aSrc), new ClassLoaderResolver(getClass().getClassLoader()), null);
+        assertNotNull(info);
+        assertFalse(info.hadParseError());
+    }
+
 
     // Examples mostly from https://openjdk.org/jeps/409
     @Test