Feat: support newly added template types - pulumi, k8s & workflow (#428)

* Feat: support newly added template types - pulumi, k8s & workflow * fixed unit test * modify comple to workflow
env0 · Jun 26, 2022 · a3036c8 · a3036c8
1 parent 8027c94
commit a3036c8
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 4 deletions.
diff --git a/env0/data_template.go b/env0/data_template.go
@@ -2,6 +2,8 @@ package env0
 
 import (
 	"context"
+	"fmt"
+	"strings"
 
 	"github.com/env0/terraform-provider-env0/client"
 	"github.com/hashicorp/terraform-plugin-sdk/v2/diag"
@@ -42,7 +44,7 @@ func dataTemplate() *schema.Resource {
 			},
 			"type": {
 				Type:        schema.TypeString,
-				Description: "'terraform' or 'terragrunt'",
+				Description: fmt.Sprintf("template type (allowed values: %s)", strings.Join(allowedTemplateTypes, ", ")),
 				Computed:    true,
 			},
 			"project_ids": {

diff --git a/env0/resource_template_test.go b/env0/resource_template_test.go
@@ -502,7 +502,7 @@ func TestUnitTemplateResource(t *testing.T) {
 						"repository": template.Repository,
 						"type":       "gruntyform",
 					}),
-					ExpectError: regexp.MustCompile(`must be one of: terragrunt, terraform`),
+					ExpectError: regexp.MustCompile(`must be one of`),
 				},
 			},
 		}

diff --git a/env0/templates.go b/env0/templates.go
@@ -1,7 +1,9 @@
 package env0
 
 import (
+	"fmt"
 	"sort"
+	"strings"
 
 	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
 )
@@ -13,6 +15,14 @@ const (
 	TemplateTypeShared = 2
 )
 
+var allowedTemplateTypes = []string{
+	"terraform",
+	"terragrunt",
+	"pulumi",
+	"k8s",
+	"workflow",
+}
+
 func getTemplateSchema(templateType TemplateType) map[string]*schema.Schema {
 	/*
 	 *	VCS Constraints:
@@ -78,10 +88,10 @@ func getTemplateSchema(templateType TemplateType) map[string]*schema.Schema {
 		},
 		"type": {
 			Type:             schema.TypeString,
-			Description:      "'terraform' or 'terragrunt'",
+			Description:      fmt.Sprintf("template type (allowed values: %s)", strings.Join(allowedTemplateTypes, ", ")),
 			Optional:         true,
 			Default:          "terraform",
-			ValidateDiagFunc: NewStringInValidator([]string{"terragrunt", "terraform"}),
+			ValidateDiagFunc: NewStringInValidator(allowedTemplateTypes),
 		},
 		"ssh_keys": {
 			Type:        schema.TypeList,