blob: 78b33922baa5ed0a2ce1e7a3042ce1c535a74528 [file] [log] [blame]
// Copyright 2016 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
//go:build unix && !android && !cgo && !darwin
package user
import (
"reflect"
"strings"
"testing"
)
var groupTests = []struct {
in string
name string
gid string
}{
{testGroupFile, "nobody", "-2"},
{testGroupFile, "kmem", "2"},
{testGroupFile, "notinthefile", ""},
{testGroupFile, "comment", ""},
{testGroupFile, "plussign", ""},
{testGroupFile, "+plussign", ""},
{testGroupFile, "-minussign", ""},
{testGroupFile, "minussign", ""},
{testGroupFile, "emptyid", ""},
{testGroupFile, "invalidgid", ""},
{testGroupFile, "indented", "7"},
{testGroupFile, "# comment", ""},
{testGroupFile, "largegroup", "1000"},
{testGroupFile, "manymembers", "777"},
{"", "emptyfile", ""},
}
func TestFindGroupName(t *testing.T) {
for _, tt := range groupTests {
got, err := findGroupName(tt.name, strings.NewReader(tt.in))
if tt.gid == "" {
if err == nil {
t.Errorf("findGroupName(%s): got nil error, expected err", tt.name)
continue
}
switch terr := err.(type) {
case UnknownGroupError:
if terr.Error() != "group: unknown group "+tt.name {
t.Errorf("findGroupName(%s): got %v, want %v", tt.name, terr, tt.name)
}
default:
t.Errorf("findGroupName(%s): got unexpected error %v", tt.name, terr)
}
} else {
if err != nil {
t.Fatalf("findGroupName(%s): got unexpected error %v", tt.name, err)
}
if got.Gid != tt.gid {
t.Errorf("findGroupName(%s): got gid %v, want %s", tt.name, got.Gid, tt.gid)
}
if got.Name != tt.name {
t.Errorf("findGroupName(%s): got name %s, want %s", tt.name, got.Name, tt.name)
}
}
}
}
var groupIdTests = []struct {
in string
gid string
name string
}{
{testGroupFile, "-2", "nobody"},
{testGroupFile, "2", "kmem"},
{testGroupFile, "notinthefile", ""},
{testGroupFile, "comment", ""},
{testGroupFile, "7", "indented"},
{testGroupFile, "4", ""},
{testGroupFile, "20", ""}, // row starts with a plus
{testGroupFile, "21", ""}, // row starts with a minus
{"", "emptyfile", ""},
}
func TestFindGroupId(t *testing.T) {
for _, tt := range groupIdTests {
got, err := findGroupId(tt.gid, strings.NewReader(tt.in))
if tt.name == "" {
if err == nil {
t.Errorf("findGroupId(%s): got nil error, expected err", tt.gid)
continue
}
switch terr := err.(type) {
case UnknownGroupIdError:
if terr.Error() != "group: unknown groupid "+tt.gid {
t.Errorf("findGroupId(%s): got %v, want %v", tt.name, terr, tt.name)
}
default:
t.Errorf("findGroupId(%s): got unexpected error %v", tt.name, terr)
}
} else {
if err != nil {
t.Fatalf("findGroupId(%s): got unexpected error %v", tt.name, err)
}
if got.Gid != tt.gid {
t.Errorf("findGroupId(%s): got gid %v, want %s", tt.name, got.Gid, tt.gid)
}
if got.Name != tt.name {
t.Errorf("findGroupId(%s): got name %s, want %s", tt.name, got.Name, tt.name)
}
}
}
}
const testUserFile = ` # Example user file
root:x:0:0:root:/root:/bin/bash
daemon:x:1:1:daemon:/usr/sbin:/usr/sbin/nologin
bin:x:2:3:bin:/bin:/usr/sbin/nologin
indented:x:3:3:indented:/dev:/usr/sbin/nologin
sync:x:4:65534:sync:/bin:/bin/sync
negative:x:-5:60:games:/usr/games:/usr/sbin/nologin
man:x:6:12:man:/var/cache/man:/usr/sbin/nologin
allfields:x:6:12:mansplit,man2,man3,man4:/home/allfields:/usr/sbin/nologin
+plussign:x:8:10:man:/var/cache/man:/usr/sbin/nologin
-minussign:x:9:10:man:/var/cache/man:/usr/sbin/nologin
malformed:x:27:12 # more:colons:after:comment
struid:x:notanumber:12 # more:colons:after:comment
# commented:x:28:12:commented:/var/cache/man:/usr/sbin/nologin
# commentindented:x:29:12:commentindented:/var/cache/man:/usr/sbin/nologin
struid2:x:30:badgid:struid2name:/home/struid:/usr/sbin/nologin
`
var userIdTests = []struct {
in string
uid string
name string
}{
{testUserFile, "-5", "negative"},
{testUserFile, "2", "bin"},
{testUserFile, "100", ""}, // not in the file
{testUserFile, "8", ""}, // plus sign, glibc doesn't find it
{testUserFile, "9", ""}, // minus sign, glibc doesn't find it
{testUserFile, "27", ""}, // malformed
{testUserFile, "28", ""}, // commented out
{testUserFile, "29", ""}, // commented out, indented
{testUserFile, "3", "indented"},
{testUserFile, "30", ""}, // the Gid is not valid, shouldn't match
{"", "1", ""},
}
func TestInvalidUserId(t *testing.T) {
_, err := findUserId("notanumber", strings.NewReader(""))
if err == nil {
t.Fatalf("findUserId('notanumber'): got nil error")
}
if want := "user: invalid userid notanumber"; err.Error() != want {
t.Errorf("findUserId('notanumber'): got %v, want %s", err, want)
}
}
func TestLookupUserId(t *testing.T) {
for _, tt := range userIdTests {
got, err := findUserId(tt.uid, strings.NewReader(tt.in))
if tt.name == "" {
if err == nil {
t.Errorf("findUserId(%s): got nil error, expected err", tt.uid)
continue
}
switch terr := err.(type) {
case UnknownUserIdError:
if want := "user: unknown userid " + tt.uid; terr.Error() != want {
t.Errorf("findUserId(%s): got %v, want %v", tt.name, terr, want)
}
default:
t.Errorf("findUserId(%s): got unexpected error %v", tt.name, terr)
}
} else {
if err != nil {
t.Fatalf("findUserId(%s): got unexpected error %v", tt.name, err)
}
if got.Uid != tt.uid {
t.Errorf("findUserId(%s): got uid %v, want %s", tt.name, got.Uid, tt.uid)
}
if got.Username != tt.name {
t.Errorf("findUserId(%s): got name %s, want %s", tt.name, got.Username, tt.name)
}
}
}
}
func TestLookupUserPopulatesAllFields(t *testing.T) {
u, err := findUsername("allfields", strings.NewReader(testUserFile))
if err != nil {
t.Fatal(err)
}
want := &User{
Username: "allfields",
Uid: "6",
Gid: "12",
Name: "mansplit",
HomeDir: "/home/allfields",
}
if !reflect.DeepEqual(u, want) {
t.Errorf("findUsername: got %#v, want %#v", u, want)
}
}
var userTests = []struct {
in string
name string
uid string
}{
{testUserFile, "negative", "-5"},
{testUserFile, "bin", "2"},
{testUserFile, "notinthefile", ""},
{testUserFile, "indented", "3"},
{testUserFile, "plussign", ""},
{testUserFile, "+plussign", ""},
{testUserFile, "minussign", ""},
{testUserFile, "-minussign", ""},
{testUserFile, " indented", ""},
{testUserFile, "commented", ""},
{testUserFile, "commentindented", ""},
{testUserFile, "malformed", ""},
{testUserFile, "# commented", ""},
{"", "emptyfile", ""},
}
func TestLookupUser(t *testing.T) {
for _, tt := range userTests {
got, err := findUsername(tt.name, strings.NewReader(tt.in))
if tt.uid == "" {
if err == nil {
t.Errorf("lookupUser(%s): got nil error, expected err", tt.uid)
continue
}
switch terr := err.(type) {
case UnknownUserError:
if want := "user: unknown user " + tt.name; terr.Error() != want {
t.Errorf("lookupUser(%s): got %v, want %v", tt.name, terr, want)
}
default:
t.Errorf("lookupUser(%s): got unexpected error %v", tt.name, terr)
}
} else {
if err != nil {
t.Fatalf("lookupUser(%s): got unexpected error %v", tt.name, err)
}
if got.Uid != tt.uid {
t.Errorf("lookupUser(%s): got uid %v, want %s", tt.name, got.Uid, tt.uid)
}
if got.Username != tt.name {
t.Errorf("lookupUser(%s): got name %s, want %s", tt.name, got.Username, tt.name)
}
}
}
}