mocks_test.go

package main

import (
	"os/user"
)

func mockLookupUser(username string) (*user.User, error) {
	if username == "alice" || username == "bob" {
		return &user.User{Username: username}, nil
	}
	return nil, user.UnknownUserError(username)
}

func mockLookupGroup(groupname string) (*user.Group, error) {
	if groupname == "dev" || groupname == "admins" {
		return &user.Group{Name: groupname}, nil
	}
	return nil, user.UnknownGroupError(groupname)
}

// setMocks overrides the real lookup functions with mocks and returns a cleanup function
func setMocks() func() {
	originalLookupUser := lookupUser
	originalLookupGroup := lookupGroup

	lookupUser = mockLookupUser
	lookupGroup = mockLookupGroup

	return func() {
		lookupUser = originalLookupUser
		lookupGroup = originalLookupGroup
	}
}