28 lines
633 B
Go
28 lines
633 B
Go
|
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)
|
||
|
|
}
|
||
|
|
}
|