2020-04-05 01:20:50 -05:00
|
|
|
// Copyright 2020 The Gitea Authors. All rights reserved.
|
2022-11-27 13:20:29 -05:00
|
|
|
// SPDX-License-Identifier: MIT
|
2020-04-05 01:20:50 -05:00
|
|
|
|
2016-11-06 03:47:25 -05:00
|
|
|
package user
|
|
|
|
|
|
|
|
import (
|
2018-10-18 02:08:20 -05:00
|
|
|
"os/exec"
|
|
|
|
"runtime"
|
|
|
|
"strings"
|
2016-11-06 03:47:25 -05:00
|
|
|
"testing"
|
|
|
|
)
|
|
|
|
|
2018-10-18 02:08:20 -05:00
|
|
|
func getWhoamiOutput() (string, error) {
|
|
|
|
output, err := exec.Command("whoami").Output()
|
|
|
|
if err != nil {
|
|
|
|
return "", err
|
|
|
|
}
|
2019-06-12 14:41:28 -05:00
|
|
|
return strings.TrimSpace(string(output)), nil
|
2018-10-18 02:08:20 -05:00
|
|
|
}
|
2016-11-06 03:47:25 -05:00
|
|
|
|
2018-10-18 02:08:20 -05:00
|
|
|
func TestCurrentUsername(t *testing.T) {
|
2016-11-06 03:47:25 -05:00
|
|
|
user := CurrentUsername()
|
2018-10-20 16:25:14 -05:00
|
|
|
if len(user) == 0 {
|
2018-10-18 02:08:20 -05:00
|
|
|
t.Errorf("expected non-empty user, got: %s", user)
|
2016-11-06 03:47:25 -05:00
|
|
|
}
|
2018-10-18 02:08:20 -05:00
|
|
|
// 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)
|
|
|
|
}
|
2024-07-17 16:07:41 -05:00
|
|
|
t.Setenv("USER", "spoofed")
|
2016-11-06 03:47:25 -05:00
|
|
|
user = CurrentUsername()
|
2018-10-18 02:08:20 -05:00
|
|
|
if user != whoami {
|
|
|
|
t.Errorf("expected %s as user, got: %s", whoami, user)
|
2016-11-06 03:47:25 -05:00
|
|
|
}
|
|
|
|
}
|