mediocre-go-lib/mnet/mnet_test.go

73 lines
1.6 KiB
Go
Raw Normal View History

2018-08-09 18:34:09 +00:00
package mnet
import (
2019-01-11 22:47:30 +00:00
"fmt"
"io/ioutil"
2018-08-09 18:34:09 +00:00
"net"
. "testing"
2019-01-11 22:47:30 +00:00
"github.com/mediocregopher/mediocre-go-lib/mcfg"
"github.com/mediocregopher/mediocre-go-lib/mctx"
"github.com/mediocregopher/mediocre-go-lib/mrun"
2018-08-09 18:34:09 +00:00
"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"),
))
}
2019-01-11 22:47:30 +00:00
func TestMListen(t *T) {
2019-01-13 02:01:16 +00:00
ctx := mctx.ChildOf(mctx.New(), "test")
l := MListen(ctx, "", "")
2019-01-11 22:47:30 +00:00
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)
}
2019-01-11 22:47:30 +00:00
}