package starnet

import (
	"context"
	"net"
	"testing"
)

func TestRequestCustomIP(t *testing.T) {
	customIPs := []string{"1.2.3.4", "5.6.7.8"}
	req := NewSimpleRequest("http://example.com", "GET").
		SetCustomIP(customIPs)

	if len(req.config.DNS.CustomIP) != 2 {
		t.Errorf("CustomIP length = %v; want 2", len(req.config.DNS.CustomIP))
	}

	for i, ip := range req.config.DNS.CustomIP {
		if ip != customIPs[i] {
			t.Errorf("CustomIP[%d] = %v; want %v", i, ip, customIPs[i])
		}
	}
}

func TestRequestCustomIPInvalid(t *testing.T) {
	req := NewSimpleRequest("http://example.com", "GET").
		SetCustomIP([]string{"invalid-ip"})

	if req.Err() == nil {
		t.Error("Expected error for invalid IP, got nil")
	}
}

func TestRequestCustomDNS(t *testing.T) {
	dnsServers := []string{"8.8.8.8", "1.1.1.1"}
	req := NewSimpleRequest("http://example.com", "GET").
		SetCustomDNS(dnsServers)

	if len(req.config.DNS.CustomDNS) != 2 {
		t.Errorf("CustomDNS length = %v; want 2", len(req.config.DNS.CustomDNS))
	}
}

func TestRequestCustomDNSInvalid(t *testing.T) {
	req := NewSimpleRequest("http://example.com", "GET").
		SetCustomDNS([]string{"invalid-dns"})

	if req.Err() == nil {
		t.Error("Expected error for invalid DNS, got nil")
	}
}

func TestRequestLookupFunc(t *testing.T) {
	called := false
	lookupFunc := func(ctx context.Context, host string) ([]net.IPAddr, error) {
		called = true
		return []net.IPAddr{
			{IP: net.ParseIP("1.2.3.4")},
		}, nil
	}

	req := NewSimpleRequest("http://example.com", "GET").
		SetLookupFunc(lookupFunc)

	if req.config.DNS.LookupFunc == nil {
		t.Error("LookupFunc not set")
	}

	// Call the function to verify it works
	ips, err := req.config.DNS.LookupFunc(context.Background(), "example.com")
	if err != nil {
		t.Errorf("LookupFunc error: %v", err)
	}
	if !called {
		t.Error("LookupFunc was not called")
	}
	if len(ips) != 1 {
		t.Errorf("IPs length = %v; want 1", len(ips))
	}
}

func TestDNSPriority(t *testing.T) {
	// CustomIP should have highest priority
	req := NewSimpleRequest("http://example.com", "GET").
		SetCustomIP([]string{"1.2.3.4"}).
		SetCustomDNS([]string{"8.8.8.8"}).
		SetLookupFunc(func(ctx context.Context, host string) ([]net.IPAddr, error) {
			return []net.IPAddr{{IP: net.ParseIP("5.6.7.8")}}, nil
		})

	// CustomIP should be set
	if len(req.config.DNS.CustomIP) == 0 {
		t.Error("CustomIP should be set")
	}

	// Others should also be set (but CustomIP takes priority in actual use)
	if len(req.config.DNS.CustomDNS) == 0 {
		t.Error("CustomDNS should be set")
	}
	if req.config.DNS.LookupFunc == nil {
		t.Error("LookupFunc should be set")
	}
}