SUMO-258848: disallow updates to partition's analytics tier other than case of migration to flex

Author: raghavxk

sumologic/resource_sumologic_partition.go

@@ -1,6 +1,7 @@
 package sumologic
 
 import (
+	"fmt"
 	"log"
 	"strings"
 
@@ -135,11 +136,23 @@ func resourceSumologicPartitionDelete(d *schema.ResourceData, meta interface{})
 	return c.DecommissionPartition(d.Id())
 }
 func resourceSumologicPartitionUpdate(d *schema.ResourceData, meta interface{}) error {
+	c := meta.(*Client)
+
+	partitionId := d.Id()
+
 	spartition := resourceToPartition(d)
+	if d.HasChange("analytics_tier") {
+		currPartitionState, err := c.GetPartition(partitionId)
+		if err != nil {
+			return fmt.Errorf("error loading partition with id %s for analytics_tier update validation", partitionId)
+		}
 
-	c := meta.(*Client)
-	err := c.UpdatePartition(spartition)
+		if !areAnalyticsTierEqual(currPartitionState.AnalyticsTier, spartition.AnalyticsTier) {
+			return fmt.Errorf("analytics_tier of a partition can only be updated post creation if partition has been moved to flex tier")
+		}
+	}
 
+	err := c.UpdatePartition(spartition)
 	if err != nil {
 		return err
 	}
@@ -183,3 +196,15 @@ func resourceToPartition(d *schema.ResourceData) Partition {
 		IsIncludedInDefaultSearch:        isIncludedInDefaultSearchPtr,
 	}
 }
+
+func areAnalyticsTierEqual(a, b *string) bool {
+	if a == nil && b == nil {
+		return true
+	}
+
+	if a == nil || b == nil {
+		return false
+	}
+
+	return strings.EqualFold(*a, *b)
+}

sumologic/resource_sumologic_partition_test.go

@@ -2,6 +2,7 @@ package sumologic
 
 import (
 	"fmt"
+	"regexp"
 	"testing"
 
 	"github.com/hashicorp/terraform-plugin-sdk/helper/acctest"
@@ -50,6 +51,28 @@ func TestAccSumoLogicPartition_basic(t *testing.T) {
 						"sumologic_partition.foo", "is_compliant", "false"),
 				),
 			},
+			// allow change in casing of analytics_tier
+			{
+				Config: updatePartitionAnalyticsTierCase(testName),
+				Check: resource.ComposeTestCheckFunc(
+					testAccCheckPartitionExists("sumologic_partition.foo"),
+					resource.TestCheckResourceAttr(
+						"sumologic_partition.foo", "name", "terraform_acctest_"+testName),
+					resource.TestCheckResourceAttr(
+						"sumologic_partition.foo", "routing_expression", "_sourcecategory=*/Terraform"),
+					resource.TestCheckResourceAttr(
+						"sumologic_partition.foo", "analytics_tier", "continuous"),
+					resource.TestCheckResourceAttr(
+						"sumologic_partition.foo", "retention_period", "366"),
+					resource.TestCheckResourceAttr(
+						"sumologic_partition.foo", "is_compliant", "false"),
+				),
+			},
+			// Update analytics tier to a different value and assert error
+			{
+				Config:      updatePartitionAnalyticsTierConfig(testName),
+				ExpectError: regexp.MustCompile(`(?i)analytics_tier of a partition can only be updated post creation if partition has been moved to flex tier`),
+			},
 		},
 	})
 }
@@ -107,3 +130,27 @@ resource "sumologic_partition" "foo" {
 }
 `, testName)
 }
+
+func updatePartitionAnalyticsTierConfig(testName string) string {
+	return fmt.Sprintf(`
+resource "sumologic_partition" "foo" {
+    name = "terraform_acctest_%s"
+    routing_expression = "_sourcecategory=*/Terraform"
+	retention_period = 365
+	is_compliant = false
+	analytics_tier = "infrequent" // Changed from "continuous" to trigger error
+}
+`, testName)
+}
+
+func updatePartitionAnalyticsTierCase(testName string) string {
+	return fmt.Sprintf(`
+	resource "sumologic_partition" "foo" {
+		name = "terraform_acctest_%s"
+		routing_expression = "_sourcecategory=*/Terraform"
+		retention_period = 366
+		is_compliant = false
+		analytics_tier = "ContinuouS" // Changed case for "continuous"
+	}
+	`, testName)
+}