feat: trim suffix from name servers

dnsimple · Mar 11, 2024 · 528a43e · 528a43e
1 parent 650a4c3
commit 528a43e
Show file tree

Hide file tree

Showing 4 changed files with 153 additions and 0 deletions.
diff --git a/internal/framework/modifiers/setplanmodifier.go b/internal/framework/modifiers/setplanmodifier.go
@@ -0,0 +1,45 @@
+package modifiers
+
+import (
+	"context"
+	"strings"
+
+	"github.com/hashicorp/terraform-plugin-framework/resource/schema/planmodifier"
+	"github.com/hashicorp/terraform-plugin-framework/types"
+)
+
+type setTrimSuffix struct {
+}
+
+func SetTrimSuffixValue() planmodifier.Set {
+	return setTrimSuffix{}
+}
+
+func (m setTrimSuffix) Description(context.Context) string {
+	return "Trim suffix from configuration value when comparing with state"
+}
+
+func (m setTrimSuffix) MarkdownDescription(ctx context.Context) string {
+	return m.Description(ctx)
+}
+
+func (m setTrimSuffix) PlanModifySet(ctx context.Context, req planmodifier.SetRequest, resp *planmodifier.SetResponse) {
+	if req.ConfigValue.IsNull() || req.PlanValue.IsUnknown() || req.PlanValue.IsNull() {
+		return
+	}
+
+	var planValue []string
+	resp.Diagnostics.Append(req.PlanValue.ElementsAs(ctx, &planValue, false)...)
+	if resp.Diagnostics.HasError() {
+		return
+	}
+
+	for i, element := range planValue {
+		planValue[i] = strings.TrimSuffix(element, ".")
+	}
+
+	serializedPlanValue, diags := types.SetValueFrom(ctx, types.StringType, planValue)
+	resp.Diagnostics.Append(diags...)
+
+	resp.PlanValue = serializedPlanValue
+}
diff --git a/internal/framework/modifiers/setplanmodifier_test.go b/internal/framework/modifiers/setplanmodifier_test.go
@@ -0,0 +1,84 @@
+package modifiers_test
+
+import (
+	"context"
+	"testing"
+
+	"github.com/hashicorp/terraform-plugin-framework/path"
+	"github.com/hashicorp/terraform-plugin-framework/resource/schema/planmodifier"
+	"github.com/hashicorp/terraform-plugin-framework/types"
+	"github.com/stretchr/testify/assert"
+	"github.com/terraform-providers/terraform-provider-dnsimple/internal/framework/modifiers"
+)
+
+func TestSetTrimSuffixValue(t *testing.T) {
+	t.Parallel()
+
+	serializeList := func(elements []string) types.Set {
+		listValue, _ := types.SetValueFrom(context.Background(), types.StringType, elements)
+		return listValue
+	}
+
+	type testCase struct {
+		plannedValue  types.Set
+		stateValue    types.Set
+		configValue   types.Set
+		expectedValue types.Set
+		expectError   bool
+	}
+	tests := map[string]testCase{
+		"trim `.` in string value ": {
+			plannedValue:  serializeList([]string{"beta.alpha.", "gamma.omega"}),
+			stateValue:    types.SetNull(types.StringType),
+			configValue:   serializeList([]string{"beta.alpha.", "gamma.omega"}),
+			expectedValue: serializeList([]string{"beta.alpha", "gamma.omega"}),
+		},
+		"when state is Null": {
+			plannedValue:  serializeList([]string{"beta.alpha.", "gamma.omega."}),
+			stateValue:    types.SetNull(types.StringType),
+			configValue:   serializeList([]string{"beta.alpha.", "gamma.omega."}),
+			expectedValue: serializeList([]string{"beta.alpha", "gamma.omega"}),
+		},
+		"when plan is Null": {
+			plannedValue:  types.SetNull(types.StringType),
+			stateValue:    types.SetNull(types.StringType),
+			configValue:   serializeList([]string{"beta.alpha.", "gamma.omega."}),
+			expectedValue: types.SetNull(types.StringType),
+		},
+		"when plan is Unknown": {
+			plannedValue:  types.SetUnknown(types.StringType),
+			stateValue:    types.SetNull(types.StringType),
+			configValue:   serializeList([]string{"beta.alpha.", "gamma.omega."}),
+			expectedValue: types.SetUnknown(types.StringType),
+		},
+	}
+
+	for name, test := range tests {
+		name, test := name, test
+		t.Run(name, func(t *testing.T) {
+			t.Parallel()
+
+			ctx := context.Background()
+			request := planmodifier.SetRequest{
+				Path:        path.Root("test"),
+				PlanValue:   test.plannedValue,
+				StateValue:  test.stateValue,
+				ConfigValue: test.configValue,
+			}
+			response := planmodifier.SetResponse{
+				PlanValue: request.PlanValue,
+			}
+			modifiers.SetTrimSuffixValue().PlanModifySet(ctx, request, &response)
+
+			if !response.Diagnostics.HasError() && test.expectError {
+				t.Fatal("expected error, got no error")
+			}
+
+			if response.Diagnostics.HasError() && !test.expectError {
+				t.Fatalf("got unexpected error: %s", response.Diagnostics)
+			}
+
+			assert.Equal(t, test.expectedValue, response.PlanValue)
+		})
+	}
+}
diff --git a/internal/framework/resources/domain_delegation_resource.go b/internal/framework/resources/domain_delegation_resource.go
@@ -15,6 +15,7 @@ import (
 	"github.com/hashicorp/terraform-plugin-framework/types"
 	"github.com/hashicorp/terraform-plugin-log/tflog"
 	"github.com/terraform-providers/terraform-provider-dnsimple/internal/framework/common"
+	"github.com/terraform-providers/terraform-provider-dnsimple/internal/framework/modifiers"
 	"github.com/terraform-providers/terraform-provider-dnsimple/internal/framework/utils"
 )
 
