From a04b919abd79670cc68273945f0259b3910ab893 Mon Sep 17 00:00:00 2001
From: Knut Wannheden <knut@moderne.io>
Date: Wed, 25 Sep 2024 15:30:04 +0200
Subject: [PATCH] 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.
---
 .../test/javascript/parser/expressionStatement.test.ts    | 6 ++++--
 openrewrite/test/javascript/parser/sourceFile.test.ts     | 8 +++++++-
 2 files changed, 11 insertions(+), 3 deletions(-)

diff --git a/openrewrite/test/javascript/parser/expressionStatement.test.ts b/openrewrite/test/javascript/parser/expressionStatement.test.ts
index 767b18b..d8581e0 100644
--- 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
index 366c8fd..ed534f5 100644
--- 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
+            `
+          )
         );
     });
 });