v2ray-core/testing/unit/intsubject.go

package unit

import (
	"strconv"
)

type IntSubject struct {
	*Subject
	value int
}

func NewIntSubject(base *Subject, value int) *IntSubject {
	return &IntSubject{
		Subject: base,
		value:   value,
	}
}

func (subject *IntSubject) Named(name string) *IntSubject {
	subject.Subject.Named(name)
	return subject
}

func (subject *IntSubject) Fail(verb string, other int) {
	subject.FailWithMessage("Not true that " + subject.DisplayString() + " " + verb + " <" + strconv.Itoa(other) + ">.")
}

func (subject *IntSubject) DisplayString() string {
	return subject.Subject.DisplayString(strconv.Itoa(subject.value))
}

func (subject *IntSubject) Equals(expectation int) {
	if subject.value != expectation {
		subject.Fail("is equal to", expectation)
	}
}

func (subject *IntSubject) GreaterThan(expectation int) {
	if subject.value <= expectation {
		subject.Fail("is greater than", expectation)
	}
}

func (subject *IntSubject) LessThan(expectation int) {
	if subject.value >= expectation {
		subject.Fail("is less than", expectation)
	}
}