try stupid AI suggestions

Author: marctc

internal/cmd/integration-tests/tests/loki-enrich/enrich_test.go

@@ -7,6 +7,7 @@ import (
 	"encoding/json"
 	"fmt"
 	"net/http"
+	"os"
 	"testing"
 	"time"
 
@@ -48,6 +49,18 @@ func TestEnrichWithMissingLabels(t *testing.T) {
 	)
 }
 
+func getLokiAPIEndpoint() string {
+	host := os.Getenv("ALLOY_HOST")
+	port := os.Getenv("ALLOY_PORT")
+
+	if host != "" && port != "" {
+		return fmt.Sprintf("http://%s:%s/loki/api/v1/push", host, port)
+	}
+
+	// Fallback for manual testing
+	return "http://localhost:1514/loki/api/v1/push"
+}
+
 func sendTestLogsForDevice(t *testing.T, hostname string) {
 	networkLogs := []string{
 		"%LINK-3-UPDOWN: Interface GigabitEthernet1/0/1, changed state to up",
@@ -77,7 +90,7 @@ func sendTestLogsForDevice(t *testing.T, hostname string) {
 	body, err := json.Marshal(pushReq)
 	require.NoError(t, err)
 
-	resp, err := http.Post("http://alloy-loki-enrich:1514/loki/api/v1/push", "application/json", bytes.NewReader(body))
+	resp, err := http.Post(getLokiAPIEndpoint(), "application/json", bytes.NewReader(body))
 	require.NoError(t, err)
 	defer resp.Body.Close()
 	require.Equal(t, http.StatusNoContent, resp.StatusCode)

internal/cmd/integration-tests/utils.go

@@ -156,6 +156,34 @@ func runSingleTest(ctx context.Context, testDir string, port int) {
 
 	testCmd := exec.Command("go", "test")
 	testCmd.Dir = testDir
+
+	if dirName == "loki-enrich" {
+		mappedPort, err := alloyContainer.MappedPort(ctx, "1514/tcp")
+		if err != nil {
+			logChan <- TestLog{
+				TestDir:  dirName,
+				AlloyLog: fmt.Sprintf("failed to get mapped port: %v", err),
+			}
+			return
+		}
+
+		host, err := alloyContainer.Host(ctx)
+		if err != nil {
+			logChan <- TestLog{
+				TestDir:  dirName,
+				AlloyLog: fmt.Sprintf("failed to get container host: %v", err),
+			}
+			return
+		}
+
+		fmt.Printf("Loki API should be available at http://%s:%s/loki/api/v1/push\n",
+			host, mappedPort.Port())
+
+		testCmd.Env = append(os.Environ(),
+			fmt.Sprintf("ALLOY_HOST=%s", host),
+			fmt.Sprintf("ALLOY_PORT=%s", mappedPort.Port()))
+	}
+
 	testOutput, errTest := testCmd.CombinedOutput()
 
 	alloyLogs, _ := alloyContainer.Logs(ctx)