test-e2e: Add check for validating kbs image and kbs image tag

Validate kbs image and kbs image tag only when deploy kbs is enabled. Signed-off-by: Kartik Joshi <kartikjoshi@microsoft.com>
confidential-containers · Mar 4, 2024 · 789ec85 · 789ec85
1 parent 5fb75bc
commit 789ec85
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/test/e2e/main_test.go b/test/e2e/main_test.go
@@ -118,9 +118,12 @@ func TestMain(m *testing.M) {
 		var err error
 
 		// Get properties
-		props := provisioner.GetProperties(ctx, cfg)
-		if props["KBS_IMAGE"] == "" || props["KBS_IMAGE_TAG"] == "" {
-			return ctx, fmt.Errorf("kbs image not provided")
+
+		if shouldDeployKbs {
+			props := provisioner.GetProperties(ctx, cfg)
+			if props["KBS_IMAGE"] == "" || props["KBS_IMAGE_TAG"] == "" {
+				return ctx, fmt.Errorf("kbs image not provided")
+			}
 		}
 
 		if shouldProvisionCluster {