githaven/modules/user/user_test.go

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)
	}
}