From 70a9f4dffda9bad7644e32309a964629984a2d86 Mon Sep 17 00:00:00 2001 From: Sage Pointer Date: Thu, 13 Sep 2018 14:36:04 +0200 Subject: [PATCH 1/2] Use native go method to get current user rather than environment var --- modules/user/user.go | 21 ++++++++++++++++++++- modules/user/user_test.go | 37 ++++++++++++++++++++++++++++--------- 2 files changed, 48 insertions(+), 10 deletions(-) diff --git a/modules/user/user.go b/modules/user/user.go index b9911e6f52563..af49d7b06a4b9 100644 --- a/modules/user/user.go +++ b/modules/user/user.go @@ -4,10 +4,29 @@ package user -import "os" +import ( + "os" + "os/user" + "runtime" + "strings" +) // CurrentUsername return current login OS user name func CurrentUsername() string { + userinfo, err := user.Current() + if err != nil { + return fallbackCurrentUsername() + } + username := userinfo.Username + if runtime.GOOS == "windows" { + parts := strings.Split(username, "\\") + username = parts[len(parts)-1] + } + return username +} + +// Old method, used if new method doesn't work on your OS for some reason +func fallbackCurrentUsername() string { curUserName := os.Getenv("USER") if len(curUserName) > 0 { return curUserName diff --git a/modules/user/user_test.go b/modules/user/user_test.go index d7df15153166a..40660a4bff47a 100644 --- a/modules/user/user_test.go +++ b/modules/user/user_test.go @@ -2,21 +2,40 @@ package user import ( "os" + "os/exec" + "runtime" + "strings" "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() - 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" { + return + } + 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 != "gitea" { - t.Errorf("expected gitea as user, got: %s", user) + if user != whoami { + t.Errorf("expected %s as user, got: %s", whoami, user) } } From 02b461735acb6d4780ec0ab7e34b9147a6b67d07 Mon Sep 17 00:00:00 2001 From: SagePtr Date: Tue, 18 Sep 2018 20:43:05 +0200 Subject: [PATCH 2/2] Use t.Skip instead of return in test --- modules/user/user_test.go | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modules/user/user_test.go b/modules/user/user_test.go index 40660a4bff47a..51f10dbbd23ed 100644 --- a/modules/user/user_test.go +++ b/modules/user/user_test.go @@ -23,7 +23,7 @@ func TestCurrentUsername(t *testing.T) { } // Windows whoami is weird, so just skip remaining tests if runtime.GOOS == "windows" { - return + t.Skip("skipped test because of weird whoami on Windows") } whoami, err := getWhoamiOutput() if err != nil {