package executor

import (
	"net/http"
	"testing"

	"github.com/router-for-me/CLIProxyAPI/v6/internal/config"
	cliproxyauth "github.com/router-for-me/CLIProxyAPI/v6/sdk/cliproxy/auth"
)

func TestProxyAwareClient_KeepAlivePoolReuse(t *testing.T) {
	cfg := &config.Config{}
	cfg.HTTPKeepAlive.Enabled = true
	cfg.HTTPKeepAlive.MaxIdleConns = 32
	cfg.HTTPKeepAlive.MaxIdleConnsPerHost = 8
	cfg.HTTPKeepAlive.IdleConnTimeoutSeconds = 30

	auth := &cliproxyauth.Auth{}
	c1 := newProxyAwareHTTPClient(t.Context(), cfg, auth, 0)
	c2 := newProxyAwareHTTPClient(t.Context(), cfg, auth, 0)
	if c1.Transport == nil || c2.Transport == nil {
		t.Fatalf("expected non-nil transports")
	}
	if c1.Transport != c2.Transport {
		t.Fatalf("expected pooled transport reuse")
	}
}

func TestProxyAwareClient_KeepAliveDisabledNoPoolReuse(t *testing.T) {
	cfg := &config.Config{}
	cfg.HTTPKeepAlive.Enabled = false
	auth := &cliproxyauth.Auth{ProxyURL: "http://127.0.0.1:8080"}

	c1 := newProxyAwareHTTPClient(t.Context(), cfg, auth, 0)
	c2 := newProxyAwareHTTPClient(t.Context(), cfg, auth, 0)
	if c1.Transport == nil || c2.Transport == nil {
		t.Fatalf("expected non-nil transports")
	}
	if c1.Transport == c2.Transport {
		t.Fatalf("expected distinct transports when keepalive pool disabled")
	}
	tr, ok := c1.Transport.(*http.Transport)
	if !ok {
		t.Fatalf("expected *http.Transport")
	}
	if !tr.DisableKeepAlives {
		t.Fatalf("expected keep-alives disabled when feature disabled")
	}
}