# $NetBSD: t_id.sh,v 1.2 2024/04/28 07:27:42 rillig Exp $
#
# Copyright (c) 2007, 2008 The NetBSD Foundation, Inc.
# All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions
# are met:
# 1. Redistributions of source code must retain the above copyright
#    notice, this list of conditions and the following disclaimer.
# 2. Redistributions in binary form must reproduce the above copyright
#    notice, this list of conditions and the following disclaimer in the
#    documentation and/or other materials provided with the distribution.
#
# THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
# ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
# TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
# PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
# BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
# CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
# SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
# CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
# ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
# POSSIBILITY OF SUCH DAMAGE.
#

create_run_id() {
       cat >run_id.sh <<EOF
#! /bin/sh
[ -f ./id ] || ln -s $(atf_get_srcdir)/h_id ./id
/id "\${@}"
EOF
       chmod +x run_id.sh
}

atf_test_case default
default_head() {
       atf_set "descr" "Checks that the output without options is correct"
}
default_body() {
       create_run_id

       echo "uid=100(test) gid=100(users) groups=100(users),0(wheel)" >expout
       atf_check -s exit:0 -o file:expout -e empty ./run_id.sh
       atf_check -s exit:0 -o file:expout -e empty ./run_id.sh 100
       atf_check -s exit:0 -o file:expout -e empty ./run_id.sh test

       echo "uid=0(root) gid=0(wheel) groups=0(wheel)" >expout
       atf_check -s exit:0 -o file:expout -e empty ./run_id.sh 0
       atf_check -s exit:0 -o file:expout -e empty ./run_id.sh root

       export LIBFAKE_EGID_ROOT=1 LIBFAKE_EUID_ROOT=1
       echo "uid=100(test) gid=100(users) euid=0(root) egid=0(wheel) groups=100(users),0(wheel)" >expout
       atf_check -s exit:0 -o file:expout -e empty ./run_id.sh
       unset LIBFAKE_EGID_ROOT LIBFAKE_EUID_ROOT

       echo 'id: nonexistent: No such user' >experr
       atf_check -s exit:1 -o empty -e file:experr ./run_id.sh nonexistent

       atf_check -s exit:1 -o empty -e save:stderr ./run_id.sh root nonexistent
       atf_check -s exit:0 -o ignore -e empty grep ^usage: stderr
}

atf_test_case primaries
primaries_head() {
       atf_set "descr" "Checks that giving multiple primaries" \
                       "simultaneously provides an error"
}
primaries_body() {
       create_run_id

       for p1 in -G -g -p -u; do
               for p2 in -G -g -p -u; do
                       if [ ${p1} != ${p2} ]; then
                               atf_check -s exit:1 -o empty -e save:stderr \
                                   ./run_id.sh ${p1} ${p2}
                               atf_check -s exit:0 -o ignore -e empty \
                                   grep ^usage: stderr
                       fi
               done
       done
}

atf_test_case Gflag
Gflag_head() {
       atf_set "descr" "Checks that the -G primary flag works"
}
Gflag_body() {
       create_run_id

       echo "100 0" >expout
       atf_check -s exit:0 -o file:expout -e empty ./run_id.sh -G
       atf_check -s exit:0 -o file:expout -e empty ./run_id.sh -G 100
       atf_check -s exit:0 -o file:expout -e empty ./run_id.sh -G test

       echo "users wheel" >expout
       atf_check -s exit:0 -o file:expout -e empty ./run_id.sh -G -n
       atf_check -s exit:0 -o file:expout -e empty ./run_id.sh -G -n 100
       atf_check -s exit:0 -o file:expout -e empty ./run_id.sh -G -n test

       echo "0" >expout
       atf_check -s exit:0 -o file:expout -e empty ./run_id.sh -G 0
       atf_check -s exit:0 -o file:expout -e empty ./run_id.sh -G root

       echo "wheel" >expout
       atf_check -s exit:0 -o file:expout -e empty ./run_id.sh -G -n 0
       atf_check -s exit:0 -o file:expout -e empty ./run_id.sh -G -n root

       echo 'id: nonexistent: No such user' >experr
       atf_check -s exit:1 -o empty -e file:experr ./run_id.sh -G nonexistent

       atf_check -s exit:1 -o empty -e save:stderr ./run_id.sh -G root nonexistent
       atf_check -s exit:0 -o ignore -e empty grep ^usage: stderr
}

