From 6ca2cd8c79a1988eae464de1f4b465fc779372e9 Mon Sep 17 00:00:00 2001
From: Esteban Ricardo Mirizio <ermirizio@gmail.com>
Date: Wed, 28 Feb 2024 11:34:02 -0300
Subject: [PATCH] adding default use case

---
 cmd/policy.go             |  6 ++++--
 policy/controller_test.go | 38 ++++++++++++++++++++++++++++++++++++--
 2 files changed, 40 insertions(+), 4 deletions(-)

diff --git a/cmd/policy.go b/cmd/policy.go
index 4be2b1d7..d7fe1214 100644
--- a/cmd/policy.go
+++ b/cmd/policy.go
@@ -56,8 +56,10 @@ var controllerSubCmd = &cobra.Command{
 		// db is the databases to use
 		// pump/sink are included only when needed
 		defaults := policy.TestVariations{
-			"pump": []string{"tykio/tyk-pump-docker-pub:v1.8", "$ECR/tyk-pump:master"},
-			"sink": []string{"tykio/tyk-mdcb-docker:v2.4", "$ECR/tyk-sink:master"},
+			params["job"] + "_conf": []string{"sha256"},
+			params["job"] + "_db":   []string{"mongo44", "postgres15"},
+			"pump":                  []string{"tykio/tyk-pump-docker-pub:v1.8", "$ECR/tyk-pump:master"},
+			"sink":                  []string{"tykio/tyk-mdcb-docker:v2.4", "$ECR/tyk-sink:master"},
 		}
 		if err := params.SetVariations(&op, defaults); err != nil {
 			return err
diff --git a/policy/controller_test.go b/policy/controller_test.go
index 78e0f6ca..daa8d473 100644
--- a/policy/controller_test.go
+++ b/policy/controller_test.go
@@ -94,6 +94,38 @@ EOF
 			baseRef: "refs/heads/main",
 		},
 		{
+			job: "ui",
+			want: `versions<<EOF
+tyk_image=$ECR/tyk:master
+tyk_analytics_image=$ECR/tyk-analytics:master
+tyk_pump_image=$ECR/tyk-pump:master
+tyk_sink_image=$ECR/tyk-sink:master
+# override default above with just built tag
+tyk_image=v1.0
+# alfa and beta have to come after the override
+tyk_alfa_image=$tyk_image
+tyk_beta_image=$tyk_image
+EOF
+gd_tag=master
+pump<<EOF
+["tykio/tyk-pump-docker-pub:v1.8","$ECR/tyk-pump:master"]
+EOF
+sink<<EOF
+["tykio/tyk-mdcb-docker:v2.4","$ECR/tyk-sink:master"]
+EOF
+ui_conf<<EOF
+["sha256"]
+EOF
+ui_db<<EOF
+["mongo44","postgres15"]
+EOF
+`,
+			trigger: "",
+			isPR:    "no",
+			isTag:   "no",
+			isLTS:   "no",
+			baseRef: "refs/heads/main",
+		}, {
 			job: "api",
 			want: `versions<<EOF
 tyk_image=$ECR/tyk:release-5-lts
@@ -148,8 +180,10 @@ EOF
 			op.WriteString("\n")
 
 			defaults := TestVariations{
-				"pump": []string{"tykio/tyk-pump-docker-pub:v1.8", "$ECR/tyk-pump:master"},
-				"sink": []string{"tykio/tyk-mdcb-docker:v2.4", "$ECR/tyk-sink:master"},
+				p["job"] + "_conf": []string{"sha256"},
+				p["job"] + "_db":   []string{"mongo44", "postgres15"},
+				"pump":             []string{"tykio/tyk-pump-docker-pub:v1.8", "$ECR/tyk-pump:master"},
+				"sink":             []string{"tykio/tyk-mdcb-docker:v2.4", "$ECR/tyk-sink:master"},
 			}
 
 			if err := p.SetVariations(&op, defaults); err != nil {