osutil: update docstring and tests (#6255)

This commit is contained in:
ᴜɴᴋɴᴡᴏɴ
2020-08-19 21:30:01 +08:00
committed by GitHub
parent 252d0fd977
commit c6143edb44
2 changed files with 4 additions and 10 deletions

View File

@@ -81,16 +81,10 @@ func TestIsExist(t *testing.T) {
}
func TestCurrentUsername(t *testing.T) {
// Make sure it does not blow up
CurrentUsername()
}
func TestCurrentUsernamePrefersEnvironmentVariable(t *testing.T) {
// Some users/scripts expect that they can change the current username via environment variables
if userBak, ok := os.LookupEnv("USER"); ok {
defer os.Setenv("USER", userBak)
if oldUser, ok := os.LookupEnv("USER"); ok {
defer func() { _ = os.Setenv("USER", oldUser) }()
} else {
defer os.Unsetenv("USER")
defer func() { _ = os.Unsetenv("USER") }()
}
if err := os.Setenv("USER", "__TESTING::USERNAME"); err != nil {