feat: added embedded nats server option to CLI (#5566)

kubeshop · Jun 11, 2024 · 6c4d1cf · 6c4d1cf
1 parent 1b44ba3
commit 6c4d1cf
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/cmd/kubectl-testkube/commands/common/helper.go b/cmd/kubectl-testkube/commands/common/helper.go
@@ -30,6 +30,7 @@ type HelmOptions struct {
 	DryRun         bool
 	MultiNamespace bool
 	NoOperator     bool
+	EmbeddedNATS   bool
 }
 
 const (
@@ -117,6 +118,12 @@ func HelmUpgradeOrInstallTestkubeCloud(options HelmOptions, cfg config.Data, isM
 	args = append(args, "--set", fmt.Sprintf("testkube-api.minio.replicas=%d", options.MinioReplicas))
 	args = append(args, "--set", fmt.Sprintf("mongodb.replicas=%d", options.MongoReplicas))
 
+	// if embedded nats is enabled disable nats chart
+	if options.EmbeddedNATS {
+		args = append(args, "--set", "testkube-api.nats.enabled=false")
+		args = append(args, "--set", "testkube-api.nats.embedded=true")
+	}
+
 	args = append(args, options.Name, options.Chart)
 
 	if options.Values != "" {
@@ -190,6 +197,7 @@ func PopulateHelmFlags(cmd *cobra.Command, options *HelmOptions) {
 	cmd.Flags().BoolVar(&options.NoMongo, "no-mongo", false, "don't install MongoDB")
 	cmd.Flags().BoolVar(&options.NoConfirm, "no-confirm", false, "don't ask for confirmation - unatended installation mode")
 	cmd.Flags().BoolVar(&options.DryRun, "dry-run", false, "dry run mode - only print commands that would be executed")
+	cmd.Flags().BoolVar(&options.EmbeddedNATS, "embedded-nats", false, "embedded NATS server in agent")
 }
 
 func PopulateLoginDataToContext(orgID, envID, token, refreshToken string, options HelmOptions, cfg config.Data) error {