correct parse nested arrays in the ndoejs driver

apache · Aug 20, 2024 · aaf33e3 · aaf33e3
1 parent 05487a0
commit aaf33e3
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/drivers/nodejs/src/antlr4/CustomAgTypeListener.ts b/drivers/nodejs/src/antlr4/CustomAgTypeListener.ts
@@ -145,6 +145,10 @@ class CustomAgTypeListener implements AgtypeListener, ParseTreeListener {
     this.mergeArray()
   }
 
+  exitArrayValue(): void {
+    this.mergeArray();
+  }
+
   exitPair (ctx: PairContext): void {
     const name = this.stripQuotes(ctx.STRING().text)
 

diff --git a/drivers/nodejs/test/Agtype.test.ts b/drivers/nodejs/test/Agtype.test.ts
@@ -101,4 +101,16 @@ describe('Parsing', () => {
       }))
     })))
   })
+
+  it("Nested Array Parsing", () => {
+    expect(
+      AGTypeParse(
+        '{"id": 1125899906842627, "label": "car", "properties": {"wheels": [ "a", ["d"] ]}}::vertex',
+      ),
+    ).toStrictEqual(new Map<string, any>(Object.entries({
+      id: 1125899906842627,
+      label: "car",
+      properties: new Map<string, any>(Object.entries({ wheels: ["a", ["d"]] })),
+    })));
+  });
 })