mirror of https://github.com/rancher/wins.git
55 lines
988 B
Go
55 lines
988 B
Go
package powershell
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestExecuteCommand(t *testing.T) {
|
|
type args struct {
|
|
command string
|
|
}
|
|
|
|
tests := []struct {
|
|
name string
|
|
args args
|
|
want string
|
|
error bool
|
|
}{
|
|
{
|
|
name: "write to standard out",
|
|
args: args{command: "Write-Output 'test-value'"},
|
|
want: "test-value",
|
|
error: false,
|
|
},
|
|
{
|
|
name: "write to host",
|
|
args: args{command: "Write-Host 'test-value'"},
|
|
want: "test-value",
|
|
error: false,
|
|
},
|
|
{
|
|
name: "write to standard error",
|
|
args: args{command: "Write-Error 'test-value'"},
|
|
want: "test-value",
|
|
error: true,
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
o, err := RunCommand(tt.args.command)
|
|
if err != nil && !tt.error {
|
|
t.Errorf("error occurred, " + err.Error())
|
|
}
|
|
|
|
if !tt.error {
|
|
got := strings.TrimSpace(string(o))
|
|
if got != tt.want {
|
|
t.Errorf("error, should be " + tt.want + ", but got " + got)
|
|
}
|
|
}
|
|
})
|
|
}
|
|
}
|