forked from Shiloh/githaven
e81ccc406b
Change all license headers to comply with REUSE specification. Fix #16132 Co-authored-by: flynnnnnnnnnn <flynnnnnnnnnn@github> Co-authored-by: John Olheiser <john.olheiser@gmail.com>
45 lines
1006 B
Go
45 lines
1006 B
Go
// Copyright 2020 The Gitea Authors. All rights reserved.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
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)
|
|
}
|
|
}
|