From 6731f050dad99d50da2d65db917fc7e2fb9f75e9 Mon Sep 17 00:00:00 2001
From: Evan Lezar <elezar@nvidia.com>
Date: Tue, 16 Nov 2021 15:00:58 +0100
Subject: [PATCH] Rework init repo for centos8 release tests

Signed-off-by: Evan Lezar <elezar@nvidia.com>
---
 test/release/Makefile                       |  2 --
 test/release/docker/centos8/Dockerfile      |  3 +--
 test/release/docker/centos8/entrypoint.sh   |  6 ++---
 test/release/docker/centos8/install_repo.sh | 25 +++++++++++++++++++++
 4 files changed, 29 insertions(+), 7 deletions(-)
 create mode 100755 test/release/docker/centos8/install_repo.sh

diff --git a/test/release/Makefile b/test/release/Makefile
index 7c58c518..c680e994 100644
--- a/test/release/Makefile
+++ b/test/release/Makefile
@@ -13,7 +13,6 @@
 # limitations under the License.
 
 WORKFLOW ?= nvidia-docker
-TEST_REPO ?= elezar.github.io
 
 DISTRIBUTIONS := ubuntu18.04 centos8
 
@@ -30,7 +29,6 @@ images: $(IMAGE_TARGETS)
 $(IMAGE_TARGETS): image-%:
 	docker build \
     --build-arg WORKFLOW="$(WORKFLOW)" \
-    --build-arg TEST_REPO="$(TEST_REPO)" \
     -t nvidia-container-toolkit-repo-test:$(*) \
     -f $(DOCKERFILE) \
     $(shell dirname $(DOCKERFILE))
diff --git a/test/release/docker/centos8/Dockerfile b/test/release/docker/centos8/Dockerfile
index e0947bbe..b6015b18 100644
--- a/test/release/docker/centos8/Dockerfile
+++ b/test/release/docker/centos8/Dockerfile
@@ -25,11 +25,10 @@ RUN fpm -s empty \
 
 
 ARG WORKFLOW=nvidia-docker
-ARG TEST_REPO=nvidia.github.io
-ENV TEST_REPO ${TEST_REPO}
 RUN curl -s -L https://nvidia.github.io/${WORKFLOW}/centos8/nvidia-docker.repo \
     | tee /etc/yum.repos.d/nvidia-docker.repo
 
 COPY entrypoint.sh /
+COPY install_repo.sh /
 
 ENTRYPOINT [ "/entrypoint.sh" ]
\ No newline at end of file
diff --git a/test/release/docker/centos8/entrypoint.sh b/test/release/docker/centos8/entrypoint.sh
index 9dd4de0e..c118b1fe 100755
--- a/test/release/docker/centos8/entrypoint.sh
+++ b/test/release/docker/centos8/entrypoint.sh
@@ -33,10 +33,10 @@ gpgcheck=0
 protect=1
 EOL
     yum-config-manager --enable local-repository
+elif [[ -n ${TEST_REPO} ]]; then
+    ./install_repo.sh ${TEST_REPO}
 else
-    echo "Setting up TEST repo: ${TEST_REPO}"
-    sed -i -e 's#nvidia\.github\.io/libnvidia-container#${TEST_REPO}/libnvidia-container#g' /etc/yum.repos.d/nvidia-docker.repo
-    yum-config-manager --enable libnvidia-container-experimental
+    echo "Skipping repo setup"
 fi
 
 exec bash $@
diff --git a/test/release/docker/centos8/install_repo.sh b/test/release/docker/centos8/install_repo.sh
new file mode 100755
index 00000000..ae65fdf3
--- /dev/null
+++ b/test/release/docker/centos8/install_repo.sh
@@ -0,0 +1,25 @@
+#!/usr/bin/env bash
+
+# Copyright (c) 2021, NVIDIA CORPORATION.  All rights reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# This script is used to build the packages for the components of the NVIDIA
+# Container Stack. These include the nvidia-container-toolkit in this repository
+# as well as the components included in the third_party folder.
+# All required packages are generated in the specified dist folder.
+
+test_repo=$1
+echo "Setting up TEST repo: ${test_repo}"
+sed -i -e "s#nvidia\.github\.io/libnvidia-container#${test_repo}/libnvidia-container#g" /etc/yum.repos.d/nvidia-docker.repo
+yum-config-manager --enable libnvidia-container-experimental