Provide more source map info about statements in JS tests

js/js.tests/test/org/jetbrains/kotlin/js/test/utils/LineCollector.kt

@@ -58,7 +58,9 @@ class LineCollector : RecursiveJsVisitor() {
 
     override fun visitExpressionStatement(x: JsExpressionStatement) {
         withStatement(x) {
-            super.visitExpressionStatement(x)
+            handleNodeLocation(x.expression)
+            lineNumbersByStatement[x]?.add(-1)
+            x.expression.acceptChildren(this)
         }
     }
 
@@ -114,12 +116,16 @@ class LineCollector : RecursiveJsVisitor() {
 
     override fun visitReturn(x: JsReturn) {
         withStatement(x) {
+            handleNodeLocation(x)
+            lineNumbersByStatement[x]?.add(-1)
             super.visitReturn(x)
         }
     }
 
     override fun visitVars(x: JsVars) {
         withStatement(x) {
+            handleNodeLocation(x)
+            lineNumbersByStatement[x]?.add(-1)
             super.visitVars(x)
         }
     }
@@ -133,6 +139,8 @@ class LineCollector : RecursiveJsVisitor() {
 
     override fun visitThrow(x: JsThrow) {
         withStatement(x) {
+            handleNodeLocation(x)
+            lineNumbersByStatement[x]?.add(-1)
             super.visitThrow(x)
         }
     }