75 lines
1.7 KiB
Go
75 lines
1.7 KiB
Go
package mnet
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"net"
|
|
. "testing"
|
|
|
|
"github.com/mediocregopher/mediocre-go-lib/mcfg"
|
|
"github.com/mediocregopher/mediocre-go-lib/mctx"
|
|
"github.com/mediocregopher/mediocre-go-lib/mlog"
|
|
"github.com/mediocregopher/mediocre-go-lib/mrun"
|
|
"github.com/mediocregopher/mediocre-go-lib/mtest/massert"
|
|
)
|
|
|
|
func TestIsReservedIP(t *T) {
|
|
assertReserved := func(ipStr string) massert.Assertion {
|
|
ip := net.ParseIP(ipStr)
|
|
if ip == nil {
|
|
panic("ip:" + ipStr + " not valid")
|
|
}
|
|
return massert.Comment(massert.Equal(true, IsReservedIP(ip)),
|
|
"ip:%q", ipStr)
|
|
}
|
|
|
|
massert.Fatal(t, massert.All(
|
|
assertReserved("127.0.0.1"),
|
|
assertReserved("::ffff:127.0.0.1"),
|
|
assertReserved("192.168.40.50"),
|
|
assertReserved("::1"),
|
|
assertReserved("100::1"),
|
|
))
|
|
|
|
massert.Fatal(t, massert.None(
|
|
assertReserved("8.8.8.8"),
|
|
assertReserved("::ffff:8.8.8.8"),
|
|
assertReserved("2600:1700:7580:6e80:21c:25ff:fe97:44df"),
|
|
))
|
|
}
|
|
|
|
func TestMListen(t *T) {
|
|
ctx := mctx.ChildOf(mctx.New(), "test")
|
|
mlog.CtxSet(ctx, mlog.From(ctx).WithMaxLevel(mlog.DebugLevel))
|
|
|
|
l := MListen(ctx, "", "")
|
|
if err := mcfg.Populate(ctx, nil); err != nil {
|
|
t.Fatal(err)
|
|
} else if err := mrun.Start(ctx); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
go func() {
|
|
conn, err := net.Dial("tcp", l.Addr().String())
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
} else if _, err = fmt.Fprint(conn, "hello world"); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
conn.Close()
|
|
}()
|
|
|
|
conn, err := l.Accept()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
} else if b, err := ioutil.ReadAll(conn); err != nil {
|
|
t.Fatal(err)
|
|
} else if string(b) != "hello world" {
|
|
t.Fatalf("read %q from conn", b)
|
|
}
|
|
|
|
if err := mrun.Stop(ctx); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|