From 6a5371de361867f333aad8270c141252ee692525 Mon Sep 17 00:00:00 2001
From: Dustin Long <dustin.long@datadoghq.com>
Date: Wed, 15 Jan 2025 15:12:37 -0500
Subject: [PATCH] Client method for getting lastAPIKey

---
 test/fakeintake/client/client.go | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/test/fakeintake/client/client.go b/test/fakeintake/client/client.go
index 83a767596e6890..62ba7acc2aeb9f 100644
--- a/test/fakeintake/client/client.go
+++ b/test/fakeintake/client/client.go
@@ -394,6 +394,19 @@ func (c *Client) ConfigureOverride(override api.ResponseOverride) error {
 	return nil
 }
 
+// GetLastAPIKey returns the last apiKey sent with a payload to the intake
+func (c *Client) GetLastAPIKey() (string, error) {
+	resp, err := http.Get(fmt.Sprintf("%s/debug/lastAPIKey", c.fakeIntakeURL))
+	if err != nil {
+		return "", err
+	}
+	body, err := io.ReadAll(resp.Body)
+	if err != nil {
+		return "", err
+	}
+	return strings.TrimSpace(string(body)), nil
+}
+
 func (c *Client) getMetric(name string) ([]*aggregator.MetricSeries, error) {
 	err := c.getMetrics()
 	if err != nil {