CLOUDP-64113: implementing e2e test for alerts ack and unack atlas (#186)

Author: andreaangiolillo

e2e/atlas/alerts_test.go

@@ -21,13 +21,14 @@ import (
 	"os/exec"
 	"path/filepath"
 	"testing"
+	"time"
 
 	"github.com/mongodb/go-client-mongodb-atlas/mongodbatlas"
 )
 
 const (
-	closed                               = "CLOSED"
-	replication_oplog_window_running_out = "REPLICATION_OPLOG_WINDOW_RUNNING_OUT"
+	open                            = "OPEN"
+	users_without_multi_factor_auth = "USERS_WITHOUT_MULTI_FACTOR_AUTH"
 )
 
 func TestAtlasAlerts(t *testing.T) {
@@ -43,9 +44,9 @@ func TestAtlasAlerts(t *testing.T) {
 
 	atlasEntity := "atlas"
 	alertsEntity := "alerts"
+	alertID := "5ecbef6b2359825e889837a7"
 
 	t.Run("Describe", func(t *testing.T) {
-		alertID := "5e4d20ff5cc174527c22c606"
 
 		cmd := exec.Command(cliPath,
 			atlasEntity,
@@ -72,12 +73,12 @@ func TestAtlasAlerts(t *testing.T) {
 			t.Errorf("got=%#v\nwant=%#v\n", alert.ID, alertID)
 		}
 
-		if alert.Status != closed {
-			t.Errorf("got=%#v\nwant=%#v\n", alert.Status, closed)
+		if alert.Status != open {
+			t.Errorf("got=%#v\nwant=%#v\n", alert.Status, open)
 		}
 
-		if alert.EventTypeName != replication_oplog_window_running_out {
-			t.Errorf("got=%#v\nwant=%#v\n", alert.EventTypeName, replication_oplog_window_running_out)
+		if alert.EventTypeName != users_without_multi_factor_auth {
+			t.Errorf("got=%#v\nwant=%#v\n", alert.EventTypeName, users_without_multi_factor_auth)
 		}
 
 	})
@@ -133,11 +134,12 @@ func TestAtlasAlerts(t *testing.T) {
 			t.Fatalf("unexpected error: %v", err)
 		}
 
-		if len(alerts.Results) > 0 {
-			t.Errorf("got=%#v\nwant=0\n", len(alerts.Results))
+		if len(alerts.Results) == 0 {
+			t.Errorf("got=0\nwant>0\n")
 		}
 
 	})
+
 	t.Run("List with status CLOSED", func(t *testing.T) {
 
 		cmd := exec.Command(cliPath,
@@ -167,4 +169,88 @@ func TestAtlasAlerts(t *testing.T) {
 		}
 
 	})
+
+	t.Run("Acknowledge", func(t *testing.T) {
+
+		cmd := exec.Command(cliPath,
+			atlasEntity,
+			alertsEntity,
+			"ack",
+			alertID,
+			"--until",
+			time.Now().Format(time.RFC3339))
+
+		cmd.Env = os.Environ()
+		resp, err := cmd.CombinedOutput()
+
+		if err != nil {
+			t.Fatalf("unexpected error: %v, resp: %v", err, string(resp))
+		}
+
+		alert := mongodbatlas.Alert{}
+		err = json.Unmarshal(resp, &alert)
+
+		if err != nil {
+			t.Fatalf("unexpected error: %v", err)
+		}
+
+		if alert.ID != alertID {
+			t.Errorf("got=%#v\nwant%v\n", alert.ID, alertID)
+		}
+	})
+
+	t.Run("Acknowledge Forever", func(t *testing.T) {
+
+		cmd := exec.Command(cliPath,
+			atlasEntity,
+			alertsEntity,
+			"ack",
+			alertID,
+			"--forever")
+
+		cmd.Env = os.Environ()
+		resp, err := cmd.CombinedOutput()
+
+		if err != nil {
+			t.Fatalf("unexpected error: %v, resp: %v", err, string(resp))
+		}
+
+		alert := mongodbatlas.Alert{}
+		err = json.Unmarshal(resp, &alert)
+
+		if err != nil {
+			t.Fatalf("unexpected error: %v", err)
+		}
+
+		if alert.ID != alertID {
+			t.Errorf("got=%#v\nwant%v\n", alert.ID, alertID)
+		}
+	})
+
+	t.Run("UnaAcknowledge", func(t *testing.T) {
+
+		cmd := exec.Command(cliPath,
+			atlasEntity,
+			alertsEntity,
+			"unack",
+			alertID)
+
+		cmd.Env = os.Environ()
+		resp, err := cmd.CombinedOutput()
+
+		if err != nil {
+			t.Fatalf("unexpected error: %v, resp: %v", err, string(resp))
+		}
+
+		alert := mongodbatlas.Alert{}
+		err = json.Unmarshal(resp, &alert)
+
+		if err != nil {
+			t.Fatalf("unexpected error: %v", err)
+		}
+
+		if alert.ID != alertID {
+			t.Errorf("got=%#v\nwant%v\n", alert.ID, alertID)
+		}
+	})
 }