net assert

common/net/testing/assert.go

@@ -0,0 +1,46 @@
+package testing
+
+import (
+	"reflect"
+
+	"v2ray.com/core/common/net"
+	"v2ray.com/ext/assert"
+)
+
+var IsIPv4 = assert.CreateOperator(reflect.TypeOf(net.Address(nil)), reflect.ValueOf(func(a net.Address) bool {
+	return a.Family().IsIPv4()
+}), 1, "is IPv4")
+
+var IsIPv6 = assert.CreateOperator(reflect.TypeOf(net.Address(nil)), reflect.ValueOf(func(a net.Address) bool {
+	return a.Family().IsIPv6()
+}), 1, "is IPv6")
+
+var IsIP = assert.CreateOperator(reflect.TypeOf(net.Address(nil)), reflect.ValueOf(func(a net.Address) bool {
+	return a.Family().IsIPv4() || a.Family().IsIPv6()
+}), 1, "is IP")
+
+var IsTCP = assert.CreateOperator(reflect.TypeOf(net.Destination{}), reflect.ValueOf(func(a net.Destination) bool {
+	return a.Network == net.Network_TCP
+}), 1, "is TCP")
+
+var IsUDP = assert.CreateOperator(reflect.TypeOf(net.Destination{}), reflect.ValueOf(func(a net.Destination) bool {
+	return a.Network == net.Network_UDP
+}), 1, "is UDP")
+
+var IsDomain = assert.CreateOperator(reflect.TypeOf(net.Address(nil)), reflect.ValueOf(func(a net.Address) bool {
+	return a.Family().IsDomain()
+}), 1, "is Domain")
+
+func init() {
+	assert.RegisterEqualsOperator(reflect.TypeOf((*net.Address)(nil)).Elem(), reflect.ValueOf(func(a, b net.Address) bool {
+		return a == b
+	}))
+
+	assert.RegisterEqualsOperator(reflect.TypeOf(net.Destination{}), reflect.ValueOf(func(a, b net.Destination) bool {
+		return a == b
+	}))
+
+	assert.RegisterEqualsOperator(reflect.TypeOf(net.Port(0)), reflect.ValueOf(func(a, b net.Port) bool {
+		return a == b
+	}))
+}