2 files changed, 70 insertions, 7 deletions

diff --git a/resource_statuscaketest.go b/resource_statuscaketest.go
index a1d7354..0cdc76d 100644
--- a/resource_statuscaketest.go
+++ b/resource_statuscaketest.go
@@ -58,6 +58,10 @@ func resourceStatusCakeTest() *schema.Resource {
                                 Type:     schema.TypeInt,
                                 Computed: true,
                         },
+                        "confirmations": &schema.Schema{
+                                Type:     schema.TypeInt,
+                                Optional: true,
+                        },
                 },
         }
 }
@@ -66,10 +70,14 @@ 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),
+                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),
         }
 
         log.Printf("[DEBUG] Creating new StatusCake Test: %s", d.Get("website_name").(string))
@@ -125,7 +133,14 @@ 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)
 
         return nil
 }
@@ -162,5 +177,8 @@ func getStatusCakeTestInput(d *schema.ResourceData) *statuscake.Test {
         if v, ok := d.GetOk("contact_id"); ok {
                 test.ContactID = v.(int)
         }
+        if v, ok := d.GetOk("confirmations"); ok {
+                test.Confirmation = v.(int)
+        }
         return test
 }
diff --git a/resource_statuscaketest_test.go b/resource_statuscaketest_test.go
index bec1a45..ac3cc92 100644
--- a/resource_statuscaketest_test.go
+++ b/resource_statuscaketest_test.go
@@ -22,6 +22,7 @@ func TestAccStatusCake_basic(t *testing.T) {
                                 Config: testAccTestConfig_basic,
                                 Check: resource.ComposeTestCheckFunc(
                                         testAccTestCheckExists("statuscake_test.google", &test),
+                                        testAccTestCheckAttributes("statuscake_test.google", &test),
                                 ),
                         },
                 },
@@ -47,9 +48,11 @@ func TestAccStatusCake_withUpdate(t *testing.T) {
                                 Config: testAccTestConfig_update,
                                 Check: resource.ComposeTestCheckFunc(
                                         testAccTestCheckExists("statuscake_test.google", &test),
+                                        testAccTestCheckAttributes("statuscake_test.google", &test),
                                         resource.TestCheckResourceAttr("statuscake_test.google", "check_rate", "500"),
                                         resource.TestCheckResourceAttr("statuscake_test.google", "paused", "true"),
-                                        resource.TestCheckResourceAttr("statuscake_test.google", "contact_id", "23456"),
+                                        resource.TestCheckResourceAttr("statuscake_test.google", "contact_id", "0"),
+                                        resource.TestCheckResourceAttr("statuscake_test.google", "confirmations", "0"),
                                 ),
                         },
                 },
@@ -75,7 +78,7 @@ func testAccTestCheckExists(rn string, test *statuscake.Test) resource.TestCheck
 
                 gotTest, err := client.Tests().Detail(testId)
                 if err != nil {
-                        return fmt.Errorf("error getting project: %s", err)
+                        return fmt.Errorf("error getting test: %s", err)
                 }
 
                 *test = *gotTest
@@ -84,6 +87,48 @@ func testAccTestCheckExists(rn string, test *statuscake.Test) resource.TestCheck
         }
 }
 
+func testAccTestCheckAttributes(rn string, test *statuscake.Test) resource.TestCheckFunc {
+        return func(s *terraform.State) error {
+                attrs := s.RootModule().Resources[rn].Primary.Attributes
+
+                check := func(key, stateValue, testValue string) error {
+                        if testValue != stateValue {
+                                return fmt.Errorf("different values for %s in state (%s) and in statuscake (%s)",
+                                        key, stateValue, testValue)
+                        }
+                        return nil
+                }
+
+                for key, value := range attrs {
+                        var err error
+
+                        switch key {
+                        case "website_name":
+                                err = check(key, value, test.WebsiteName)
+                        case "website_url":
+                                err = check(key, value, test.WebsiteURL)
+                        case "check_rate":
+                                err = check(key, value, strconv.Itoa(test.CheckRate))
+                        case "test_type":
+                                err = check(key, value, test.TestType)
+                        case "paused":
+                                err = check(key, value, strconv.FormatBool(test.Paused))
+                        case "timeout":
+                                err = check(key, value, strconv.Itoa(test.Timeout))
+                        case "contact_id":
+                                err = check(key, value, strconv.Itoa(test.ContactID))
+                        case "confirmations":
+                                err = check(key, value, strconv.Itoa(test.Confirmation))
+                        }
+
+                        if err != nil {
+                                return err
+                        }
+                }
+                return nil
+        }
+}
+
 func testAccTestCheckDestroy(test *statuscake.Test) resource.TestCheckFunc {
         return func(s *terraform.State) error {
                 client := testAccProvider.Meta().(*statuscake.Client)
@@ -103,6 +148,7 @@ resource "statuscake_test" "google" {
         test_type = "HTTP"
         check_rate = 300
         contact_id = 12345
+        confirmations = 1
 }
 `
 
@@ -113,6 +159,5 @@ resource "statuscake_test" "google" {
         test_type = "HTTP"
         check_rate = 500
         paused = true
-        contact_id = 23456
 }
 `