test: make the root/data partition size configurable per test

(cherry picked from commit 0334afe)

Author: mrc0mmand

test/TEST-58-REPART/test.sh

@@ -4,6 +4,7 @@ set -e
 
 TEST_DESCRIPTION="test systemd-repart"
 IMAGE_NAME="repart"
+IMAGE_ADDITIONAL_ROOT_SIZE=1000
 TEST_FORCE_NEWIMAGE=1
 
 # shellcheck source=test/test-functions

test/test-functions

@@ -1445,8 +1445,12 @@ create_empty_image() {
         root_size=$((4 * root_size))
         data_size=$((2 * data_size))
     fi
-    if [ "$IMAGE_NAME" = "repart" ]; then
-        root_size=$((root_size+=1000))
+
+    if [[ "${IMAGE_ADDITIONAL_ROOT_SIZE:-0}" -gt 0 ]]; then
+        root_size=$((root_size + IMAGE_ADDITIONAL_ROOT_SIZE))
+    fi
+    if [[ "${IMAGE_ADDITIONAL_DATA_SIZE:-0}" -gt 0 ]]; then
+        data_size=$((data_size + IMAGE_ADDITIONAL_DATA_SIZE))
     fi
 
     echo "Setting up ${IMAGE_PUBLIC:?} (${root_size} MB)"