[flow][enums] Add parser tests for bigint enums, and big int enums to custom_ast_types.js

gkz · facebook-github-bot · commit 786fcb99bf42 · 2024-04-05T11:28:54.000-07:00
Summary:
Add parser tests for bigint enums, and big int enums to custom_ast_types.js

Changelog: [internal]

Reviewed By: SamChou19815

Differential Revision: D55774602

fbshipit-source-id: a3e71050b2ee86edb76506550ba016a30cfdc1a1
diff --git a/packages/flow-parser/test/custom_ast_types.js b/packages/flow-parser/test/custom_ast_types.js
@@ -132,9 +132,31 @@ function custom_ast_types(fork) {
         def('EnumNumberBody'),
         def('EnumStringBody'),
         def('EnumSymbolBody'),
+        def('EnumBigIntBody'),
       ),
     );
 
+  def('EnumDeclaration').field(
+    'body',
+    or(
+      def('EnumBooleanBody'),
+      def('EnumNumberBody'),
+      def('EnumStringBody'),
+      def('EnumSymbolBody'),
+      def('EnumBigIntBody'), // <-- ADDITION
+    ),
+  );
+
+  def('EnumBigIntBody')
+    .build('members', 'explicitType')
+    .field('members', [def('EnumBigIntMember')])
+    .field('explicitType', Boolean);
+
+  def('EnumBigIntMember')
+    .build('id', 'init')
+    .field('id', def('Identifier'))
+    .field('init', def('Literal'));
+
   def('DeclareNamespace')
     .bases('Declaration')
     .build('id', 'body')
