Use TokenKind enum inside lexer/parser tests

Replicates graphql/graphql-js@aba1066

Author: Cito

tests/language/test_lexer.py

@@ -322,8 +322,8 @@ def lex_reports_useful_unknown_character_error():
 
     # noinspection PyArgumentEqualDefault
     def lex_reports_useful_information_for_dashes_in_names():
-        q = "a-b"
-        lexer = Lexer(Source(q))
+        source = Source("a-b")
+        lexer = Lexer(source)
         first_token = lexer.advance()
         assert first_token == Token(TokenKind.NAME, 0, 1, 1, 1, None, "a")
         with raises(GraphQLSyntaxError) as exc_info:
@@ -335,7 +335,15 @@ def lex_reports_useful_information_for_dashes_in_names():
         assert error.locations == [(1, 3)]
 
     def produces_double_linked_list_of_tokens_including_comments():
-        lexer = Lexer(Source("{\n      #comment\n      field\n    }"))
+        source = Source(
+            """
+            {
+              #comment
+              field
+            }
+            """
+        )
+        lexer = Lexer(source)
         start_token = lexer.token
         while True:
             end_token = lexer.advance()
@@ -350,11 +358,11 @@ def produces_double_linked_list_of_tokens_including_comments():
             assert not tokens or tok.prev == tokens[-1]
             tokens.append(tok)
             tok = tok.next
-        assert [tok.kind.value for tok in tokens] == [
-            "<SOF>",
-            "{",
-            "Comment",
-            "Name",
-            "}",
-            "<EOF>",
+        assert [tok.kind for tok in tokens] == [
+            TokenKind.SOF,
+            TokenKind.BRACE_L,
+            TokenKind.COMMENT,
+            TokenKind.NAME,
+            TokenKind.BRACE_R,
+            TokenKind.EOF,
         ]

tests/language/test_parser.py

@@ -22,6 +22,7 @@
     StringValueNode,
     ValueNode,
     Token,
+    TokenKind,
     parse,
     parse_type,
     parse_value,
@@ -376,10 +377,10 @@ def contains_references_to_start_and_end_tokens():
         result = parse("{ id }")
         start_token = result.loc.start_token
         assert isinstance(start_token, Token)
-        assert start_token.desc == "<SOF>"
+        assert start_token.kind == TokenKind.SOF
         end_token = result.loc.end_token
         assert isinstance(end_token, Token)
-        assert end_token.desc == "<EOF>"
+        assert end_token.kind == TokenKind.EOF
 
 
 def describe_parse_value():