From 8a86628cbf8bda095486ef405b1dbabbb995105a Mon Sep 17 00:00:00 2001
From: Yariv Rachmani <yrachman@redhat.com>
Date: Wed, 20 Dec 2023 09:01:08 +0200
Subject: [PATCH] Add a fix in setup

Signed-off-by: Yariv Rachmani <yrachman@redhat.com>
---
 plans/e2e/ffi.fmf       |  5 ++++-
 plans/e2e/tier-0.fmf    |  8 +++++---
 setup                   | 32 +++++++++++++++++++++++++++-----
 tests/e2e/lib/container |  2 +-
 4 files changed, 37 insertions(+), 10 deletions(-)

diff --git a/plans/e2e/ffi.fmf b/plans/e2e/ffi.fmf
index 6cff6b9f..0df948cb 100644
--- a/plans/e2e/ffi.fmf
+++ b/plans/e2e/ffi.fmf
@@ -11,6 +11,10 @@ provision:
    how: local
 
 adjust:
+ - when: run == manual
+   environment+:
+       USE_QM_COPR: yes
+ - when: scenario == ffi and distro == centos-stream-9, fedora
    prepare+:
       - name: Install rpms
         how: install
@@ -23,7 +27,6 @@ adjust:
         script: |
            cd tests/e2e
            ./set-ffi-env-e2e --set-qm-disk-part=yes
-   when: scenario == ffi and distro == centos-stream-9, fedora
 
 execute:
     how: tmt
diff --git a/plans/e2e/tier-0.fmf b/plans/e2e/tier-0.fmf
index 8792dd2d..d94d43d4 100644
--- a/plans/e2e/tier-0.fmf
+++ b/plans/e2e/tier-0.fmf
@@ -8,9 +8,11 @@ provision:
    how: local
 
 adjust:
- - when: scenario == manual
+ - when: run == manual
+   environment+:
+       USE_QM_COPR: yes
    prepare+:
-     - name: Prepare Repos
+     - name: Prepare QM Repos
        how: shell
        order: 10
        script: |
@@ -33,7 +35,7 @@ adjust:
        how: shell
        script: |
           cd tests/e2e
-          ./run-test-e2e --skip-tests=yes
+          ./run-test-e2e --use-qm-repo --skip-tests=yes
 
 execute:
     how: tmt
diff --git a/setup b/setup
index 6ab55cc0..4f2380f7 100755
--- a/setup
+++ b/setup
@@ -11,7 +11,7 @@ INSTALLDIR="/usr/share/qm"
 ROOTFS="/usr/lib/qm/rootfs"
 RWETCFS="/etc/qm"
 RWVARFS="/var/qm"
-TMP_QM_IMG_DIR="tmp-img-dir"
+TMP_QM_IMG_DIR="tmp.images"
 AGENT_HOSTNAME="$(hostname)"
 AGENTCONF="/etc/bluechi/agent.conf.d/agent.conf"
 QM_CONTAINER_IDS=1000000000:1500000000
@@ -98,12 +98,34 @@ storage() {
 	touch "${ROOTFS}/var/lib/shared/overlay-images/images.lock" \
 	      "${ROOTFS}/var/lib/shared/overlay-layers/layers.lock"
 
-	sed -e '/additionalimage.*/a "/var/lib/shared",' \
-	    -e 's|^#.*transient_store.*|transient_store=true|g' \
-	    "${ROOTFS}/usr/share/containers/storage.conf" \
-	    > "${ROOTFS}/etc/containers/storage.conf"
+        sed -e '/additionalimage.*/a "/var/lib/shared",' \
+            -e 's|^#.*transient_store.*|transient_store=true|g' \
+            "${ROOTFS}/usr/share/containers/storage.conf" \
+            > "${ROOTFS}/etc/containers/storage.conf"
         sed -i "s|^#env = \[\]|env = \[\"TMPDIR=/var/${TMP_QM_IMG_DIR}\"\]|" \
             "${ROOTFS}/usr/share/containers/containers.conf"
+
+
+        sed -e '/^\s*#/d' -e  '/^$/d' \
+            "${ROOTFS}/usr/share/containers/storage.conf" \
+            > "${ROOTFS}/etc/containers/storage.conf"
+        if ! grep -q "additionalimagestores" "${ROOTFS}/etc/containers/storage.conf"; then
+            cat >> "${ROOTFS}/etc/containers/storage.conf" << EOF
+additionalimagestores = [
+/var/lib/shared,
+]
+EOF
+        else
+           sed -i '/additionalimage.*/a "/var/lib/shared",'
+           "${ROOTFS}/etc/containers/storage.conf"
+        fi
+        if ! grep -q "transient_store" "${ROOTFS}/etc/containers/storage.conf"; then
+           echo transient_store=true >> "${ROOTFS}/etc/containers/storage.conf"
+        fi
+        if ! grep -q "env" "${ROOTFS}/etc/containers/containers.conf"; then
+            echo "env = \[\"TMPDIR=/var/${TMP_QM_IMG_DIR}\"\]" \
+            >>  "${ROOTFS}/etc/containers/containers.conf"
+        fi
     fi
 }
 
diff --git a/tests/e2e/lib/container b/tests/e2e/lib/container
index 93f66d98..a95294f5 100644
--- a/tests/e2e/lib/container
+++ b/tests/e2e/lib/container
@@ -80,7 +80,7 @@ setup_node() {
         # copy control as template to node${number}
         cp ./lib/ContainerFile.template ContainerFile.node"${nodeID}"
         if_error_exit "failed to copy ContainerFile control template to node!"
-        if [ -n ${use_qm-copr} ]; then
+        if [ -n ${use_qm_copr} ]; then
           sed -e 's/crb/&\nRUN dnf -y copr enable rhcontainerbot\/qm centos-stream-9/' -i ContainerFile.node"${nodeID}"
           if_error_exit "failed to sed ContainerFile node template!"
         fi