v2ray-core/common/bitmask/byte_test.go

package bitmask_test

import (
	"testing"

	. "v2ray.com/core/common/bitmask"
	. "v2ray.com/ext/assert"
)

func TestBitmaskByte(t *testing.T) {
	assert := With(t)

	b := Byte(0)
	b.Set(Byte(1))
	assert(b.Has(1), IsTrue)

	b.Set(Byte(2))
	assert(b.Has(2), IsTrue)
	assert(b.Has(1), IsTrue)

	b.Clear(Byte(1))
	assert(b.Has(2), IsTrue)
	assert(b.Has(1), IsFalse)

	b.Toggle(Byte(2))
	assert(b.Has(2), IsFalse)
}