commit b5768af15f2949385b3d782199336d70e56e00e2
author fzaiser <fzaiser@google.com> Mon Oct 09 15:16:50 2017 +0200
committer Marcel Hlopko <hlopko@google.com> Tue Oct 10 11:22:04 2017 +0200
tree ac2eb5c63147cd8a27933f528f4475fcd771f94f
parent 8880b7abceb78b22b56be53d02bf97d9e8cb74cb

Fix the location of load statements in Skylark.

Previously, the closing parenthesis was not included in the location.

RELNOTES: none
PiperOrigin-RevId: 171527078

src/test/java/com/google/devtools/build/lib/syntax/ParserTest.java

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 20ea5c0..858e0ca 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
@@ -555,6 +555,16 @@
     assertExpressionLocationCorrect("not True");
   }
 
+  @Test
+  public void testLoadStatementPosition() throws Exception {
+    String input = "load(':foo.bzl', 'bar')";
+    LoadStatement stmt = (LoadStatement) parseFile(input).get(0);
+    assertThat(getText(input, stmt)).isEqualTo(input);
+    // Also try it with another token at the end (newline), which broke the location in the past.
+    stmt = (LoadStatement) parseFile(input + "\n").get(0);
+    assertThat(getText(input, stmt)).isEqualTo(input);
+  }
+
   private void assertExpressionLocationCorrect(String exprStr) {
     Expression expr = parseExpression(exprStr);
     assertThat(getText(exprStr, expr)).isEqualTo(exprStr);