|
@ -2,21 +2,40 @@ package user |
|
|
|
|
|
|
|
|
import ( |
|
|
import ( |
|
|
"os" |
|
|
"os" |
|
|
|
|
|
"os/exec" |
|
|
|
|
|
"runtime" |
|
|
|
|
|
"strings" |
|
|
"testing" |
|
|
"testing" |
|
|
) |
|
|
) |
|
|
|
|
|
|
|
|
func TestCurrentUsername(t *testing.T) { |
|
|
|
|
|
os.Setenv("USER", "") |
|
|
|
|
|
os.Setenv("USERNAME", "foobar") |
|
|
|
|
|
|
|
|
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() |
|
|
user := CurrentUsername() |
|
|
if user != "foobar" { |
|
|
|
|
|
t.Errorf("expected foobar as user, got: %s", user) |
|
|
|
|
|
|
|
|
if len(user) <= 0 { |
|
|
|
|
|
t.Errorf("expected non-empty user, got: %s", user) |
|
|
} |
|
|
} |
|
|
|
|
|
|
|
|
os.Setenv("USER", "gitea") |
|
|
|
|
|
|
|
|
// 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() |
|
|
user = CurrentUsername() |
|
|
if user != "gitea" { |
|
|
|
|
|
t.Errorf("expected gitea as user, got: %s", user) |
|
|
|
|
|
|
|
|
if user != whoami { |
|
|
|
|
|
t.Errorf("expected %s as user, got: %s", whoami, user) |
|
|
} |
|
|
} |
|
|
} |
|
|
} |