@@ -60,6 +61,9 @@ func (r *DomainDelegationResource) Schema(_ context.Context, _ resource.SchemaRe
 			"name_servers": schema.SetAttribute{
 				Required:    true,
 				ElementType: types.StringType,
+				PlanModifiers: []planmodifier.Set{
+					modifiers.SetTrimSuffixValue(),
+				},
 			},
 		},
 	}

diff --git a/internal/framework/resources/domain_delegation_resource_test.go b/internal/framework/resources/domain_delegation_resource_test.go
@@ -43,6 +43,18 @@ func TestAccDomainDelegationResource(t *testing.T) {
 					resource.TestCheckTypeSetElemAttr(resourceName, "name_servers.*", "ns-1556.awsdns-02.co.uk"),
 				),
 			},
+			{
+				Config:             testAccDomainDelegationResourceConfigWithSuffix(domainId),
+				ExpectNonEmptyPlan: false,
+				PlanOnly:           true,
+				Check: resource.ComposeAggregateTestCheckFunc(
+					resource.TestCheckResourceAttrSet(resourceName, "id"),
+					resource.TestCheckResourceAttr(resourceName, "domain", domainId),
+					resource.TestCheckResourceAttr(resourceName, "name_servers.#", "2"),
+					resource.TestCheckTypeSetElemAttr(resourceName, "name_servers.*", "ns-998.awsdns-60.net"),
+					resource.TestCheckTypeSetElemAttr(resourceName, "name_servers.*", "ns-1556.awsdns-02.co.uk"),
+				),
+			},
 			{
 				ResourceName:      resourceName,
 				ImportStateIdFunc: testAccDomainDelegationImportStateIDFunc(resourceName),
@@ -89,3 +101,11 @@ resource "dnsimple_domain_delegation" "test" {
 	name_servers = ["ns-1556.awsdns-02.co.uk", "ns-998.awsdns-60.net"]
 }`, domainId)
 }
+
+func testAccDomainDelegationResourceConfigWithSuffix(domainId string) string {
+	return fmt.Sprintf(`
+resource "dnsimple_domain_delegation" "test" {
+	domain = %[1]q
+	name_servers = ["ns-1556.awsdns-02.co.uk.", "ns-998.awsdns-60.net"]
+}`, domainId)
+}