Signed-off-by: Andrew Ammerlaan <andrewammerl...@gentoo.org>
---
 eclass/tests/dist-kernel-utils.sh | 59 +++++++++++++++++++++++++++++++
 1 file changed, 59 insertions(+)

diff --git a/eclass/tests/dist-kernel-utils.sh 
b/eclass/tests/dist-kernel-utils.sh
index 50ba001f8e9c..28c8f7213a53 100755
--- a/eclass/tests/dist-kernel-utils.sh
+++ b/eclass/tests/dist-kernel-utils.sh
@@ -20,9 +20,68 @@ test_PV_to_KV() {
        tend $?
 }
 
+test_compressed_module_cleanup() {
+       mkdir -p "${tmpdir}/source" || die
+       pushd "${tmpdir}" >/dev/null || die
+
+       local module option fail=0
+       for option in NONE GZIP XZ ZSTD; do
+               tbegin "CONFIG_MODULE_COMPRESS_${option}"
+               echo "CONFIG_MODULE_COMPRESS_${option}=y" > source/.config
+
+               touch a.ko b.ko.gz c.ko.xz d.ko.gz e.ko f.ko.xz || die
+               # ensure some files are older
+               touch -d "2 hours ago" d.ko e.ko.xz f.ko.gz || die
+
+               IUSE=modules-compress dist-kernel_compressed_module_cleanup .
+
+               local to_keep=( a.ko b.ko.gz c.ko.xz )
+               local to_remove=()
+
+               case ${option} in
+                       NONE)
+                               to_keep+=( d.ko e.ko f.ko.xz )
+                               to_remove+=( d.ko.gz e.ko.xz f.ko.gz )
+                               ;;
+                       GZIP)
+                               to_keep+=( d.ko.gz e.ko f.ko.gz )
+                               to_remove+=( d.ko e.ko.xz f.ko.xz )
+                               ;;
+                       XZ)
+                               to_keep+=( d.ko.gz e.ko.xz f.ko.xz )
+                               to_remove+=( d.ko e.ko f.ko.gz )
+                               ;;
+                       ZSTD)
+                               to_keep+=( d.ko.gz e.ko f.ko.xz )
+                               to_remove+=( d.ko e.ko.xz f.ko.gz )
+                               ;;
+               esac
+
+               for module in "${to_keep[@]}"; do
+                       if [[ ! -f ${module} ]]; then
+                               eerror "Module ${module} was removed"
+                               fail=1
+                       fi
+               done
+
+               for module in "${to_remove[@]}"; do
+                       if [[ -f ${module} ]]; then
+                               eerror "Module ${module} was not removed"
+                               fail=1
+                       fi
+               done
+               tend ${fail}
+       done
+
+       popd >/dev/null || die
+}
+
+
 test_PV_to_KV 6.0_rc1 6.0.0-rc1
 test_PV_to_KV 6.0 6.0.0
 test_PV_to_KV 6.0.1_rc1 6.0.1-rc1
 test_PV_to_KV 6.0.1 6.0.1
 
+test_compressed_module_cleanup
+
 texit
-- 
2.45.2


Reply via email to