From 943ee93346bf8f626e5453f833f4c87c982d9114 Mon Sep 17 00:00:00 2001
From: Jakub Warczarek <jakub.warczarek@konghq.com>
Date: Tue, 22 Oct 2024 21:52:06 +0200
Subject: [PATCH] avoid kong:3.6

---
 test/e2e/test_helm_install_upgrade.go | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/test/e2e/test_helm_install_upgrade.go b/test/e2e/test_helm_install_upgrade.go
index beb473502..cd8ca84f0 100644
--- a/test/e2e/test_helm_install_upgrade.go
+++ b/test/e2e/test_helm_install_upgrade.go
@@ -33,10 +33,10 @@ func init() {
 }
 
 func TestHelmUpgrade(t *testing.T) {
+	var image = "docker.io/kong/gateway-operator-oss"
 	const (
 		// Rel: https://github.com/Kong/charts/tree/main/charts/gateway-operator
 		chart = "kong/gateway-operator"
-		image = "docker.io/kong/gateway-operator-oss"
 
 		waitTime = 3 * time.Minute
 	)
@@ -376,6 +376,9 @@ func TestHelmUpgrade(t *testing.T) {
 
 	for _, tc := range testCases {
 		t.Run(tc.name, func(t *testing.T) {
+			if helpers.GetDefaultDataPlaneBaseImage() == consts.DefaultDataPlaneBaseEnterpriseImage {
+				image = "docker.io/kong/gateway-operator"
+			}
 			var (
 				tag              string
 				targetRepository = image