mirror of https://github.com/v2ray/v2ray-core
47 lines
1.5 KiB
Go
47 lines
1.5 KiB
Go
|
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
|
||
|
|
}))
|
||
|
|
}
|