Merge pull request #709 from SumoLogic/priyansh-field

SUMO-252740 Add method to update status of fields resource

Author: priyanshgoel

CHANGELOG.md

@@ -1,5 +1,6 @@
 ## X.Y.Z (Unreleased)
 * Add new change notes here
+* Add support for update method to change state of the fields resource
 
 ## 3.0.0 (December 09, 2024)
 **REMOVALS:**

sumologic/resource_sumologic_field.go

@@ -1,6 +1,7 @@
 package sumologic
 
 import (
+	"errors"
 	"fmt"
 
 	"github.com/hashicorp/terraform-plugin-sdk/helper/schema"
@@ -10,6 +11,7 @@ func resourceSumologicField() *schema.Resource {
 	return &schema.Resource{
 		Create: resourceSumologicFieldCreate,
 		Read:   resourceSumologicFieldRead,
+		Update: resourceSumologicFieldUpdate,
 		Delete: resourceSumologicFieldDelete,
 		Importer: &schema.ResourceImporter{
 			State: resourceSumologicFieldImport,
@@ -38,7 +40,6 @@ func resourceSumologicField() *schema.Resource {
 			"state": {
 				Type:     schema.TypeString,
 				Optional: true,
-				ForceNew: true,
 			},
 		},
 	}
@@ -116,6 +117,43 @@ func resourceSumologicFieldImport(d *schema.ResourceData, meta interface{}) ([]*
 	return []*schema.ResourceData{d}, nil
 }
 
+func resourceSumologicFieldUpdate(d *schema.ResourceData, meta interface{}) error {
+	c := meta.(*Client)
+
+	id := d.Get("field_id").(string)
+	name := d.Get("field_name").(string)
+	state := d.Get("state").(string)
+	if id == "" {
+		newId, err := c.FindFieldId(name)
+		if err != nil {
+			return err
+		}
+		id = newId
+	}
+	_, err := c.GetField(id)
+
+	if err != nil {
+		return err
+	}
+
+	if state == "Enabled" {
+		err := c.EnableField(id)
+		if err != nil {
+			return err
+		}
+	} else if state == "Disabled" {
+		err := c.DisableField(id)
+		if err != nil {
+			return err
+		}
+	} else {
+		return errors.New("Invalid value of state field. Only Enabled or Disabled values are accepted")
+	}
+
+	return resourceSumologicFieldRead(d, meta)
+
+}
+
 func resourceToField(d *schema.ResourceData) Field {
 	return Field{
 		DataType:  d.Get("data_type").(string),

sumologic/resource_sumologic_field_test.go

@@ -56,6 +56,41 @@ func TestAccSumologicField_create(t *testing.T) {
 	})
 }
 
+func TestAccSumologicField_update(t *testing.T) {
+	var field Field
+	testFieldName := "fields_provider_test"
+	testDataType := "String"
+	testState := "Enabled"
+	updatedState := "Disabled"
+	resource.Test(t, resource.TestCase{
+		PreCheck:     func() { testAccPreCheck(t) },
+		Providers:    testAccProviders,
+		CheckDestroy: testAccCheckFieldDestroy(field),
+		Steps: []resource.TestStep{
+			{
+				Config: testAccSumologicField(testFieldName, testDataType, testState),
+				Check: resource.ComposeTestCheckFunc(
+					testAccCheckFieldExists("sumologic_field.test", &field, t),
+					testAccCheckFieldAttributes("sumologic_field.test"),
+					resource.TestCheckResourceAttr("sumologic_field.test", "field_name", testFieldName),
+					resource.TestCheckResourceAttr("sumologic_field.test", "data_type", testDataType),
+					resource.TestCheckResourceAttr("sumologic_field.test", "state", testState),
+				),
+			},
+			{
+				Config: testAccSumologicFieldUpdate(testFieldName, testDataType, updatedState),
+				Check: resource.ComposeTestCheckFunc(
+					testAccCheckFieldExists("sumologic_field.test", &field, t),
+					testAccCheckFieldAttributes("sumologic_field.test"),
+					resource.TestCheckResourceAttr("sumologic_field.test", "field_name", testFieldName),
+					resource.TestCheckResourceAttr("sumologic_field.test", "data_type", testDataType),
+					resource.TestCheckResourceAttr("sumologic_field.test", "state", updatedState),
+				),
+			},
+		},
+	})
+}
+
 func testAccCheckFieldDestroy(field Field) resource.TestCheckFunc {
 	return func(s *terraform.State) error {
 		client := testAccProvider.Meta().(*Client)
@@ -116,6 +151,16 @@ resource "sumologic_field" "test" {
 `, fieldName, dataType, state)
 }
 
+func testAccSumologicFieldUpdate(fieldName string, dataType string, state string) string {
+	return fmt.Sprintf(`
+resource "sumologic_field" "test" {
+    field_name = "%s"
+    data_type = "%s"
+    state = "%s"
+}
+`, fieldName, dataType, state)
+}
+
 func testAccCheckFieldAttributes(name string) resource.TestCheckFunc {
 	return func(s *terraform.State) error {
 		f := resource.ComposeTestCheckFunc(

sumologic/sumologic_field.go

@@ -94,6 +94,20 @@ func (s *Client) FindFieldId(name string) (string, error) {
 	return "", fmt.Errorf("Field \"%s\" not found", name)
 }
 
+func (s *Client) DisableField(id string) error {
+	urlWithParams := fmt.Sprintf("v1/fields/%s/disable", id)
+
+	_, err := s.Delete(urlWithParams)
+	return err
+}
+
+func (s *Client) EnableField(id string) error {
+	urlWithParams := fmt.Sprintf("v1/fields/%s/enable", id)
+
+	_, err := s.Put(urlWithParams, nil)
+	return err
+}
+
 type Field struct {
 	FieldId   string `json:"fieldId"`
 	DataType  string `json:"dataType"`