// Package proxy 提供 DNS 代理功能的测试。
//
// 该文件测试 proxy_dns.go 中的 DNS 相关功能，包括：
//   - DNS 解析器设置和获取
//   - DNS 缓存刷新机制
//   - HostClient 地址更新
//   - 错误处理场景
//
// 作者：xfy
package proxy

import (
	"context"
	"errors"
	"sync"
	"testing"
	"time"

	"rua.plus/lolly/internal/config"
	"rua.plus/lolly/internal/loadbalance"
	"rua.plus/lolly/internal/resolver"
)

// mockResolver 实现 resolver.Resolver 接口的模拟解析器。
type mockResolver struct {
	mu                   sync.RWMutex
	lookupResults        map[string][]string
	lookupError          error
	startErr             error
	stopErr              error
	lookupHostCalls      int
	lookupWithCacheCalls int
	startCalls           int
	stopCalls            int
}

func (m *mockResolver) LookupHost(_ context.Context, host string) ([]string, error) {
	m.mu.Lock()
	m.lookupHostCalls++
	m.mu.Unlock()
	return m.resolve(host)
}

func (m *mockResolver) LookupHostWithCache(_ context.Context, host string) ([]string, error) {
	m.mu.Lock()
	m.lookupWithCacheCalls++
	m.mu.Unlock()
	return m.resolve(host)
}

func (m *mockResolver) resolve(host string) ([]string, error) {
	m.mu.RLock()
	defer m.mu.RUnlock()

	if m.lookupError != nil {
		return nil, m.lookupError
	}
	if ips, ok := m.lookupResults[host]; ok {
		return ips, nil
	}
	return nil, errors.New("host not found in mock resolver")
}

func (m *mockResolver) Refresh(_ string) error {
	return nil
}

func (m *mockResolver) Start() error {
	m.mu.Lock()
	m.startCalls++
	m.mu.Unlock()
	return m.startErr
}

func (m *mockResolver) Stop() error {
	m.mu.Lock()
	m.stopCalls++
	m.mu.Unlock()
	return m.stopErr
}

func (m *mockResolver) Stats() resolver.Stats {
	m.mu.RLock()
	defer m.mu.RUnlock()
	return resolver.Stats{
		CacheHits:   int64(m.lookupWithCacheCalls),
		CacheMisses: int64(m.lookupHostCalls),
	}
}

func (m *mockResolver) getLookupWithCacheCalls() int {
	m.mu.RLock()
	defer m.mu.RUnlock()
	return m.lookupWithCacheCalls
}

// TestSetResolver 测试设置 DNS 解析器。
func TestSetResolver(t *testing.T) {
	cfg := &config.ProxyConfig{
		Path:        "/api",
		LoadBalance: "round_robin",
		Timeout:     config.ProxyTimeout{Connect: 5 * time.Second},
	}
	targets := []*loadbalance.Target{
		{URL: "http://localhost:8080"},
	}

	p, err := NewProxy(cfg, targets, nil, nil)
	if err != nil {
		t.Fatalf("NewProxy() error: %v", err)
	}

	// 初始状态 resolver 为 nil
	if p.resolver != nil {
		t.Error("resolver should be nil initially")
	}

	// 设置解析器
	mr := &mockResolver{}
	p.SetResolver(mr)

	if p.resolver != mr {
		t.Error("SetResolver() did not set resolver correctly")
	}
}

// TestGetResolverStats_NoResolver 测试没有解析器时返回空统计。


// TestGetResolverStats_WithResolver 测试有解析器时返回统计。


// TestStartWithResolver 测试启动代理时解析器正确启动。


// TestStartResolverFails 测试解析器启动失败时代理返回错误。


// TestStartIdempotent 测试 Start 是幂等的。


// TestStopIdempotent 测试 Stop 是幂等的。


// TestStopWithoutResolver 测试没有解析器时停止代理。


// TestRefreshDNS_Success 测试 DNS 刷新成功场景。


// TestRefreshDNS_LookupError 测试 DNS 刷新时查找失败场景。


// TestRefreshDNS_NoResolver 测试没有解析器时刷新不执行任何操作。


// TestRefreshDNS_IPAddressTarget 测试 IP 类型的目标不需要解析。


// TestRefreshDNS_RecentlyResolved 测试最近已解析的目标不需要再次解析。


// TestRefreshDNS_ExpiredResolve 测试 TTL 过期后需要重新解析。
// 该测试验证 TTL 过期检查的正确性：
// - 短时间内（< TTL）不需要重新解析
// - 长时间后（> TTL）需要重新解析


// TestUpdateHostClientAddr_HTTP 测试 HTTP 目标地址更新。


// TestUpdateHostClientAddr_HTTPS 测试 HTTPS 目标地址更新。


// TestUpdateHostClientAddr_DefaultPort 测试没有端口时使用默认端口。


// TestUpdateHostClientAddr_NonExistentTarget 测试不存在的目标不更新。


// TestGetResolverTTL 测试 TTL 获取。


// TestDNSRefreshLoop_StartStop 测试 DNS 刷新循环的启动和停止。


// TestMultipleTargets_Refresh 测试多目标刷新。


// TestStopResolverFails 测试停止解析器失败时返回错误。