1// Copyright 2021 The Go Authors. All rights reserved. 2// Use of this source code is governed by a BSD-style 3// license that can be found in the LICENSE file. 4 5//go:build ((darwin || dragonfly || freebsd || (js && wasm) || (!android && linux) || netbsd || openbsd || solaris) && (!cgo || osusergo)) || aix || illumos 6 7package user 8 9import ( 10 "fmt" 11 "sort" 12 "strings" 13 "testing" 14) 15 16var testGroupFile = `# See the opendirectoryd(8) man page for additional 17# information about Open Directory. 18## 19nobody:*:-2: 20nogroup:*:-1: 21wheel:*:0:root 22emptyid:*::root 23invalidgid:*:notanumber:root 24+plussign:*:20:root 25-minussign:*:21:root 26# Next line is invalid (empty group name) 27:*:22:root 28 29daemon:*:1:root 30 indented:*:7:root 31# comment:*:4:found 32 # comment:*:4:found 33kmem:*:2:root 34manymembers:x:777:jill,jody,john,jack,jov,user777 35` + largeGroup() 36 37func largeGroup() (res string) { 38 var b strings.Builder 39 b.WriteString("largegroup:x:1000:user1") 40 for i := 2; i <= 7500; i++ { 41 fmt.Fprintf(&b, ",user%d", i) 42 } 43 return b.String() 44} 45 46var listGroupsTests = []struct { 47 // input 48 in string 49 user string 50 gid string 51 // output 52 gids []string 53 err bool 54}{ 55 {in: testGroupFile, user: "root", gid: "0", gids: []string{"0", "1", "2", "7"}}, 56 {in: testGroupFile, user: "jill", gid: "33", gids: []string{"33", "777"}}, 57 {in: testGroupFile, user: "jody", gid: "34", gids: []string{"34", "777"}}, 58 {in: testGroupFile, user: "john", gid: "35", gids: []string{"35", "777"}}, 59 {in: testGroupFile, user: "jov", gid: "37", gids: []string{"37", "777"}}, 60 {in: testGroupFile, user: "user777", gid: "7", gids: []string{"7", "777", "1000"}}, 61 {in: testGroupFile, user: "user1111", gid: "1111", gids: []string{"1111", "1000"}}, 62 {in: testGroupFile, user: "user1000", gid: "1000", gids: []string{"1000"}}, 63 {in: testGroupFile, user: "user7500", gid: "7500", gids: []string{"1000", "7500"}}, 64 {in: testGroupFile, user: "no-such-user", gid: "2345", gids: []string{"2345"}}, 65 {in: "", user: "no-such-user", gid: "2345", gids: []string{"2345"}}, 66 // Error cases. 67 {in: "", user: "", gid: "2345", err: true}, 68 {in: "", user: "joanna", gid: "bad", err: true}, 69} 70 71func TestListGroups(t *testing.T) { 72 for _, tc := range listGroupsTests { 73 u := &User{Username: tc.user, Gid: tc.gid} 74 got, err := listGroupsFromReader(u, strings.NewReader(tc.in)) 75 if tc.err { 76 if err == nil { 77 t.Errorf("listGroups(%q): got nil; want error", tc.user) 78 } 79 continue // no more checks 80 } 81 if err != nil { 82 t.Errorf("listGroups(%q): got %v error, want nil", tc.user, err) 83 continue // no more checks 84 } 85 checkSameIDs(t, got, tc.gids) 86 } 87} 88 89func checkSameIDs(t *testing.T, got, want []string) { 90 t.Helper() 91 if len(got) != len(want) { 92 t.Errorf("ID list mismatch: got %v; want %v", got, want) 93 return 94 } 95 sort.Strings(got) 96 sort.Strings(want) 97 mismatch := -1 98 for i, g := range want { 99 if got[i] != g { 100 mismatch = i 101 break 102 } 103 } 104 if mismatch != -1 { 105 t.Errorf("ID list mismatch (at index %d): got %v; want %v", mismatch, got, want) 106 } 107} 108