refactor: enhance stdin script handling by improving string trimming logic

1 files changed, 6 insertions, 1 deletions

diff --git a/tests/modules/stdin/test.lua b/tests/modules/stdin/test.lua
index 6b1b4026b..b0c908c3c 100644
--- a/tests/modules/stdin/test.lua
+++ b/tests/modules/stdin/test.lua
@@ -113,8 +113,13 @@ function main(t)
             "2"
         )
         test_shell(
+            "pwsh_main",
+            string.format('%s -c "echo \\"function main() print(\'in_pwsh_main\') end\\" | %s"', pwsh, run_stdin),
+            "in_pwsh_main"
+        )
+        test_shell(
             "pwsh_multi",
-            string.format("%s -c \"echo \\\"print('pline1')\\nprint('pline2')\\\" | %s l --stdin\"", pwsh, xmake),
+            string.format('%s -c "echo \\"print(\'pline1\')\\" \\"print(\'pline2\')\\" | %s"', pwsh, run_stdin),
             "pline1[\r\n]+pline2"
         )
     else