mirror of
https://github.com/taigrr/wtf
synced 2025-01-18 04:03:14 -08:00
51 lines
1009 B
Go
51 lines
1009 B
Go
package utils
|
|
|
|
import (
|
|
"os/exec"
|
|
"testing"
|
|
|
|
. "github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func Test_ExecuteCommand(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
cmd *exec.Cmd
|
|
expected string
|
|
}{
|
|
{
|
|
name: "with nil command",
|
|
cmd: nil,
|
|
expected: "",
|
|
},
|
|
{
|
|
name: "with defined command",
|
|
cmd: exec.Command("echo", "cats"),
|
|
expected: "cats\n",
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
actual := ExecuteCommand(tt.cmd)
|
|
|
|
if tt.expected != actual {
|
|
t.Errorf("\nexpected: %s\n got: %s", tt.expected, actual)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func Test_FindMatch(t *testing.T) {
|
|
var result [][]string
|
|
|
|
expected := [][]string([][]string{[]string{"SSID: 7E5B5C", "7E5B5C"}})
|
|
result = FindMatch(`s*SSID: (.+)s*`, "SSID: 7E5B5C")
|
|
Equal(t, expected, result)
|
|
}
|
|
|
|
func Test_ExcludeWhenTrue(t *testing.T) {
|
|
Equal(t, true, Exclude([]string{"cat", "dog", "rat"}, "bat"))
|
|
Equal(t, false, Exclude([]string{"cat", "dog", "rat"}, "dog"))
|
|
}
|