Handle test/kw names followed by data

yanne · yanne · commit 9b29b7fa8f9a · 2019-05-30T19:01:24.000+03:00
diff --git a/src/robot/parsing/lexer/__init__.py b/src/robot/parsing/lexer/__init__.py
@@ -46,9 +46,18 @@ def get_tokens(self):
         else:
             ignore = {Token.IGNORE}
         for statement in self._handle_old_for(self.statements):
+            name_eos = None
             for token in statement:
-                if token.type not in ignore:
-                    yield token
+                if token.type in ignore:
+                    continue
+                if name_eos:
+                    yield name_eos
+                    name_eos = None
+                if token.type == Token.NAME:
+                    name_eos = Token(Token.EOS,
+                                     lineno=token.lineno,
+                                     columnno=token.columnno + len(token.value))
+                yield token
             yield Token(Token.EOS,
                         lineno=token.lineno,
                         columnno=token.columnno + len(token.value))
