package recon

import "testing"

func TestHostReachableHook(t *testing.T) {
	SetHostReachableHook(func(host string) bool { return host == "10.0.0.5" })
	t.Cleanup(func() { SetHostReachableHook(nil) })
	if !HostReachable("10.0.0.5") || HostReachable("10.0.0.6") {
		t.Fatal("reachability hook mismatch")
	}
}

func TestMergeAgentRelayScanWireguardHint(t *testing.T) {
	payload := []byte(`{"open_ports":[22],"udp_hints":[{"port":51820,"open":true,"service":"wireguard"}]}`)
	merged, err := MergeAgentRelayScan(RelayScanReport{Host: "10.1.2.50", ScannedVia: "relay"}, payload, false)
	if err != nil || len(merged.PathTracerHints) != 1 || merged.PathTracerHints[0] != "wireguard" {
		t.Fatalf("merged=%+v err=%v", merged, err)
	}
}