Improve test case

As AST implementations like Roslyn directly tie a trailing line comment to the preceding node, we need to make sure that the comments on different lines are also all correctly handled.
openrewrite · Sep 25, 2024 · a04b919 · a04b919
1 parent 235e63c
commit a04b919
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/openrewrite/test/javascript/parser/expressionStatement.test.ts b/openrewrite/test/javascript/parser/expressionStatement.test.ts
@@ -13,8 +13,10 @@ describe('expression statement mapping', () => {
           javaScript(
             //language=ts
             `
-                1 ; // foo
-                2 ;`
+                1; // foo
+                // bar
+                /*baz*/
+                2;`
           )
         );
     });

diff --git a/openrewrite/test/javascript/parser/sourceFile.test.ts b/openrewrite/test/javascript/parser/sourceFile.test.ts
@@ -4,7 +4,13 @@ describe('source file mapping', () => {
     test('whitespace after last statement', () => {
         rewriteRunWithOptions(
           {normalizeIndent: false},
-          javaScript('1 ; // comment')
+          javaScript(
+            //language=typescript
+            `
+                1; /* comment 1 */
+                // comment 2
+            `
+          )
         );
     });
 });