Add projects

backend/sdk/cliproxy/pprof_server_test.go

@@ -0,0 +1,74 @@
+package cliproxy
+
+import (
+	"context"
+	"net/http"
+	"testing"
+
+	"github.com/router-for-me/CLIProxyAPI/v7/internal/config"
+)
+
+func TestPprofServerStopOwnedServerKeepsReplacement(t *testing.T) {
+	pprof := newPprofServer()
+	oldServer := &http.Server{}
+	replacement := &http.Server{}
+	pprof.server = replacement
+
+	if errStop := pprof.stopOwnedServerWithContext(context.Background(), oldServer, "old", "canceled", 1); errStop != nil {
+		t.Fatalf("stopOwnedServerWithContext() error = %v", errStop)
+	}
+	pprof.mu.Lock()
+	current := pprof.server
+	pprof.mu.Unlock()
+	if current != replacement {
+		t.Fatal("stopping a stale pprof server removed the replacement server")
+	}
+}
+
+func TestPprofServerSamePointerOwnerTransferKeepsCurrentServer(t *testing.T) {
+	pprof := newPprofServer()
+	server := &http.Server{}
+	pprof.server = server
+	pprof.addr = "127.0.0.1:6060"
+	pprof.enabled = true
+	pprof.owner = 1
+
+	cfg := &config.Config{}
+	cfg.Pprof.Enable = true
+	cfg.Pprof.Addr = "127.0.0.1:6060"
+	if !pprof.ApplyContext(context.Background(), cfg) {
+		t.Fatal("ApplyContext() = false, want same-pointer owner transfer")
+	}
+
+	pprof.mu.Lock()
+	owner := pprof.owner
+	pprof.mu.Unlock()
+	if owner == 1 {
+		t.Fatal("ApplyContext() did not transfer same-server ownership")
+	}
+	if errStop := pprof.stopOwnedServerWithContext(context.Background(), server, cfg.Pprof.Addr, "canceled", 1); errStop != nil {
+		t.Fatalf("stopOwnedServerWithContext() error = %v", errStop)
+	}
+	pprof.mu.Lock()
+	current := pprof.server
+	pprof.mu.Unlock()
+	if current != server {
+		t.Fatal("stale owner stopped the current same-pointer server")
+	}
+}
+
+func TestPprofServerServeFailureClearsTransferredOwner(t *testing.T) {
+	pprof := newPprofServer()
+	server := &http.Server{}
+	pprof.server = server
+	pprof.owner = 2
+
+	pprof.clearFailedServer(server)
+
+	pprof.mu.Lock()
+	current := pprof.server
+	pprof.mu.Unlock()
+	if current != nil {
+		t.Fatal("serve failure retained a server after ownership transferred")
+	}
+}