[lexer] Supports root-node token

Author: springcomp

jmespath/lexer.py

@@ -17,6 +17,7 @@ class Lexer(object):
         ',': 'comma',
         ':': 'colon',
         '@': 'current',
+        '$': 'root',
         '(': 'lparen',
         ')': 'rparen',
         '{': 'lbrace',

tests/test_lexer.py

@@ -143,6 +143,23 @@ def test_adds_quotes_when_invalid_json(self):
              ]
         )
 
+    def test_root_reference(self):
+        tokens = list(self.lexer.tokenize('$[0]'))
+        self.assertEqual(
+            tokens,
+            [{'type': 'root', 'value': '$',
+                'start': 0, 'end': 1},
+            {'type': 'lbracket', 'value':
+                '[', 'start': 1, 'end': 2},
+            {'type': 'number', 'value': 0,
+                'start': 2, 'end': 3},
+            {'type': 'rbracket', 'value': ']',
+                'start': 3, 'end': 4},
+            {'type': 'eof', 'value': '',
+                'start': 4, 'end': 4}
+            ]
+        )
+
     def test_unknown_character(self):
         with self.assertRaises(LexerError) as e:
             tokens = list(self.lexer.tokenize('foo[0^]'))