Allow alert notification reminder to be turned on

Author: John Shahid

grafana/resource_alert_notification.go

@@ -1,14 +1,20 @@
 package grafana
 
 import (
+	"errors"
 	"fmt"
 	"log"
 	"strconv"
+	"time"
 
 	"github.com/hashicorp/terraform/helper/schema"
 	gapi "github.com/nytm/go-grafana-api"
 )
 
+var (
+	ErrFrequenceMustBeSet = errors.New("frequency must be set when send_reminder is set to 'true'")
+)
+
 func ResourceAlertNotification() *schema.Resource {
 	return &schema.Resource{
 		Create: CreateAlertNotification,
@@ -33,6 +39,18 @@ func ResourceAlertNotification() *schema.Resource {
 				Default:  false,
 			},
 
+			"send_reminder": {
+				Type:     schema.TypeBool,
+				Optional: true,
+				Default:  false,
+			},
+
+			"frequency": {
+				Type:     schema.TypeString,
+				Optional: true,
+				Default:  "",
+			},
+
 			"settings": {
 				Type:      schema.TypeMap,
 				Optional:  true,
@@ -143,11 +161,26 @@ func makeAlertNotification(d *schema.ResourceData) (*gapi.AlertNotification, err
 		}
 	}
 
+	sendReminder := d.Get("send_reminder").(bool)
+	frequency := d.Get("frequency").(string)
+
+	if sendReminder {
+		if frequency == "" {
+			return nil, ErrFrequenceMustBeSet
+		}
+
+		if _, err := time.ParseDuration(frequency); err != nil {
+			return nil, err
+		}
+	}
+
 	return &gapi.AlertNotification{
-		Id:        id,
-		Name:      d.Get("name").(string),
-		Type:      d.Get("type").(string),
-		IsDefault: d.Get("is_default").(bool),
-		Settings:  settings,
+		Id:           id,
+		Name:         d.Get("name").(string),
+		Type:         d.Get("type").(string),
+		IsDefault:    d.Get("is_default").(bool),
+		SendReminder: sendReminder,
+		Frequency:    frequency,
+		Settings:     settings,
 	}, err
 }

grafana/resource_alert_notification_test.go

@@ -24,12 +24,19 @@ func TestAccAlertNotification_basic(t *testing.T) {
 				Config: testAccAlertNotificationConfig_basic,
 				Check: resource.ComposeTestCheckFunc(
 					testAccAlertNotificationCheckExists("grafana_alert_notification.test", &alertNotification),
+					testAccAlertNotificationDefinition(&alertNotification),
 					resource.TestCheckResourceAttr(
 						"grafana_alert_notification.test", "type", "email",
 					),
 					resource.TestMatchResourceAttr(
 						"grafana_alert_notification.test", "id", regexp.MustCompile(`\d+`),
 					),
+					resource.TestCheckResourceAttr(
+						"grafana_alert_notification.test", "send_reminder", "true",
+					),
+					resource.TestCheckResourceAttr(
+						"grafana_alert_notification.test", "frequency", "12h",
+					),
 					resource.TestCheckResourceAttr(
 						"grafana_alert_notification.test", "settings.addresses", "[email protected]",
 					),
@@ -39,6 +46,38 @@ func TestAccAlertNotification_basic(t *testing.T) {
 	})
 }
 
+func TestAccAlertNotification_invalid_frequence(t *testing.T) {
+	var alertNotification gapi.AlertNotification
+
+	resource.Test(t, resource.TestCase{
+		PreCheck:     func() { testAccPreCheck(t) },
+		Providers:    testAccProviders,
+		CheckDestroy: testAccAlertNotificationCheckDestroy(&alertNotification),
+		Steps: []resource.TestStep{
+			{
+				ExpectError: regexp.MustCompile("invalid duration hi"),
+				Config:      testAccAlertNotificationConfig_invalid_frequency,
+			},
+		},
+	})
+}
+
+func TestAccAlertNotification_reminder_no_frequence(t *testing.T) {
+	var alertNotification gapi.AlertNotification
+
+	resource.Test(t, resource.TestCase{
+		PreCheck:     func() { testAccPreCheck(t) },
+		Providers:    testAccProviders,
+		CheckDestroy: testAccAlertNotificationCheckDestroy(&alertNotification),
+		Steps: []resource.TestStep{
+			{
+				ExpectError: regexp.MustCompile("frequency must be set when send_reminder is set to 'true'"),
+				Config:      testAccAlertNotificationConfig_reminder_no_frequency,
+			},
+		},
+	})
+}
+
 func testAccAlertNotificationCheckExists(rn string, a *gapi.AlertNotification) resource.TestCheckFunc {
 	return func(s *terraform.State) error {
 		rs, ok := s.RootModule().Resources[rn]
@@ -67,6 +106,16 @@ func testAccAlertNotificationCheckExists(rn string, a *gapi.AlertNotification) r
 	}
 }
 
+func testAccAlertNotificationDefinition(a *gapi.AlertNotification) resource.TestCheckFunc {
+	return func(s *terraform.State) error {
+		if !a.SendReminder {
+			return fmt.Errorf("send_reminder is not set properly")
+		}
+
+		return nil
+	}
+}
+
 func testAccAlertNotificationCheckDestroy(a *gapi.AlertNotification) resource.TestCheckFunc {
 	return func(s *terraform.State) error {
 		client := testAccProvider.Meta().(*gapi.Client)
@@ -82,6 +131,35 @@ const testAccAlertNotificationConfig_basic = `
 resource "grafana_alert_notification" "test" {
     type = "email"
     name = "terraform-acc-test"
+		send_reminder = true
+		frequency = "12h"
+    settings = {
+			"addresses" = "[email protected]"
+			"uploadImage" = "false"
+			"autoResolve" = "true"
+		}
+}
+`
+
+const testAccAlertNotificationConfig_invalid_frequency = `
+resource "grafana_alert_notification" "test" {
+    type = "email"
+    name = "terraform-acc-test"
+		send_reminder = true
+		frequency = "hi"
+    settings = {
+			"addresses" = "[email protected]"
+			"uploadImage" = "false"
+			"autoResolve" = "true"
+		}
+}
+`
+
+const testAccAlertNotificationConfig_reminder_no_frequency = `
+resource "grafana_alert_notification" "test" {
+    type = "email"
+    name = "terraform-acc-test"
+		send_reminder = true
     settings = {
 			"addresses" = "[email protected]"
 			"uploadImage" = "false"