Mojo: Mojom: Fix line number tracking when there are comments.

mojo/public/tools/bindings/pylib/mojom_tests/parse/parser_unittest.py

 # Copyright 2014 The Chromium Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
 import imp
 import os.path
 import sys
 import unittest
 
 # Disable lint check for finding modules:
@@ skipping 43 matching lines @@
     source = """\
 // This is a comment.
 
 module my_module {
 """
     with self.assertRaisesRegexp(
         parser.ParseError,
         r"^my_file\.mojom: Error: Unexpected end of file$"):
       parser.Parse(source, "my_file.mojom")
 
+  def testCommentLineNumbers(self):
+    """Tests that line numbers are correctly tracked when comments are
+    present."""
+    source1 = """\
+// Isolated C++-style comments.
+
+// Foo.
+asdf1
+"""
+    with self.assertRaisesRegexp(
+        parser.ParseError,
+        r"^my_file\.mojom:4: Error: Unexpected 'asdf1':\nasdf1$"):
+      parser.Parse(source1, "my_file.mojom")
+
+    source2 = """\
+// Consecutive C++-style comments.
+// Foo.
+  // Bar.
+
+struct Yada {  // Baz.
+// Quux.
+  int32 x;
+};
+
+asdf2
+"""
+    with self.assertRaisesRegexp(
+        parser.ParseError,
+        r"^my_file\.mojom:10: Error: Unexpected 'asdf2':\nasdf2$"):
+      parser.Parse(source2, "my_file.mojom")
+
+    source3 = """\
+/* Single-line C-style comments. */
+/* Foobar. */
+
+/* Baz. */
+asdf3
+"""
+    with self.assertRaisesRegexp(
+        parser.ParseError,
+        r"^my_file\.mojom:5: Error: Unexpected 'asdf3':\nasdf3$"):
+      parser.Parse(source3, "my_file.mojom")
+
+    source4 = """\
+/* Multi-line C-style comments.
+*/
+/*
+Foo.
+Bar.
+*/
+
+/* Baz
+   Quux. */
+asdf4
+"""
+    with self.assertRaisesRegexp(
+        parser.ParseError,
+        r"^my_file\.mojom:10: Error: Unexpected 'asdf4':\nasdf4$"):
+      parser.Parse(source4, "my_file.mojom")
+
+
   def testSimpleStruct(self):
     """Tests a simple .mojom source that just defines a struct."""
     source = """\
 module my_module {
 
 struct MyStruct {
   int32 a;
   double b;
 };
 
@@ skipping 294 matching lines @@
     with self.assertRaisesRegexp(
         parser.ParseError,
         r"^my_file\.mojom:2: Error: "
             r"Invalid handle type 'wtf_is_this':\n"
             r"  handle<wtf_is_this> foo;$"):
       parser.Parse(source, "my_file.mojom")
 
 
 if __name__ == "__main__":
   unittest.main()