Adding test to ensure we save notifications as alerted

alert_task.go

@@ -8,6 +8,8 @@ func SendAlerts() {
 		Preload("Tags.AlertGroups").
 		Where(Notification{Alerted: false, Read: false}).
 		Find(&n)
+
+	FilterCorrespondence(n)
 }
 
 func FilterCorrespondence(n []Notification) {
@@ -44,6 +46,17 @@ func FilterCorrespondence(n []Notification) {
 			}
 		}
 	}
+
+	UpdateNotifications(s)
+}
+
+func UpdateNotifications(n []Notification) {
+	for _, v := range n {
+		v.Alerted = true
+		v.Read = true
+
+		App.db.Save(&v)
+	}
 }
 
 func SendPushNotification(a AlertGroup, n []Notification) bool {

alert_task_test.go

@@ -1,16 +1,37 @@
 package main
 
 import (
-	"github.com/stretchr/testify/assert"
-	"testing"
+    "github.com/stretchr/testify/assert"
+    "testing"
 )
 
 func init() {
-	App = NewApp(":8888", "sqlite3", "/tmp/test.db", "test", "test")
+    App = NewApp(":8888", "sqlite3", "/tmp/test.db", "test", "test")
 
-	Migrate()
+    Migrate()
 
-	App.db.Delete(&Notification{})
-	App.db.Delete(&AlertGroup{})
-	App.db.Delete(&Tag{})
+    App.db.Delete(&Notification{})
+    App.db.Delete(&AlertGroup{})
+    App.db.Delete(&Tag{})
+}
+
+func TestItShouldUpdateNotificationsOnSend(t *testing.T) {
+    a := AlertGroup{Name: "NewTestGroup", Type: "email", Emails: "danzabian@gmail.com"}
+    App.db.Create(&a)
+
+    f := Tag{Name: "test"}
+    f.AlertGroups = append(f.AlertGroups, a)
+    App.db.Create(&f)
+
+    n := Notification{Message: "Test message"}
+    n.Tags = append(n.Tags, f)
+    App.db.Create(&n)
+
+    SendAlerts()
+
+    var o Notification
+    App.db.Where("ext_id = ?", n.ExtId).Find(&o)
+
+    assert.Equal(t, true, o.Alerted)
+    assert.Equal(t, true, o.Read)
 }

tag_test.go

@@ -49,7 +49,7 @@ func TestPostTagSuccess(t *testing.T) {
 	json.NewDecoder(resp.Body).Decode(&tag)
 
 	assert.Equal(t, 200, resp.StatusCode)
-	assert.Equal(t, "Test", tag.Name)
+	assert.Equal(t, "test1", tag.Name)
 }
 
 func TestPostTagValidation(t *testing.T) {