#       $NetBSD: t_sets.sh,v 1.9 2024/05/10 03:29:47 riastradh Exp $
#
# Copyright (c) 2024 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.
#

check_mtree()
{
       local set=$1

       cd /
       atf_check -o empty -s exit:0 \
               mtree -e </etc/mtree/set."$set"
}

set_case()
{
       local set=$1

       eval "set_${set}_head() { atf_set descr \"/etc/mtree/set.${set}\"; }"
       eval "set_${set}_body() { check_mtree ${set}; }"
       eval "set_${set}_defined="
}

set_case base
set_case base32
set_case base64
set_case comp
set_case debug
set_case debug32
set_case debug64
set_case dtb
#set_case etc
set_case games
set_case gpufw
set_case man
set_case manhtml
set_case misc
set_case modules
set_case rescue
set_case tests
set_case text
set_case xbase
set_case xcomp
set_case xdebug
#set_case xetc
set_case xfont
set_case xserver

sets_unknown=

sets_unknown_head()
{
       atf_set descr "Verify this tests lists all sets"
}
sets_unknown_body()
{
       test -z "$sets_unknown" || atf_fail "Unknown sets: ${sets_unknown}"
}

atf_init_test_cases()
{
       local mtree set defined

       atf_add_test_case sets_unknown

       # base is always installed -- hard-code this in case we make a
       # mistake with the automatic set detection.
       atf_add_test_case set_base

       # Test all of the sets that are installed, except for some
       # special cases.
       for mtree in /etc/mtree/set.*; do
               set=${mtree#/etc/mtree/set.}
               case $set in
               base)   # Handled above already.
                       continue
                       ;;
               dtb)
                       # contents of this set go to the boot partition,
                       # which may not be mounted during normal operation
                       if [ ! -d /boot/dtb ]; then
                               continue;
                       fi
                       ;;
               etc|xetc)
                       # etc and xetc have files that may be modified
                       # on installation, and also contain log files,
                       # so let's skip them for now.
                       continue
                       ;;
               *)      ;;
               esac

               # If we have a test for this set, add it.  Otherwise,
               # add it to the unknown list to make the test suite
               # fail.
               eval 'defined=${set_'"$set"'_defined+yes}'
               if [ -n "$defined" ]; then
                       atf_add_test_case set_"${set}"
               else
                       sets_unknown="${sets_unknown}${sets_unknown:+ }${set}"
               fi
       done
}