Merge pull request #1166 from nxf5025/feature/private-helm-repo-add

Allow for ExtraArgs in Helm Module for AddRepo

modules/helm/options.go

@@ -15,5 +15,5 @@ type Options struct {
 	EnvVars        map[string]string   // Environment variables to set when running helm
 	Version        string              // Version of chart
 	Logger         *logger.Logger      // Set a non-default logger that should be used. See the logger package for more info. Use logger.Discard to not print the output while executing the command.
-	ExtraArgs      map[string][]string // Extra arguments to pass to the helm install/upgrade/rollback/delete command. The key signals the command (e.g., install) while the values are the extra arguments to pass through.
+	ExtraArgs      map[string][]string // Extra arguments to pass to the helm install/upgrade/rollback/delete and helm repo add commands. The key signals the command (e.g., install) while the values are the extra arguments to pass through.
 }

modules/helm/repo.go

@@ -14,7 +14,16 @@ func AddRepo(t testing.TestingT, options *Options, repoName string, repoURL stri
 
 // AddRepoE will setup the provided helm repository to the local helm client configuration.
 func AddRepoE(t testing.TestingT, options *Options, repoName string, repoURL string) error {
-	_, err := RunHelmCommandAndGetOutputE(t, options, "repo", "add", repoName, repoURL)
+	// Set required args
+	args := []string{"add", repoName, repoURL}
+
+	// Append helm repo add ExtraArgs if available
+	if options.ExtraArgs != nil {
+		if repoAddArgs, ok := options.ExtraArgs["repoAdd"]; ok {
+			args = append(args, repoAddArgs...)
+		}
+	}
+	_, err := RunHelmCommandAndGetOutputE(t, options, "repo", args...)
 	return err
 }