Merge remote-tracking branch 'origin/master' into statuscake_adding_contact_group

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

diff --git a/resource_statuscaketest.go b/resource_statuscaketest.go
index 683413583542bcb6652c2e0c51d670f42330cca2..389a7d6867fbd0e25d6a982b585156772a751c2b 100644 (file)
--- a/resource_statuscaketest.go
+++ b/resource_statuscaketest.go
@@ -6,7 +6,7 @@ import (
 
        "log"
 
-       wtf "github.com/DreamItGetIT/statuscake"
+       "github.com/DreamItGetIT/statuscake"
        "github.com/hashicorp/terraform/helper/schema"
 )
 
@@ -33,6 +33,11 @@ func resourceStatusCakeTest() *schema.Resource {
                                Required: true,
                        },
 
+                       "contact_id": &schema.Schema{
+                               Type:     schema.TypeInt,
+                               Optional: true,
+                       },
+
                        "check_rate": &schema.Schema{
                                Type:     schema.TypeInt,
                                Optional: true,
@@ -53,23 +58,28 @@ func resourceStatusCakeTest() *schema.Resource {
                                Type:     schema.TypeInt,
                                Computed: true,
                        },
+                       "contact_id": &schema.Schema{
+                               Type:     schema.TypeInt,
+                               Optional: true,
+                       },
                },
        }
 }
 
 func CreateTest(d *schema.ResourceData, meta interface{}) error {
-       client := meta.(*wtf.Client)
+       client := meta.(*statuscake.Client)
 
-       newTest := &wtf.Test{
+       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),
        }
 
        log.Printf("[DEBUG] Creating new StatusCake Test: %s", d.Get("website_name").(string))
 
-       response, err := client.Tests().Put(newTest)
+       response, err := client.Tests().Update(newTest)
        if err != nil {
                return fmt.Errorf("Error creating StatusCake Test: %s", err.Error())
        }
@@ -81,12 +91,12 @@ func CreateTest(d *schema.ResourceData, meta interface{}) error {
 }
 
 func UpdateTest(d *schema.ResourceData, meta interface{}) error {
-       client := meta.(*wtf.Client)
+       client := meta.(*statuscake.Client)
 
        params := getStatusCakeTestInput(d)
 
        log.Printf("[DEBUG] StatusCake Test Update for %s", d.Id())
-       _, err := client.Tests().Put(params)
+       _, err := client.Tests().Update(params)
        if err != nil {
                return fmt.Errorf("Error Updating StatusCake Test: %s", err.Error())
        }
@@ -94,7 +104,7 @@ func UpdateTest(d *schema.ResourceData, meta interface{}) error {
 }
 
 func DeleteTest(d *schema.ResourceData, meta interface{}) error {
-       client := meta.(*wtf.Client)
+       client := meta.(*statuscake.Client)
 
        testId, parseErr := strconv.Atoi(d.Id())
        if parseErr != nil {
@@ -110,13 +120,13 @@ func DeleteTest(d *schema.ResourceData, meta interface{}) error {
 }
 
 func ReadTest(d *schema.ResourceData, meta interface{}) error {
-       client := meta.(*wtf.Client)
+       client := meta.(*statuscake.Client)
 
        testId, parseErr := strconv.Atoi(d.Id())
        if parseErr != nil {
                return parseErr
        }
-       testResp, err := client.Tests().Details(testId)
+       testResp, err := client.Tests().Detail(testId)
        if err != nil {
                return fmt.Errorf("Error Getting StatusCake Test Details for %s: Error: %s", d.Id(), err)
        }
@@ -125,12 +135,12 @@ func ReadTest(d *schema.ResourceData, meta interface{}) error {
        return nil
 }
 
-func getStatusCakeTestInput(d *schema.ResourceData) *wtf.Test {
+func getStatusCakeTestInput(d *schema.ResourceData) *statuscake.Test {
        testId, parseErr := strconv.Atoi(d.Id())
        if parseErr != nil {
                log.Printf("[DEBUG] Error Parsing StatusCake TestID: %s", d.Id())
        }
-       test := &wtf.Test{
+       test := &statuscake.Test{
                TestID: testId,
        }
        if v, ok := d.GetOk("website_name"); ok {
@@ -142,6 +152,9 @@ func getStatusCakeTestInput(d *schema.ResourceData) *wtf.Test {
        if v, ok := d.GetOk("check_rate"); ok {
                test.CheckRate = v.(int)
        }
+       if v, ok := d.GetOk("contact_id"); ok {
+               test.ContactID = v.(int)
+       }
        if v, ok := d.GetOk("test_type"); ok {
                test.TestType = v.(string)
        }
@@ -151,5 +164,8 @@ func getStatusCakeTestInput(d *schema.ResourceData) *wtf.Test {
        if v, ok := d.GetOk("timeout"); ok {
                test.Timeout = v.(int)
        }
+       if v, ok := d.GetOk("contact_id"); ok {
+               test.ContactID = v.(int)
+       }
        return test
 }