From ffd3ceef0dd021142a83d7816e282c1edcb44edf Mon Sep 17 00:00:00 2001
From: Alexander Hellbom <alexander.hellbom@gmail.com>
Date: Mon, 17 Oct 2016 23:35:38 +0200
Subject: [PATCH] Add schedule test

---
 .../resource_pagerduty_schedule_test.go       | 106 ++++++++++++++++++
 1 file changed, 106 insertions(+)
 create mode 100644 builtin/providers/pagerduty/resource_pagerduty_schedule_test.go

diff --git a/builtin/providers/pagerduty/resource_pagerduty_schedule_test.go b/builtin/providers/pagerduty/resource_pagerduty_schedule_test.go
new file mode 100644
index 0000000000..52d3321e03
--- /dev/null
+++ b/builtin/providers/pagerduty/resource_pagerduty_schedule_test.go
@@ -0,0 +1,106 @@
+package pagerduty
+
+import (
+	"fmt"
+	"testing"
+
+	"github.com/PagerDuty/go-pagerduty"
+	"github.com/hashicorp/terraform/helper/resource"
+	"github.com/hashicorp/terraform/terraform"
+)
+
+func TestAccPagerDutySchedule_Basic(t *testing.T) {
+	resource.Test(t, resource.TestCase{
+		PreCheck:     func() { testAccPreCheck(t) },
+		Providers:    testAccProviders,
+		CheckDestroy: testAccCheckPagerDutyScheduleDestroy,
+		Steps: []resource.TestStep{
+			resource.TestStep{
+				Config: testAccCheckPagerDutyScheduleConfig,
+				Check: resource.ComposeTestCheckFunc(
+					testAccCheckPagerDutyScheduleExists("pagerduty_schedule.foo"),
+					resource.TestCheckResourceAttr(
+						"pagerduty_schedule.foo", "name", "foo"),
+					resource.TestCheckResourceAttr(
+						"pagerduty_schedule.foo", "description", "Managed by Terraform"),
+					resource.TestCheckResourceAttr(
+						"pagerduty_schedule.foo", "time_zone", "Europe/Berlin"),
+					resource.TestCheckResourceAttr(
+						"pagerduty_schedule.foo", "schedule_layer.#", "1"),
+				),
+			},
+		},
+	})
+}
+
+func testAccCheckPagerDutyScheduleDestroy(s *terraform.State) error {
+	client := testAccProvider.Meta().(*pagerduty.Client)
+	for _, r := range s.RootModule().Resources {
+		if r.Type != "pagerduty_schedule" {
+			continue
+		}
+
+		_, err := client.GetSchedule(r.Primary.ID, pagerduty.GetScheduleOptions{})
+
+		if err == nil {
+			return fmt.Errorf("Schedule still exists")
+		}
+
+	}
+	return nil
+}
+
+func testAccCheckPagerDutyScheduleExists(n string) resource.TestCheckFunc {
+	return func(s *terraform.State) error {
+		rs, ok := s.RootModule().Resources[n]
+		if !ok {
+			return fmt.Errorf("Not found: %s", n)
+		}
+		if rs.Primary.ID == "" {
+			return fmt.Errorf("No Schedule ID is set")
+		}
+
+		client := testAccProvider.Meta().(*pagerduty.Client)
+
+		found, err := client.GetSchedule(rs.Primary.ID, pagerduty.GetScheduleOptions{})
+		if err != nil {
+			return err
+		}
+
+		if found.ID != rs.Primary.ID {
+			return fmt.Errorf("Schedule not found: %v - %v", rs.Primary.ID, found)
+		}
+
+		return nil
+	}
+}
+
+const testAccCheckPagerDutyScheduleConfig = `
+resource "pagerduty_user" "foo" {
+  name        = "foo"
+  email       = "foo@bar.com"
+  color       = "green"
+  role        = "user"
+  job_title   = "foo"
+  description = "foo"
+}
+
+resource "pagerduty_schedule" "foo" {
+  name      = "foo"
+  time_zone = "Europe/Berlin"
+
+  schedule_layer {
+    name                         = "foo"
+    start                        = "2015-11-06T20:00:00-05:00"
+    rotation_virtual_start       = "2015-11-06T20:00:00-05:00"
+    rotation_turn_length_seconds = 86401
+    users                        = ["${pagerduty_user.foo.id}"]
+
+    restriction {
+      type              = "daily_restriction"
+      start_time_of_day = "08:00:00"
+      duration_seconds  = 32101
+    }
+  }
+}
+`