provider/statuscake: use default status code list when updating test (#12375)

[github/fretlink/terraform-provider-statuscake.git] / resource_statuscaketest.go

diff --git a/resource_statuscaketest.go b/resource_statuscaketest.go
index d403848c042fdd2452f0afcf21ec3443e5f14e33..4912758d4049fb39dbf9e5fed196e36ea12cfdb5 100644 (file)
--- a/resource_statuscaketest.go
+++ b/resource_statuscaketest.go
@@ -18,45 +18,57 @@ func resourceStatusCakeTest() *schema.Resource {
                Read:   ReadTest,
 
                Schema: map[string]*schema.Schema{
-                       "test_id": &schema.Schema{
+                       "test_id": {
                                Type:     schema.TypeString,
                                Computed: true,
                        },
 
-                       "website_name": &schema.Schema{
+                       "website_name": {
                                Type:     schema.TypeString,
                                Required: true,
                        },
 
-                       "website_url": &schema.Schema{
+                       "website_url": {
                                Type:     schema.TypeString,
                                Required: true,
                        },
 
-                       "contact_id": &schema.Schema{
+                       "contact_id": {
                                Type:     schema.TypeInt,
                                Optional: true,
                        },
 
-                       "check_rate": &schema.Schema{
+                       "check_rate": {
                                Type:     schema.TypeInt,
                                Optional: true,
                                Default:  300,
                        },
 
-                       "test_type": &schema.Schema{
+                       "test_type": {
                                Type:     schema.TypeString,
                                Required: true,
                        },
 
-                       "paused": &schema.Schema{
+                       "paused": {
                                Type:     schema.TypeBool,
                                Optional: true,
                                Default:  false,
                        },
-                       "timeout": &schema.Schema{
+
+                       "timeout": {
                                Type:     schema.TypeInt,
-                               Computed: true,
+                               Optional: true,
+                               Default:  40,
+                       },
+
+                       "confirmations": {
+                               Type:     schema.TypeInt,
+                               Optional: true,
+                       },
+
+                       "port": {
+                               Type:     schema.TypeInt,
+                               Optional: true,
                        },
                },
        }
@@ -66,11 +78,15 @@ func CreateTest(d *schema.ResourceData, meta interface{}) error {
        client := meta.(*statuscake.Client)
 
        newTest := &statuscake.Test{
-               WebsiteName: d.Get("website_name").(string),
-               WebsiteURL:  d.Get("website_url").(string),
-               TestType:    d.Get("test_type").(string),
-               CheckRate:   d.Get("check_rate").(int),
-               ContactID:   d.Get("contact_id").(int),
+               WebsiteName:  d.Get("website_name").(string),
+               WebsiteURL:   d.Get("website_url").(string),
+               CheckRate:    d.Get("check_rate").(int),
+               TestType:     d.Get("test_type").(string),
+               Paused:       d.Get("paused").(bool),
+               Timeout:      d.Get("timeout").(int),
+               ContactID:    d.Get("contact_id").(int),
+               Confirmation: d.Get("confirmations").(int),
+               Port:         d.Get("port").(int),
        }
 
        log.Printf("[DEBUG] Creating new StatusCake Test: %s", d.Get("website_name").(string))
@@ -126,7 +142,15 @@ func ReadTest(d *schema.ResourceData, meta interface{}) error {
        if err != nil {
                return fmt.Errorf("Error Getting StatusCake Test Details for %s: Error: %s", d.Id(), err)
        }
+       d.Set("website_name", testResp.WebsiteName)
+       d.Set("website_url", testResp.WebsiteURL)
        d.Set("check_rate", testResp.CheckRate)
+       d.Set("test_type", testResp.TestType)
+       d.Set("paused", testResp.Paused)
+       d.Set("timeout", testResp.Timeout)
+       d.Set("contact_id", testResp.ContactID)
+       d.Set("confirmations", testResp.Confirmation)
+       d.Set("port", testResp.Port)
 
        return nil
 }
@@ -160,5 +184,22 @@ func getStatusCakeTestInput(d *schema.ResourceData) *statuscake.Test {
        if v, ok := d.GetOk("timeout"); ok {
                test.Timeout = v.(int)
        }
+       if v, ok := d.GetOk("contact_id"); ok {
+               test.ContactID = v.(int)
+       }
+       if v, ok := d.GetOk("confirmations"); ok {
+               test.Confirmation = v.(int)
+       }
+       if v, ok := d.GetOk("port"); ok {
+               test.Port = v.(int)
+       }
+
+       defaultStatusCodes := "204, 205, 206, 303, 400, 401, 403, 404, 405, 406, " +
+               "408, 410, 413, 444, 429, 494, 495, 496, 499, 500, 501, 502, 503, " +
+               "504, 505, 506, 507, 508, 509, 510, 511, 521, 522, 523, 524, 520, " +
+               "598, 599"
+
+       test.StatusCodes = defaultStatusCodes
+
        return test
 }