mirror of https://github.com/v2ray/v2ray-core
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
40 lines
792 B
40 lines
792 B
9 years ago
|
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)
|
||
|
|
}
|
||
|
|
}
|