agent pool support for stack resource

Author: Maed223

internal/provider/resource_tfe_stack.go

@@ -80,6 +80,10 @@ func (r *resourceTFEStack) Schema(ctx context.Context, req resource.SchemaReques
 					stringplanmodifier.RequiresReplace(),
 				},
 			},
+			"agent_pool_id": schema.StringAttribute{
+				Description: "The time when the stack was last updated.",
+				Optional:    true,
+			},
 			"name": schema.StringAttribute{
 				Description: "Name of the Stack",
 				Required:    true,
@@ -155,6 +159,12 @@ func (r *resourceTFEStack) Create(ctx context.Context, req resource.CreateReques
 		}
 	}
 
+	if !plan.AgentPoolID.IsNull() {
+		options.AgentPool = &tfe.AgentPool{
+			ID: plan.AgentPoolID.ValueString(),
+		}
+	}
+
 	if !plan.Description.IsNull() {
 		options.Description = tfe.String(plan.Description.ValueString())
 	}
@@ -251,6 +261,12 @@ func (r *resourceTFEStack) Update(ctx context.Context, req resource.UpdateReques
 		options.VCSRepo = nil
 	}
 
+	if !plan.AgentPoolID.IsNull() {
+		options.AgentPool = &tfe.AgentPool{
+			ID: plan.AgentPoolID.ValueString(),
+		}
+	}
+
 	tflog.Debug(ctx, "Updating stack")
 	stack, err := r.config.Client.Stacks.Update(ctx, state.ID.ValueString(), options)
 	if err != nil {

internal/provider/resource_tfe_stack_test.go

@@ -77,6 +77,63 @@ resource "tfe_stack" "foobar" {
 `, orgName, ghToken, ghRepoIdentifier)
 }
 
+func TestAccTFEStackResource_withAgentPool(t *testing.T) {
+	skipUnlessBeta(t)
+
+	rInt := rand.New(rand.NewSource(time.Now().UnixNano())).Int()
+	orgName := fmt.Sprintf("tst-terraform-%d", rInt)
+
+	resource.Test(t, resource.TestCase{
+		PreCheck:                 func() { testAccPreCheck(t) },
+		ProtoV5ProviderFactories: testAccMuxedProviders,
+		Steps: []resource.TestStep{
+			{
+				Config: testAccTFEStackResourceConfigWithAgentPool(orgName),
+				Check: resource.ComposeAggregateTestCheckFunc(
+					resource.TestCheckResourceAttrSet("tfe_stack.foobar", "id"),
+					resource.TestCheckResourceAttrSet("tfe_stack.foobar", "project_id"),
+					resource.TestCheckResourceAttrSet("tfe_stack.foobar", "agent_pool_id"),
+					resource.TestCheckResourceAttr("tfe_stack.foobar", "name", "example-stack"),
+					resource.TestCheckResourceAttr("tfe_stack.foobar", "description", "Just an ordinary stack"),
+					resource.TestCheckResourceAttrSet("tfe_stack.foobar", "created_at"),
+					resource.TestCheckResourceAttrSet("tfe_stack.foobar", "updated_at"),
+				),
+			},
+			{
+				ResourceName:      "tfe_stack.foobar",
+				ImportState:       true,
+				ImportStateVerify: true,
+			},
+		},
+	})
+}
+
+func testAccTFEStackResourceConfigWithAgentPool(orgName string) string {
+	return fmt.Sprintf(`
+resource "tfe_organization" "foobar" {
+  name  = "%s"
+  email = "[email protected]"
+}
+
+resource "tfe_agent_pool" "foobar" {
+  name                  = "agent-pool-test-example"
+  organization          = tfe_organization.foobar.name
+}
+
+resource "tfe_project" "example" {
+	name         = "example"
+	organization = tfe_organization.foobar.name
+}
+
+resource "tfe_stack" "foobar" {
+	name        = "example-stack"
+	description = "Just an ordinary stack"
+    project_id  = tfe_project.example.id
+    agent_pool_id = tfe_agent_pool.foobar.id
+}
+`, orgName)
+}
+
 func TestAccTFEStackResource_noVCSRepo(t *testing.T) {
 	skipUnlessBeta(t)
 

internal/provider/stack.go

@@ -23,6 +23,7 @@ type modelTFEStackVCSRepo struct {
 type modelTFEStack struct {
 	ID              types.String          `tfsdk:"id"`
 	ProjectID       types.String          `tfsdk:"project_id"`
+	AgentPoolID     types.String          `tfsdk:"agent_pool_id"`
 	Name            types.String          `tfsdk:"name"`
 	Description     types.String          `tfsdk:"description"`
 	DeploymentNames types.Set             `tfsdk:"deployment_names"`
@@ -42,6 +43,7 @@ func modelFromTFEStack(v *tfe.Stack) modelTFEStack {
 	result := modelTFEStack{
 		ID:              types.StringValue(v.ID),
 		ProjectID:       types.StringValue(v.Project.ID),
+		AgentPoolID:     types.StringNull(),
 		Name:            types.StringValue(v.Name),
 		Description:     types.StringNull(),
 		DeploymentNames: types.SetValueMust(types.StringType, names),
@@ -58,6 +60,10 @@ func modelFromTFEStack(v *tfe.Stack) modelTFEStack {
 		}
 	}
 
+	if v.AgentPool != nil {
+		result.AgentPoolID = types.StringValue(v.AgentPool.ID)
+	}
+
 	if v.Description != "" {
 		result.Description = types.StringValue(v.Description)
 	}