From 22c589e2cf84cee9a99c7e0829e6a7e4445780ea Mon Sep 17 00:00:00 2001
From: Daniel Hougaard <62331820+DanielHougaard@users.noreply.github.com>
Date: Thu, 18 Apr 2024 15:01:31 +0200
Subject: [PATCH] Update tests.go

---
 cli/packages/util/tests.go | 32 ++++++++++++++++++++++++++++----
 1 file changed, 28 insertions(+), 4 deletions(-)

diff --git a/cli/packages/util/tests.go b/cli/packages/util/tests.go
index 816df573e..989376e58 100644
--- a/cli/packages/util/tests.go
+++ b/cli/packages/util/tests.go
@@ -9,10 +9,11 @@ import (
 	"github.com/spf13/cobra"
 )
 
-func HandleSendTestSecrets(cmd *cobra.Command, secrets []models.SingleEnvironmentVariable) {
-	isTestMode := os.Getenv("TEST_MODE")
+var IS_TEST_MODE = os.Getenv("TEST_MODE") == "true"
 
-	if isTestMode != "true" {
+func HandleSendTestSecrets(cmd *cobra.Command, secrets []models.SingleEnvironmentVariable) {
+
+	if !IS_TEST_MODE {
 		return
 	}
 
@@ -22,5 +23,28 @@ func HandleSendTestSecrets(cmd *cobra.Command, secrets []models.SingleEnvironmen
 	}
 
 	fmt.Fprint(cmd.OutOrStdout(), string(jsonOut))
-
+}
+
+func HandleSendTestEnvVars(cmd *cobra.Command, envs []string) {
+
+	if !IS_TEST_MODE {
+		return
+	}
+
+	stringEnvVars := ""
+
+	for _, env := range envs {
+		stringEnvVars += env + "\n"
+	}
+
+	fmt.Fprint(cmd.OutOrStdout(), string(stringEnvVars))
+}
+
+func HandleSendUniversalAuthToken(cmd *cobra.Command, token string) {
+
+	if !IS_TEST_MODE {
+		return
+	}
+
+	fmt.Fprint(cmd.OutOrStdout(), token)
 }