* Use native go method to get current user rather than environment var * Use t.Skip instead of return in test
41 lines
918 B
Go
41 lines
918 B
Go
package user
|
|
|
|
import (
|
|
"os"
|
|
"os/exec"
|
|
"runtime"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func getWhoamiOutput() (string, error) {
|
|
output, err := exec.Command("whoami").Output()
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
return strings.TrimSpace(string(output[:])), nil
|
|
}
|
|
|
|
func TestCurrentUsername(t *testing.T) {
|
|
user := CurrentUsername()
|
|
if len(user) <= 0 {
|
|
t.Errorf("expected non-empty user, got: %s", user)
|
|
}
|
|
// Windows whoami is weird, so just skip remaining tests
|
|
if runtime.GOOS == "windows" {
|
|
t.Skip("skipped test because of weird whoami on Windows")
|
|
}
|
|
whoami, err := getWhoamiOutput()
|
|
if err != nil {
|
|
t.Errorf("failed to run whoami to test current user: %f", err)
|
|
}
|
|
user = CurrentUsername()
|
|
if user != whoami {
|
|
t.Errorf("expected %s as user, got: %s", whoami, user)
|
|
}
|
|
os.Setenv("USER", "spoofed")
|
|
user = CurrentUsername()
|
|
if user != whoami {
|
|
t.Errorf("expected %s as user, got: %s", whoami, user)
|
|
}
|
|
}
|