diff --git a/src/parser/test/flow/enums/enum-valid-bigint-explicit.js b/src/parser/test/flow/enums/enum-valid-bigint-explicit.js
@@ -0,0 +1,4 @@
+enum E of bigint {
+  A = 1n,
+  B = 2n,
+}
diff --git a/src/parser/test/flow/enums/enum-valid-bigint-explicit.options.json b/src/parser/test/flow/enums/enum-valid-bigint-explicit.options.json
@@ -0,0 +1,3 @@
+{
+  "enums": true
+}
diff --git a/src/parser/test/flow/enums/enum-valid-bigint-explicit.tree.json b/src/parser/test/flow/enums/enum-valid-bigint-explicit.tree.json
@@ -0,0 +1,72 @@
+{
+  "type":"Program",
+  "loc":{"source":null,"start":{"line":1,"column":0},"end":{"line":4,"column":1}},
+  "range":[0,40],
+  "body":[
+    {
+      "type":"EnumDeclaration",
+      "loc":{"source":null,"start":{"line":1,"column":0},"end":{"line":4,"column":1}},
+      "range":[0,40],
+      "id":{
+        "type":"Identifier",
+        "loc":{"source":null,"start":{"line":1,"column":5},"end":{"line":1,"column":6}},
+        "range":[5,6],
+        "name":"E",
+        "typeAnnotation":null,
+        "optional":false
+      },
+      "body":{
+        "type":"EnumBigIntBody",
+        "loc":{"source":null,"start":{"line":1,"column":7},"end":{"line":4,"column":1}},
+        "range":[7,40],
+        "members":[
+          {
+            "type":"EnumBigIntMember",
+            "loc":{"source":null,"start":{"line":2,"column":2},"end":{"line":2,"column":8}},
+            "range":[21,27],
+            "id":{
+              "type":"Identifier",
+              "loc":{"source":null,"start":{"line":2,"column":2},"end":{"line":2,"column":3}},
+              "range":[21,22],
+              "name":"A",
+              "typeAnnotation":null,
+              "optional":false
+            },
+            "init":{
+              "type":"Literal",
+              "loc":{"source":null,"start":{"line":2,"column":6},"end":{"line":2,"column":8}},
+              "range":[25,27],
+              "value":null,
+              "bigint":"1",
+              "raw":"1n"
+            }
+          },
+          {
+            "type":"EnumBigIntMember",
+            "loc":{"source":null,"start":{"line":3,"column":2},"end":{"line":3,"column":8}},
+            "range":[31,37],
+            "id":{
+              "type":"Identifier",
+              "loc":{"source":null,"start":{"line":3,"column":2},"end":{"line":3,"column":3}},
+              "range":[31,32],
+              "name":"B",
+              "typeAnnotation":null,
+              "optional":false
+            },
+            "init":{
+              "type":"Literal",
+              "loc":{"source":null,"start":{"line":3,"column":6},"end":{"line":3,"column":8}},
+              "range":[35,37],
+              "value":null,
+              "bigint":"2",
+              "raw":"2n"
+            }
+          }
+        ],
+        "explicitType":true,
+        "hasUnknownMembers":false
+      }
+    }
+  ],
+  "comments":[]
+}
diff --git a/src/parser/test/flow/enums/enum-valid-bigint-implicit.js b/src/parser/test/flow/enums/enum-valid-bigint-implicit.js
@@ -0,0 +1,4 @@
+enum E {
+  A = 1n,
+  B = 2n,
+}
diff --git a/src/parser/test/flow/enums/enum-valid-bigint-implicit.options.json b/src/parser/test/flow/enums/enum-valid-bigint-implicit.options.json
@@ -0,0 +1,3 @@
+{
+  "enums": true
+}
diff --git a/src/parser/test/flow/enums/enum-valid-bigint-implicit.tree.json b/src/parser/test/flow/enums/enum-valid-bigint-implicit.tree.json
@@ -0,0 +1,72 @@
+{
+  "type":"Program",
+  "loc":{"source":null,"start":{"line":1,"column":0},"end":{"line":4,"column":1}},
+  "range":[0,30],
+  "body":[
+    {
+      "type":"EnumDeclaration",
+      "loc":{"source":null,"start":{"line":1,"column":0},"end":{"line":4,"column":1}},
+      "range":[0,30],
+      "id":{
+        "type":"Identifier",
+        "loc":{"source":null,"start":{"line":1,"column":5},"end":{"line":1,"column":6}},
+        "range":[5,6],
+        "name":"E",
+        "typeAnnotation":null,
+        "optional":false
+      },
+      "body":{
+        "type":"EnumBigIntBody",
+        "loc":{"source":null,"start":{"line":1,"column":7},"end":{"line":4,"column":1}},
+        "range":[7,30],
+        "members":[
+          {
+            "type":"EnumBigIntMember",
+            "loc":{"source":null,"start":{"line":2,"column":2},"end":{"line":2,"column":8}},
+            "range":[11,17],
+            "id":{
+              "type":"Identifier",
+              "loc":{"source":null,"start":{"line":2,"column":2},"end":{"line":2,"column":3}},
+              "range":[11,12],
+              "name":"A",
+              "typeAnnotation":null,
+              "optional":false
+            },
+            "init":{
+              "type":"Literal",
+              "loc":{"source":null,"start":{"line":2,"column":6},"end":{"line":2,"column":8}},
+              "range":[15,17],
+              "value":null,
+              "bigint":"1",
+              "raw":"1n"
+            }
+          },
+          {
+            "type":"EnumBigIntMember",
+            "loc":{"source":null,"start":{"line":3,"column":2},"end":{"line":3,"column":8}},
+            "range":[21,27],
+            "id":{
+              "type":"Identifier",
+              "loc":{"source":null,"start":{"line":3,"column":2},"end":{"line":3,"column":3}},
+              "range":[21,22],
+              "name":"B",
+              "typeAnnotation":null,
+              "optional":false
+            },
+            "init":{
+              "type":"Literal",
+              "loc":{"source":null,"start":{"line":3,"column":6},"end":{"line":3,"column":8}},
+              "range":[25,27],
+              "value":null,
+              "bigint":"2",
+              "raw":"2n"
+            }
+          }
+        ],
+        "explicitType":false,
+        "hasUnknownMembers":false
+      }
+    }
+  ],
+  "comments":[]
+}

-Original file line number
+Diff line change
 +enum E of bigint {
 +  A = 1n,
 +  B = 2n,
 +}