Make parser tests a bit nicer

openrewrite · Sep 24, 2024 · fbc42e6 · fbc42e6
1 parent 2727213
commit fbc42e6
Show file tree

Hide file tree

Showing 3 changed files with 38 additions and 6 deletions.
diff --git a/openrewrite/package-lock.json b/openrewrite/package-lock.json
diff --git a/openrewrite/package.json b/openrewrite/package.json
@@ -19,6 +19,7 @@
   },
   "dependencies": {
     "@types/node": "^22.5.4",
+    "dedent": "^1.5.3",
     "diff": "^7.0.0",
     "uuid": "^10.0.0"
   },

diff --git a/openrewrite/test/javascript/parser.test.ts b/openrewrite/test/javascript/parser.test.ts
@@ -2,23 +2,54 @@ import {InMemoryExecutionContext, ParserInput} from '../../src/core';
 import {JavaScriptParser} from "../../src/javascript";
 import * as J from "../../src/java/tree";
 import * as JS from "../../src/javascript/tree";
+import dedent from "dedent";
+
+describe('Parser API', () => {
+    const parser = JavaScriptParser.builder().build();
 
-describe('JavaScriptParser', () => {
     test('parseInputs', () => {
-        const parser = JavaScriptParser.builder().build();
-        const [sourceFile] = parser.parseInputs([new ParserInput('foo.ts', null, true, () => Buffer.from('1', 'utf8'))], null, new InMemoryExecutionContext());
-        console.log(sourceFile);
+        const [sourceFile] = parser.parseInputs(
+            [new ParserInput('foo.ts', null, true, () => Buffer.from('1', 'utf8'))],
+            null,
+            new InMemoryExecutionContext()
+        ) as Iterable<JS.CompilationUnit>;
+        expect(sourceFile).toBeDefined();
     });
 
     test('parseStrings', () => {
-        const parser = JavaScriptParser.builder().build();
         const [sourceFile] = parser.parseStrings(`
         const c = 1;
         /* c1*/  /*c2 */const d = 1;`) as Iterable<JS.CompilationUnit>;
         expect(sourceFile).toBeDefined();
+    });
+});
+
+describe('LST mapping', () => {
+    const parser = JavaScriptParser.builder().build();
+
+    test('parseInputs', () => {
+        const sourceFile = javaScript('1');
+        expect(sourceFile).toBeDefined();
+        expect(sourceFile.statements).toHaveLength(1);
+        expect(sourceFile.statements[0]).toBeInstanceOf(JS.ExpressionStatement);
+    });
+
+    test('parseStrings', () => {
+        //language=typescript
+        const sourceFile = javaScript(`
+            const c = 1;
+            /* c1*/  /*c2 */
+            const d = 1;
+        `);
+        expect(sourceFile).toBeDefined();
         expect(sourceFile.statements).toHaveLength(2);
         sourceFile.statements.forEach(statement => {
             expect(statement).toBeInstanceOf(J.Unknown);
         })
     });
+
+    function javaScript(source: string): JS.CompilationUnit {
+        const [sourceFile] = parser.parseStrings(dedent(source)) as Iterable<JS.CompilationUnit>;
+        return sourceFile;
+    }
 });