Fix location for Break and Continue AST nodes.

--
MOS_MIGRATED_REVID=106291132

1 files changed, 9 insertions, 5 deletions

diff --git a/src/test/java/com/google/devtools/build/lib/syntax/ParserTest.java b/src/test/java/com/google/devtools/build/lib/syntax/ParserTest.java
index 8407823ad6..e5d676969d 100644
--- a/src/test/java/com/google/devtools/build/lib/syntax/ParserTest.java
+++ b/src/test/java/com/google/devtools/build/lib/syntax/ParserTest.java
@@ -469,19 +469,23 @@ public class ParserTest extends EvaluationTestCase {
         "def foo():",
         "  for i in [1, 2]:",
         "    break",
-        "    continue");
+        "    continue",
+        "    break");
     assertThat(file).hasSize(1);
     List<Statement> body = ((FunctionDefStatement) file.get(0)).getStatements();
     assertThat(body).hasSize(1);
 
     List<Statement> loop = ((ForStatement) body.get(0)).block();
-    assertThat(loop).hasSize(2);
+    assertThat(loop).hasSize(3);
 
-    assertThat(loop.get(0)).isEqualTo(FlowStatement.BREAK);
+    assertThat(((FlowStatement) loop.get(0)).getKind()).isEqualTo(FlowStatement.Kind.BREAK);
     assertLocation(34, 40, loop.get(0).getLocation());
 
-    assertThat(loop.get(1)).isEqualTo(FlowStatement.CONTINUE);
-    assertLocation(44, 52, loop.get(1).getLocation());
+    assertThat(((FlowStatement) loop.get(1)).getKind()).isEqualTo(FlowStatement.Kind.CONTINUE);
+    assertLocation(44, 53, loop.get(1).getLocation());
+
+    assertThat(((FlowStatement) loop.get(2)).getKind()).isEqualTo(FlowStatement.Kind.BREAK);
+    assertLocation(57, 62, loop.get(2).getLocation());
   }
 
   @Test