Fix: set workspace in environment create (#189)

* Create Environment resource with custom workspace name * UT
env0 · Dec 20, 2021 · d36c056 · d36c056
1 parent a7ead19
commit d36c056
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 11 deletions.
diff --git a/env0/resource_environment.go b/env0/resource_environment.go
@@ -4,12 +4,13 @@ import (
 	"context"
 	"errors"
 	"fmt"
+	"log"
+	"regexp"
+
 	"github.com/env0/terraform-provider-env0/client"
 	"github.com/google/uuid"
 	"github.com/hashicorp/terraform-plugin-sdk/v2/diag"
 	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
-	"log"
-	"regexp"
 )
 
 func resourceEnvironment() *schema.Resource {
@@ -346,6 +347,10 @@ func getCreatePayload(d *schema.ResourceData, apiClient client.ApiClientInterfac
 		payload.ProjectId = projectId.(string)
 	}
 
+	if workspace, ok := d.GetOk("workspace"); ok {
+		payload.WorkspaceName = workspace.(string)
+	}
+
 	if d.HasChange("deploy_on_push") {
 		continuousDeployment := d.Get("deploy_on_push").(bool)
 		payload.ContinuousDeployment = &continuousDeployment

diff --git a/env0/resource_environment_test.go b/env0/resource_environment_test.go
@@ -3,12 +3,13 @@ package env0
 import (
 	"errors"
 	"fmt"
-	"github.com/env0/terraform-provider-env0/client"
-	"github.com/golang/mock/gomock"
-	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/resource"
 	"regexp"
 	"strings"
 	"testing"
+
+	"github.com/env0/terraform-provider-env0/client"
+	"github.com/golang/mock/gomock"
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/resource"
 )
 
 func TestUnitEnvironmentResource(t *testing.T) {
@@ -18,18 +19,20 @@ func TestUnitEnvironmentResource(t *testing.T) {
 	templateId := "template-id"
 
 	environment := client.Environment{
-		Id:        "id0",
-		Name:      "my-environment",
-		ProjectId: "project-id",
+		Id:            "id0",
+		Name:          "my-environment",
+		ProjectId:     "project-id",
+		WorkspaceName: "workspace-name",
 		LatestDeploymentLog: client.DeploymentLog{
 			BlueprintId: templateId,
 		},
 	}
 
 	updatedEnvironment := client.Environment{
-		Id:        environment.Id,
-		Name:      "my-updated-environment-name",
-		ProjectId: "project-id",
+		Id:            environment.Id,
+		Name:          "my-updated-environment-name",
+		ProjectId:     "project-id",
+		WorkspaceName: environment.WorkspaceName,
 		LatestDeploymentLog: client.DeploymentLog{
 			BlueprintId: templateId,
 		},
@@ -40,6 +43,7 @@ func TestUnitEnvironmentResource(t *testing.T) {
 			"name":          environment.Name,
 			"project_id":    environment.ProjectId,
 			"template_id":   environment.LatestDeploymentLog.BlueprintId,
+			"workspace":     environment.WorkspaceName,
 			"force_destroy": true,
 		})
 	}
@@ -55,6 +59,7 @@ func TestUnitEnvironmentResource(t *testing.T) {
 						resource.TestCheckResourceAttr(accessor, "name", environment.Name),
 						resource.TestCheckResourceAttr(accessor, "project_id", environment.ProjectId),
 						resource.TestCheckResourceAttr(accessor, "template_id", templateId),
+						resource.TestCheckResourceAttr(accessor, "workspace", environment.WorkspaceName),
 					),
 				},
 				{
@@ -64,6 +69,7 @@ func TestUnitEnvironmentResource(t *testing.T) {
 						resource.TestCheckResourceAttr(accessor, "name", updatedEnvironment.Name),
 						resource.TestCheckResourceAttr(accessor, "project_id", updatedEnvironment.ProjectId),
 						resource.TestCheckResourceAttr(accessor, "template_id", templateId),
+						resource.TestCheckResourceAttr(accessor, "workspace", environment.WorkspaceName),
 					),
 				},
 			},
@@ -73,6 +79,7 @@ func TestUnitEnvironmentResource(t *testing.T) {
 			mock.EXPECT().EnvironmentCreate(client.EnvironmentCreate{
 				Name:                        environment.Name,
 				ProjectId:                   environment.ProjectId,
+				WorkspaceName:               environment.WorkspaceName,
 				AutoDeployOnPathChangesOnly: &autoDeployOnPathChangesOnlyDefault,
 				AutoDeployByCustomGlob:      autoDeployByCustomGlobDefault,
 				DeployRequest: &client.DeployRequest{
@@ -219,6 +226,7 @@ func TestUnitEnvironmentResource(t *testing.T) {
 			mock.EXPECT().EnvironmentCreate(client.EnvironmentCreate{
 				Name:                        environment.Name,
 				ProjectId:                   environment.ProjectId,
+				WorkspaceName:               environment.WorkspaceName,
 				AutoDeployOnPathChangesOnly: &autoDeployOnPathChangesOnlyDefault,
 				AutoDeployByCustomGlob:      autoDeployByCustomGlobDefault,
 				DeployRequest: &client.DeployRequest{
@@ -680,6 +688,7 @@ func TestUnitEnvironmentResource(t *testing.T) {
 			mock.EXPECT().EnvironmentCreate(client.EnvironmentCreate{
 				Name:                        environment.Name,
 				ProjectId:                   environment.ProjectId,
+				WorkspaceName:               environment.WorkspaceName,
 				AutoDeployOnPathChangesOnly: &autoDeployOnPathChangesOnlyDefault,
 				AutoDeployByCustomGlob:      autoDeployByCustomGlobDefault,
 				DeployRequest: &client.DeployRequest{
@@ -707,6 +716,7 @@ func TestUnitEnvironmentResource(t *testing.T) {
 			mock.EXPECT().EnvironmentCreate(client.EnvironmentCreate{
 				Name:                        environment.Name,
 				ProjectId:                   environment.ProjectId,
+				WorkspaceName:               environment.WorkspaceName,
 				AutoDeployOnPathChangesOnly: &autoDeployOnPathChangesOnlyDefault,
 				AutoDeployByCustomGlob:      autoDeployByCustomGlobDefault,
 				DeployRequest: &client.DeployRequest{
@@ -781,6 +791,7 @@ func TestUnitEnvironmentResource(t *testing.T) {
 			mock.EXPECT().EnvironmentCreate(client.EnvironmentCreate{
 				Name:                        environment.Name,
 				ProjectId:                   environment.ProjectId,
+				WorkspaceName:               environment.WorkspaceName,
 				AutoDeployOnPathChangesOnly: &autoDeployOnPathChangesOnlyDefault,
 				AutoDeployByCustomGlob:      autoDeployByCustomGlobDefault,
 				DeployRequest: &client.DeployRequest{