Add and implement normal RunPowerShell function and tests related to it.

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

Author: Aliwoto

ssg/helpers.go

@@ -256,6 +256,10 @@ func RunCommand(command string) *ExecuteCommandResult {
 	return shellUtils.RunCommand(command)
 }
 
+func RunPowerShell(command string) *ExecuteCommandResult {
+	return shellUtils.RunPowerShell(command)
+}
+
 func RunCommandAsync(command string) *ExecuteCommandResult {
 	return shellUtils.RunCommandAsync(command)
 }

ssg/shellUtils/helpers.go

@@ -13,6 +13,10 @@ func RunCommand(command string) *ExecuteCommandResult {
 	return runCommand(command, false, nil)
 }
 
+func RunPowerShell(command string) *ExecuteCommandResult {
+	return runPowerShell(command, false, nil)
+}
+
 func RunCommandAsync(command string) *ExecuteCommandResult {
 	return runCommand(command, true, nil)
 }
@@ -73,7 +77,7 @@ func runPowerShell(command string, isAsync bool, finishedChan chan bool) *Execut
 		return result
 	} else {
 		result = executePowerShell(command, &ExecuteCommandConfig{
-			TargetRunner:           GetCommandTargetRunner(),
+			TargetRunner:           GetPowerShellRunner(),
 			PrimaryArgs:            GetPowerShellPrimaryArgs(),
 			Stdout:                 stdout,
 			Stderr:                 stderr,
@@ -194,7 +198,7 @@ func executePowerShell(
 
 	if config.RemovePowerShellPrompt && !strings.Contains(command, "function prompt") {
 		// hacky way of getting rid of powershell prompt
-		command = PowerShellPromptOverride + command
+		command = PowerShellPromptOverride + "\n" + command
 	}
 
 	cmd = exec.Command(config.TargetRunner, config.PrimaryArgs...)

tests/powershell_test.go

@@ -1,6 +1,7 @@
 package tests
 
 import (
+	"fmt"
 	"log"
 	"testing"
 
@@ -24,5 +25,23 @@ func TestPowerShell01(t *testing.T) {
 
 	result.PurifyPowerShellOutput()
 
+	fmt.Println(result.Stdout)
+}
+
+func TestPowerShell02(t *testing.T) {
+	result := ssg.RunPowerShell("$PSVersionTable.PSVersion")
+	result.PurifyPowerShellOutput()
+
+	fmt.Println(result.Stdout)
+}
+
+func TestPowerShell03(t *testing.T) {
+	finishedChan := make(chan bool)
+	result := ssg.RunPowerShellAsyncWithChan("$PSVersionTable.PSVersion", finishedChan)
+
+	<-finishedChan
+
+	result.PurifyPowerShellOutput()
+
 	log.Println(result.Stdout)
 }