Fix TestPostData

Author: andy-tier1app

internal/agent/ondemand/ondemand_test.go

@@ -2,27 +2,25 @@ package ondemand
 
 import (
 	"fmt"
+	"io"
+	"net/http"
+	"net/http/httptest"
 	"os"
+	"sync"
 	"testing"
 	"time"
 	"yc-agent/internal/capture"
 	"yc-agent/internal/capture/executils"
 
 	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
 )
 
 const (
 	api  = "tier1app@12312-12233-1442134-112"
 	host = "https://test.gceasy.io"
 )
 
-func init() {
-	// err := os.Chdir("testdata")
-	// if err != nil {
-	// 	panic(err)
-	// }
-}
-
 func TestFindGCLog(t *testing.T) {
 	// Skip: Test requires Java environment with compiled MyClass.java to run background Java processes.
 	// This is an integration test that verifies GC log detection from running Java processes.
@@ -90,82 +88,78 @@ func TestFindGCLog(t *testing.T) {
 }
 
 func TestPostData(t *testing.T) {
-	// Skip: Test makes external HTTP calls to test.gceasy.io which is flaky in CI.
-	// This is an integration test that should be run manually or in a dedicated integration test suite.
-	t.Skip("Skipping: test makes external HTTP calls to test.gceasy.io")
+	// Track received requests for assertions
+	var mu sync.Mutex
+	receivedRequests := make(map[string]struct {
+		method   string
+		bodySize int
+	})
 
+	// Create mock HTTP server
+	server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		// Read body to get size
+		body, _ := io.ReadAll(r.Body)
+		defer r.Body.Close()
+
+		// Extract dt parameter
+		dt := r.URL.Query().Get("dt")
+
+		mu.Lock()
+		receivedRequests[dt] = struct {
+			method   string
+			bodySize int
+		}{
+			method:   r.Method,
+			bodySize: len(body),
+		}
+		mu.Unlock()
+
+		w.WriteHeader(http.StatusOK)
+		w.Write([]byte("success"))
+	}))
+	defer server.Close()
+
+	// Build endpoint using mock server URL
 	timestamp := time.Now().Format("2006-01-02T15-04-05")
 	parameters := fmt.Sprintf("de=%s&ts=%s", capture.GetOutboundIP().String(), timestamp)
-	endpoint := fmt.Sprintf("%s/ycrash-receiver?apiKey=%s&%s", host, api, parameters)
+	endpoint := fmt.Sprintf("%s/ycrash-receiver?apiKey=%s&%s", server.URL, api, parameters)
+
+	// Test data files - path relative to ondemand package
+	testFiles := []struct {
+		path   string
+		dtType string
+	}{
+		{"../testdata/top.out", "top"},
+		{"../testdata/disk.out", "df"},
+		{"../testdata/netstat.out", "ns"},
+		{"../testdata/ps.out", "ps"},
+		{"../testdata/vmstat.out", "vmstat"},
+		{"../testdata/gc.log", "gc"},
+		{"../testdata/threaddump.out", "td"},
+	}
 
-	t.Run("requestFin", func(t *testing.T) {
-		finEp := fmt.Sprintf("%s/yc-fin?apiKey=%s&%s", host, api, parameters)
-		RequestFin(finEp)
-	})
+	for _, tf := range testFiles {
+		t.Run(tf.dtType, func(t *testing.T) {
+			file, err := os.Open(tf.path)
+			require.NoError(t, err, "Failed to open test file: %s", tf.path)
+			defer file.Close()
 
-	vmstat, err := os.Open("testdata/vmstat.out")
-	if err != nil {
-		return
-	}
-	defer vmstat.Close()
-	ps, err := os.Open("testdata/ps.out")
-	if err != nil {
-		t.Fatal(err)
-	}
-	defer ps.Close()
-	top, err := os.Open("testdata/top.out")
-	if err != nil {
-		t.Fatal(err)
-	}
-	defer top.Close()
-	df, err := os.Open("testdata/disk.out")
-	if err != nil {
-		t.Fatal(err)
-	}
-	defer df.Close()
-	netstat, err := os.Open("testdata/netstat.out")
-	if err != nil {
-		t.Fatal(err)
+			msg, ok := capture.PostData(endpoint, tf.dtType, file)
+			assert.True(t, ok, "PostData failed for %s: %s", tf.dtType, msg)
+		})
 	}
-	defer netstat.Close()
-	gc, err := os.Open("testdata/gc.log")
-	if err != nil {
-		t.Fatal(err)
-	}
-	defer gc.Close()
-	td, err := os.Open("testdata/threaddump.out")
-	if err != nil {
-		t.Fatal(err)
-	}
-	defer td.Close()
 
-	msg, ok := capture.PostData(endpoint, "top", top)
-	if !ok {
-		t.Fatal("post data failed", msg)
-	}
-	msg, ok = capture.PostData(endpoint, "df", df)
-	if !ok {
-		t.Fatal("post data failed", msg)
-	}
-	msg, ok = capture.PostData(endpoint, "ns", netstat)
-	if !ok {
-		t.Fatal("post data failed", msg)
-	}
-	msg, ok = capture.PostData(endpoint, "ps", ps)
-	if !ok {
-		t.Fatal("post data failed", msg)
-	}
-	msg, ok = capture.PostData(endpoint, "vmstat", vmstat)
-	if !ok {
-		t.Fatal("post data failed", msg)
-	}
-	msg, ok = capture.PostData(endpoint, "gc", gc)
-	if !ok {
-		t.Fatal("post data failed", msg)
-	}
-	msg, ok = capture.PostData(endpoint, "td", td)
-	if !ok {
-		t.Fatal("post data failed", msg)
+	// Verify all requests were received correctly
+	mu.Lock()
+	defer mu.Unlock()
+
+	for _, tf := range testFiles {
+		req, exists := receivedRequests[tf.dtType]
+		assert.True(t, exists, "Request for dt=%s was not received", tf.dtType)
+		if exists {
+			assert.Equal(t, "POST", req.method, "Expected POST method for dt=%s", tf.dtType)
+			assert.Greater(t, req.bodySize, 0, "Expected non-empty body for dt=%s", tf.dtType)
+		}
 	}
 }