atf_test_case gflag
gflag_head() {
       atf_set "descr" "Checks that the -g primary flag works"
}
gflag_body() {
       create_run_id

       echo "100" >expout
       atf_check -s exit:0 -o file:expout -e empty ./run_id.sh -g
       atf_check -s exit:0 -o file:expout -e empty ./run_id.sh -g 100
       atf_check -s exit:0 -o file:expout -e empty ./run_id.sh -g test

       echo "users" >expout
       atf_check -s exit:0 -o file:expout -e empty ./run_id.sh -g -n
       atf_check -s exit:0 -o file:expout -e empty ./run_id.sh -g -n 100
       atf_check -s exit:0 -o file:expout -e empty ./run_id.sh -g -n test

       echo "0" >expout
       atf_check -s exit:0 -o file:expout -e empty ./run_id.sh -g 0
       atf_check -s exit:0 -o file:expout -e empty ./run_id.sh -g root

       echo "wheel" >expout
       atf_check -s exit:0 -o file:expout -e empty ./run_id.sh -g -n 0
       atf_check -s exit:0 -o file:expout -e empty ./run_id.sh -g -n root

       echo "100" >expout
       atf_check -s exit:0 -o file:expout -e empty ./run_id.sh -g -r

       echo "users" >expout
       atf_check -s exit:0 -o file:expout -e empty ./run_id.sh -g -r -n

       echo "100" >expout
       atf_check -s exit:0 -o file:expout -e empty ./run_id.sh -g -r 100
       atf_check -s exit:0 -o file:expout -e empty ./run_id.sh -g -r test

       echo "users" >expout
       atf_check -s exit:0 -o file:expout -e empty ./run_id.sh -g -r -n 100
       atf_check -s exit:0 -o file:expout -e empty ./run_id.sh -g -r -n test

       export LIBFAKE_EGID_ROOT=1 LIBFAKE_EUID_ROOT=1

       echo "0" >expout
       atf_check -s exit:0 -o file:expout -e empty ./run_id.sh -g

       echo "wheel" >expout
       atf_check -s exit:0 -o file:expout -e empty ./run_id.sh -g -n

       echo "100" >expout
       atf_check -s exit:0 -o file:expout -e empty ./run_id.sh -g -r

       echo "users" >expout
       atf_check -s exit:0 -o file:expout -e empty ./run_id.sh -g -r -n

       echo "100" >expout
       atf_check -s exit:0 -o file:expout -e empty ./run_id.sh -g -r 100
       atf_check -s exit:0 -o file:expout -e empty ./run_id.sh -g -r test

       echo "users" >expout
       atf_check -s exit:0 -o file:expout -e empty ./run_id.sh -g -r -n 100
       atf_check -s exit:0 -o file:expout -e empty ./run_id.sh -g -r -n test

       unset LIBFAKE_EGID_ROOT LIBFAKE_EUID_ROOT

       echo 'id: nonexistent: No such user' >experr
       atf_check -s exit:1 -o empty -e file:experr ./run_id.sh -g nonexistent

       atf_check -s exit:1 -o empty -e save:stderr ./run_id.sh -g root nonexistent
       atf_check -s exit:0 -o ignore -e empty grep ^usage: stderr
}

