Browse Source

osutil: update docstring and tests (#6255)

ᴜɴᴋɴᴡᴏɴ 3 years ago
parent
commit
c6143edb44
2 changed files with 4 additions and 10 deletions
  1. 1 1
      internal/osutil/osutil.go
  2. 3 9
      internal/osutil/osutil_test.go

+ 1 - 1
internal/osutil/osutil.go

@@ -34,7 +34,7 @@ func IsExist(path string) bool {
 	return err == nil || os.IsExist(err)
 }
 
-// CurrentUsername returns the current system user
+// CurrentUsername returns the username of the current user.
 func CurrentUsername() string {
 	username := os.Getenv("USER")
 	if len(username) > 0 {

+ 3 - 9
internal/osutil/osutil_test.go

@@ -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 {