summaryrefslogtreecommitdiffstats
path: root/modules/user/user_test.go
diff options
context:
space:
mode:
Diffstat (limited to '')
-rw-r--r--modules/user/user_test.go43
1 files changed, 43 insertions, 0 deletions
diff --git a/modules/user/user_test.go b/modules/user/user_test.go
new file mode 100644
index 0000000..372a675
--- /dev/null
+++ b/modules/user/user_test.go
@@ -0,0 +1,43 @@
+// Copyright 2020 The Gitea Authors. All rights reserved.
+// SPDX-License-Identifier: MIT
+
+package user
+
+import (
+ "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)
+ }
+ t.Setenv("USER", "spoofed")
+ user = CurrentUsername()
+ if user != whoami {
+ t.Errorf("expected %s as user, got: %s", whoami, user)
+ }
+}