package peer

import "testing"

func TestStateServer_peerUpdate(t *testing.T) {
	h := NewPeerStateTestHarness()
	h.ConfigServer_Public(t)
	h.PeerUpdate(nil)
	assertType[*stateDisconnected](t, h.State)
}

func TestStateServer_onInit(t *testing.T) {
	h := NewPeerStateTestHarness()
	h.ConfigServer_Public(t)

	msg := controlMsg[packetInit]{
		SrcIP:   3,
		SrcAddr: addrPort4(1, 2, 3, 4, 1000),
		Packet: packetInit{
			TraceID: newTraceID(),
			Direct:  true,
			Version: 4,
		},
	}

	h.OnInit(msg)
	assertEqual(t, len(h.Sent), 1)
	assertEqual(t, h.Sent[0].Peer.DirectAddr, msg.SrcAddr)
	resp := assertType[packetInit](t, h.Sent[0].Packet)
	assertEqual(t, msg.Packet.TraceID, resp.TraceID)
	assertEqual(t, resp.Version, version)
}