mirror of
https://github.com/v2fly/v2ray-core.git
synced 2024-11-07 10:47:48 -05:00
40 lines
792 B
Go
40 lines
792 B
Go
package assert
|
|
|
|
import (
|
|
"bytes"
|
|
"net"
|
|
|
|
"github.com/v2ray/v2ray-core/common/serial"
|
|
"github.com/v2ray/v2ray-core/testing/assert"
|
|
)
|
|
|
|
func IP(value net.IP) *IPSubject {
|
|
return &IPSubject{value: value}
|
|
}
|
|
|
|
type IPSubject struct {
|
|
assert.Subject
|
|
value net.IP
|
|
}
|
|
|
|
func (subject *IPSubject) Named(name string) *IPSubject {
|
|
subject.Subject.Named(name)
|
|
return subject
|
|
}
|
|
|
|
func (subject *IPSubject) DisplayString() string {
|
|
return subject.Subject.DisplayString(subject.value.String())
|
|
}
|
|
|
|
func (subject *IPSubject) IsNil() {
|
|
if subject.value != nil {
|
|
subject.Fail(subject.DisplayString(), "is", serial.StringLiteral("nil"))
|
|
}
|
|
}
|
|
|
|
func (subject *IPSubject) Equals(ip net.IP) {
|
|
if !bytes.Equal([]byte(subject.value), []byte(ip)) {
|
|
subject.Fail(subject.DisplayString(), "equals to", ip)
|
|
}
|
|
}
|