X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=manager%2Fcmd%2Fmanager_test.go;h=2589a9173ea442b31d40cf7cd7b85b50d667ca0a;hb=105030feb8fabd8b4ddff552c53e905146b2ea5f;hp=34b3902ef56da91d4dc4bf55a3c83df300af1327;hpb=bb99fb5f8451f6c53ca31524648d2e1b2dab525b;p=ric-plt%2Falarm-go.git

diff --git a/manager/cmd/manager_test.go b/manager/cmd/manager_test.go
index 34b3902..2589a91 100755
--- a/manager/cmd/manager_test.go
+++ b/manager/cmd/manager_test.go
@@ -35,6 +35,7 @@ import (
 	"net/http"
 	"net/http/httptest"
 	"os"
+	"os/exec"
 	"strconv"
 	"strings"
 	"testing"
@@ -47,7 +48,7 @@ var eventChan chan string
 
 // Test cases
 func TestMain(M *testing.M) {
-	alarmManager = NewAlarmManager("localhost:9093", 500)
+	alarmManager = NewAlarmManager("localhost:9093", 500, false)
 	alarmManager.alertInterval = 20000
 	go alarmManager.Run(false)
 	time.Sleep(time.Duration(10) * time.Second)
@@ -331,6 +332,20 @@ func TestActiveAlarmMaxThresholds(t *testing.T) {
 	ts.Close()
 }
 
+func TestGetPrometheusAlerts(t *testing.T) {
+	time.Sleep(1 * time.Second)
+	xapp.Logger.Info("TestGetPrometheusAlerts")
+	ts := CreatePromAlertSimulator2(t, "GET", "/alerts?active=true&inhibited=true&silenced=true&unprocessed=true")
+
+	commandReady := make(chan bool, 1)
+	command := "cli/alarm-cli"
+	args := []string{"gapam", "--active", "true", "--inhibited", "true", "--silenced", "--unprocessed", "true", "true", "--host", "localhost", "--port", "9093", "flushall"}
+	ExecCLICommand(commandReady, command, args...)
+	<-commandReady
+
+	ts.Close()
+}
+
 func VerifyAlarm(t *testing.T, a alarm.Alarm, expectedCount int) string {
 	receivedAlert := waitForEvent()
 
@@ -370,6 +385,33 @@ func CreatePromAlertSimulator(t *testing.T, method, url string, status int, resp
 	return ts
 }
 
+func CreatePromAlertSimulator2(t *testing.T, method, url string) *httptest.Server {
+	l, err := net.Listen("tcp", "localhost:9093")
+	if err != nil {
+		t.Error("Failed to create listener: " + err.Error())
+	}
+	ts := httptest.NewUnstartedServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		assert.Equal(t, r.Method, method)
+		assert.Equal(t, r.URL.String(), url)
+
+		w.Header().Add("Content-Type", "application/json")
+		w.WriteHeader(200)
+		// Read alerts from file
+		payload, err := readJSONFromFile("../testresources/prometheus-alerts.json")
+		if err != nil {
+			t.Error("Failed to send response: ", err)
+		}
+		_, err = w.Write(payload)
+		if err != nil {
+			t.Error("Failed to send response: " + err.Error())
+		}
+	}))
+	ts.Listener.Close()
+	ts.Listener = l
+	ts.Start()
+	return ts
+}
+
 func waitForEvent() string {
 	receivedAlert := <-eventChan
 	return receivedAlert
@@ -398,3 +440,27 @@ func checkResponseCode(t *testing.T, expected, actual int) bool {
 	}
 	return true
 }
+
+func ExecCLICommand(commandReady chan bool, command string, args ...string) {
+	go func() {
+		xapp.Logger.Info("Giving CLI command")
+		cmd := exec.Command(command, args...)
+		cmd.Dir = "../"
+		output, err := cmd.CombinedOutput()
+		if err != nil {
+			xapp.Logger.Info("CLI command failed out: %s", err)
+		}
+		xapp.Logger.Info("CLI command output: %s", output)
+		commandReady <- true
+		xapp.Logger.Info("CLI command completed")
+	}()
+}
+
+func readJSONFromFile(filename string) ([]byte, error) {
+	file, err := ioutil.ReadFile(filename)
+	if err != nil {
+		err := fmt.Errorf("readJSONFromFile() failed: Error: %v", err)
+		return nil, err
+	}
+	return file, nil
+}