From 4178f73a0325f120d6dd16b97e08bc5755b0687e Mon Sep 17 00:00:00 2001
From: Markus Muellner <mmuellner@interact.nl>
Date: Thu, 31 Mar 2022 17:54:24 +0200
Subject: [PATCH] Add functional test for metrics endpoint

---
 system/lib.py             |  1 +
 system/t12_api/metrics.py | 26 ++++++++++++++++++++++++++
 2 files changed, 27 insertions(+)
 create mode 100644 system/t12_api/metrics.py

diff --git a/system/lib.py b/system/lib.py
index 9ffaf610d..013cfc3a8 100644
--- a/system/lib.py
+++ b/system/lib.py
@@ -142,6 +142,7 @@ class BaseTest(object):
         "gpgDisableSign": False,
         "ppaDistributorID": "ubuntu",
         "ppaCodename": "",
+        "enableMetricsEndpoint": True,
     }
     configOverride = {}
     environmentOverride = {}
diff --git a/system/t12_api/metrics.py b/system/t12_api/metrics.py
new file mode 100644
index 000000000..829ac390e
--- /dev/null
+++ b/system/t12_api/metrics.py
@@ -0,0 +1,26 @@
+from api_lib import APITest
+
+
+class MetricsEnabledAPITest(APITest):
+    """
+    GET /metrics
+    """
+
+    def check(self):
+        resp = self.get("/api/metrics")
+        self.check_equal(resp.status_code, 200)
+
+        apiRequestsInFlightGauge = "# TYPE aptly_api_http_requests_in_flight gauge"
+        self.check_in(apiRequestsInFlightGauge, resp.text)
+
+        apiRequestsTotalCounter = "# TYPE aptly_api_http_requests_total counter"
+        self.check_in(apiRequestsTotalCounter, resp.text)
+
+        apiRequestSizeSummary = "# TYPE aptly_api_http_request_size_bytes summary"
+        self.check_in(apiRequestSizeSummary, resp.text)
+
+        apiResponseSizeSummary = "# TYPE aptly_api_http_response_size_bytes summary"
+        self.check_in(apiResponseSizeSummary, resp.text)
+
+        apiRequestsDurationSummary = "# TYPE aptly_api_http_request_duration_seconds summary"
+        self.check_in(apiRequestsDurationSummary, resp.text)