Fix minor issues.

Fix the issue that was causing the GetPurifiedPowerShellOutput function to return invalid value.

Signed-off-by: Aliwoto <[email protected]>

Author: Aliwoto

ssg/shellUtils/helpers.go

@@ -198,7 +198,7 @@ func executePowerShell(
 
 	if config.RemovePowerShellPrompt && !strings.Contains(command, "function prompt") {
 		// hacky way of getting rid of powershell prompt
-		command = PowerShellPromptOverride + "\n" + command
+		command = PowerShellPromptOverride + "\n" + command + "\n"
 	}
 
 	cmd = exec.Command(config.TargetRunner, config.PrimaryArgs...)

tests/powershell_test.go

@@ -29,6 +29,7 @@ func TestPowerShell01(t *testing.T) {
 }
 
 func TestPowerShell02(t *testing.T) {
+	//Write-Output ("ok" + $PSVersionTable.PSVersion.Major.ToString() + "`n`n" + $PSVersionTable.PSVersion)
 	result := ssg.RunPowerShell("$PSVersionTable.PSVersion")
 	result.PurifyPowerShellOutput()
 
@@ -45,3 +46,10 @@ func TestPowerShell03(t *testing.T) {
 
 	log.Println(result.Stdout)
 }
+
+func TestPowerShell04(t *testing.T) {
+	result := ssg.RunPowerShell("Write-Output (\"ok\" + $PSVersionTable.PSVersion.Major.ToString() + \"`n`n\" + $PSVersionTable.PSVersion)")
+	result.PurifyPowerShellOutput()
+
+	fmt.Println(result.Stdout)
+}