atf_test_case pflag
pflag_head() {
       atf_set "descr" "Checks that the -p primary flag works"
}
pflag_body() {
       create_run_id

       cat >expout <<EOF
uid     test
groups  users wheel
EOF
       atf_check -s exit:0 -o file:expout -e empty ./run_id.sh -p
       atf_check -s exit:0 -o file:expout -e empty ./run_id.sh -p 100
       atf_check -s exit:0 -o file:expout -e empty ./run_id.sh -p test

       cat >expout <<EOF
uid     root
groups  wheel
EOF
       atf_check -s exit:0 -o file:expout -e empty ./run_id.sh -p 0
       atf_check -s exit:0 -o file:expout -e empty ./run_id.sh -p root

       export LIBFAKE_EGID_ROOT=1 LIBFAKE_EUID_ROOT=1
       cat >expout <<EOF
uid     test
euid    root
rgid    users
groups  users wheel
EOF
       atf_check -s exit:0 -o file:expout -e empty ./run_id.sh -p
       unset LIBFAKE_EGID_ROOT LIBFAKE_EUID_ROOT

       echo 'id: nonexistent: No such user' >experr
       atf_check -s exit:1 -o empty -e file:experr ./run_id.sh -p nonexistent

       atf_check -s exit:1 -o empty -e save:stderr ./run_id.sh -p root nonexistent
       atf_check -s exit:0 -o ignore -e empty grep ^usage: stderr
}

atf_test_case uflag
uflag_head() {
       atf_set "descr" "Checks that the -u primary flag works"
}
uflag_body() {
       create_run_id

       echo "100" >expout
       atf_check -s exit:0 -o file:expout -e empty ./run_id.sh -u
       atf_check -s exit:0 -o file:expout -e empty ./run_id.sh -u 100
       atf_check -s exit:0 -o file:expout -e empty ./run_id.sh -u test

       echo "test" >expout
       atf_check -s exit:0 -o file:expout -e empty ./run_id.sh -u -n
       atf_check -s exit:0 -o file:expout -e empty ./run_id.sh -u -n 100
       atf_check -s exit:0 -o file:expout -e empty ./run_id.sh -u -n test

       echo "0" >expout
       atf_check -s exit:0 -o file:expout -e empty ./run_id.sh -u 0
       atf_check -s exit:0 -o file:expout -e empty ./run_id.sh -u root

       echo "root" >expout
       atf_check -s exit:0 -o file:expout -e empty ./run_id.sh -u -n 0
       atf_check -s exit:0 -o file:expout -e empty ./run_id.sh -u -n root

       echo "100" >expout
       atf_check -s exit:0 -o file:expout -e empty ./run_id.sh -u -r

       echo "test" >expout
       atf_check -s exit:0 -o file:expout -e empty ./run_id.sh -u -r -n

       echo "100" >expout
       atf_check -s exit:0 -o file:expout -e empty ./run_id.sh -u -r 100
       atf_check -s exit:0 -o file:expout -e empty ./run_id.sh -u -r test

       echo "test" >expout
       atf_check -s exit:0 -o file:expout -e empty ./run_id.sh -u -r -n 100
       atf_check -s exit:0 -o file:expout -e empty ./run_id.sh -u -r -n test

       export LIBFAKE_EGID_ROOT=1 LIBFAKE_EUID_ROOT=1

       echo "0" >expout
       atf_check -s exit:0 -o file:expout -e empty ./run_id.sh -u

       echo "root" >expout
       atf_check -s exit:0 -o file:expout -e empty ./run_id.sh -u -n

       echo "100" >expout
       atf_check -s exit:0 -o file:expout -e empty ./run_id.sh -u -r

       echo "test" >expout
       atf_check -s exit:0 -o file:expout -e empty ./run_id.sh -u -r -n

       echo "100" >expout
       atf_check -s exit:0 -o file:expout -e empty ./run_id.sh -u -r 100
       atf_check -s exit:0 -o file:expout -e empty ./run_id.sh -u -r test

       echo "test" >expout
       atf_check -s exit:0 -o file:expout -e empty ./run_id.sh -u -r -n 100
       atf_check -s exit:0 -o file:expout -e empty ./run_id.sh -u -r -n test

       unset LIBFAKE_EGID_ROOT LIBFAKE_EUID_ROOT

       echo 'id: nonexistent: No such user' >experr
       atf_check -s exit:1 -o empty -e file:experr ./run_id.sh -u nonexistent

       atf_check -s exit:1 -o empty -e save:stderr \
           ./run_id.sh -u root nonexistent
       atf_check -s exit:0 -o ignore -e empty grep ^usage: stderr
}

atf_init_test_cases()
{
       atf_add_test_case default
       atf_add_test_case primaries
       atf_add_test_case Gflag
       atf_add_test_case gflag
       atf_add_test_case pflag
       atf_add_test_case uflag
}