package consul

import (
	"context"
	"net"
	"net/netip"
	"testing"
)

func TestRemoteAddrFromContext_Found(t *testing.T) {
	in := net.TCPAddrFromAddrPort(netip.MustParseAddrPort("1.2.3.4:8080"))
	ctx := ContextWithRemoteAddr(context.Background(), in)
	out, ok := RemoteAddrFromContext(ctx)
	if !ok {
		t.Fatalf("cannot get remote addr from context")
	}
	if in != out {
		t.Fatalf("expected %s but got %s instead", in, out)
	}
}

func TestRemoteAddrFromContext_NotFound(t *testing.T) {
	out, ok := RemoteAddrFromContext(context.Background())
	if ok || out != nil {
		t.Fatalf("expected remote addr %s to not be in context", out)
	}
}