From 35f9f7ee5bcce4d8c0873d8515a178e6efbe40a2 Mon Sep 17 00:00:00 2001
From: 8naama <naama.bendalak@logz.io>
Date: Tue, 10 Dec 2024 11:42:23 +0200
Subject: [PATCH] test

---
 tests/apm_metrics_e2e_test.go | 106 ++++++++++++++++++++++++++++++++++
 1 file changed, 106 insertions(+)

diff --git a/tests/apm_metrics_e2e_test.go b/tests/apm_metrics_e2e_test.go
index 63cf0399..19a8e57d 100644
--- a/tests/apm_metrics_e2e_test.go
+++ b/tests/apm_metrics_e2e_test.go
@@ -1,8 +1,13 @@
 package tests
 
 import (
+	"encoding/json"
 	"fmt"
+	"go.uber.org/zap"
+	"io"
+	"net/http"
 	"os"
+	"strings"
 	"testing"
 )
 
@@ -33,3 +38,104 @@ func TestServiceGraphMetricsApm(t *testing.T) {
 	query := fmt.Sprintf(`{client_env_id='%s'}`, envId)
 	testMetrics(t, requiredMetrics, query)
 }
+
+func testMetrics(t *testing.T, requiredMetrics map[string][]string, query string) {
+	metricsApiKey := os.Getenv("LOGZIO_METRICS_API_KEY")
+	if metricsApiKey == "" {
+		t.Fatalf("LOGZIO_METRICS_API_KEY environment variable not set")
+	}
+
+	metricResponse, err := fetchMetrics(metricsApiKey, query)
+	if err != nil {
+		t.Fatalf("Failed to fetch metrics: %v", err)
+	}
+
+	if metricResponse.Status != "success" {
+		t.Errorf("No metrics found")
+	}
+	logger.Info("Found metrics", zap.Int("metrics_count", len(metricResponse.Data.Result)))
+	// Verify required metrics
+	missingMetrics := verifyMetrics(metricResponse, requiredMetrics)
+	if len(missingMetrics) > 0 {
+		var sb strings.Builder
+		for _, metric := range missingMetrics {
+			sb.WriteString(metric + "\n")
+		}
+		t.Errorf("Missing metrics or labels:\n%s", sb.String())
+	}
+}
+
+// fetchMetrics fetches the metrics from the logz.io API
+func fetchMetrics(metricsApiKey string, query string) (*MetricResponse, error) {
+	url := fmt.Sprintf("%s/metrics/prometheus/api/v1/query?query=%s", BaseLogzioApiUrl, query)
+	client := &http.Client{}
+	logger.Info("sending api request", zap.String("url", url))
+	req, err := http.NewRequest("GET", url, nil)
+	if err != nil {
+		return nil, err
+	}
+	req.Header.Set("Accept", "application/json")
+	req.Header.Set("X-API-TOKEN", metricsApiKey)
+
+	resp, err := client.Do(req)
+	if err != nil {
+		return nil, err
+	}
+	defer resp.Body.Close()
+
+	if resp.StatusCode != http.StatusOK {
+		return nil, fmt.Errorf("unexpected status code: %d", resp.StatusCode)
+	}
+
+	body, err := io.ReadAll(resp.Body)
+	if err != nil {
+		return nil, err
+	}
+
+	var metricResponse MetricResponse
+	err = json.Unmarshal(body, &metricResponse)
+	if err != nil {
+		return nil, err
+	}
+
+	return &metricResponse, nil
+}
+
+// verifyMetrics checks if the required metrics and their labels are present in the response
+func verifyMetrics(metricResponse *MetricResponse, requiredMetrics map[string][]string) []string {
+	missingMetrics := []string{}
+
+	for metricName, requiredLabels := range requiredMetrics {
+		found := false
+		for _, result := range metricResponse.Data.Result {
+			if result.Metric["__name__"] == metricName {
+				found = true
+				for _, label := range requiredLabels {
+					if _, exists := result.Metric[label]; !exists {
+						missingMetrics = append(missingMetrics, fmt.Sprintf("%s (missing label: %s)", metricName, label))
+					}
+				}
+			}
+		}
+		if !found {
+			missingMetrics = append(missingMetrics, metricName+" (not found)")
+		}
+	}
+	return deduplicate(missingMetrics)
+}
+
+// deduplicate removes duplicate strings from the input array.
+func deduplicate(data []string) []string {
+	uniqueMap := make(map[string]bool)
+	var uniqueList []string
+
+	for _, item := range data {
+		trimmedItem := strings.TrimSpace(item)
+		if _, exists := uniqueMap[trimmedItem]; !exists {
+			uniqueMap[trimmedItem] = true
+			uniqueList = append(uniqueList, trimmedItem)
+		}
+	}
+
+	return uniqueList
+}