From 0334afe4d01b0f24d06995bb3ee4f5c2ff7d58c6 Mon Sep 17 00:00:00 2001
From: Frantisek Sumsal <frantisek@sumsal.cz>
Date: Mon, 5 Jun 2023 09:39:51 +0200
Subject: [PATCH] test: make the root/data partition size configurable per test

---
 test/TEST-58-REPART/test.sh | 1 +
 test/test-functions         | 7 +++++--
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/test/TEST-58-REPART/test.sh b/test/TEST-58-REPART/test.sh
index 7549de568c2..d6e6ac0b226 100755
--- a/test/TEST-58-REPART/test.sh
+++ b/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
diff --git a/test/test-functions b/test/test-functions
index 09e5e0d38ef..5aa99522fc7 100644
--- a/test/test-functions
+++ b/test/test-functions
@@ -1442,8 +1442,11 @@ create_empty_image() {
         fi
     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)"
-- 
2.47.3