mirror of
https://github.com/v2fly/v2ray-core.git
synced 2024-08-28 23:34:16 -04:00
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.CreateMatcher(reflect.TypeOf(net.Address(nil)), reflect.ValueOf(func(a net.Address) bool {
|
|
return a.Family().IsIPv4()
|
|
}), 1, "is IPv4")
|
|
|
|
var IsIPv6 = assert.CreateMatcher(reflect.TypeOf(net.Address(nil)), reflect.ValueOf(func(a net.Address) bool {
|
|
return a.Family().IsIPv6()
|
|
}), 1, "is IPv6")
|
|
|
|
var IsIP = assert.CreateMatcher(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.CreateMatcher(reflect.TypeOf(net.Destination{}), reflect.ValueOf(func(a net.Destination) bool {
|
|
return a.Network == net.Network_TCP
|
|
}), 1, "is TCP")
|
|
|
|
var IsUDP = assert.CreateMatcher(reflect.TypeOf(net.Destination{}), reflect.ValueOf(func(a net.Destination) bool {
|
|
return a.Network == net.Network_UDP
|
|
}), 1, "is UDP")
|
|
|
|
var IsDomain = assert.CreateMatcher(reflect.TypeOf(net.Address(nil)), reflect.ValueOf(func(a net.Address) bool {
|
|
return a.Family().IsDomain()
|
|
}), 1, "is Domain")
|
|
|
|
func init() {
|
|
assert.RegisterEqualsMatcher(reflect.TypeOf((*net.Address)(nil)).Elem(), reflect.ValueOf(func(a, b net.Address) bool {
|
|
return a == b
|
|
}))
|
|
|
|
assert.RegisterEqualsMatcher(reflect.TypeOf(net.Destination{}), reflect.ValueOf(func(a, b net.Destination) bool {
|
|
return a == b
|
|
}))
|
|
|
|
assert.RegisterEqualsMatcher(reflect.TypeOf(net.Port(0)), reflect.ValueOf(func(a, b net.Port) bool {
|
|
return a == b
|
|
}))
|
|
}
|