2021-02-08 02:23:43 -05:00
|
|
|
package d2datautils
|
|
|
|
|
2021-03-23 22:30:57 -04:00
|
|
|
import "testing"
|
2021-02-08 02:23:43 -05:00
|
|
|
|
2021-03-23 22:30:57 -04:00
|
|
|
var testData = []byte{33, 23, 4, 33, 192, 243}
|
2021-02-08 02:23:43 -05:00
|
|
|
|
|
|
|
func TestBitmuncherCopy(t *testing.T) {
|
2021-03-23 22:30:57 -04:00
|
|
|
bm1 := CreateBitMuncher(testData, 0)
|
2021-02-08 02:23:43 -05:00
|
|
|
bm2 := CopyBitMuncher(bm1)
|
|
|
|
|
|
|
|
for i := range bm1.data {
|
|
|
|
if bm1.data[i] != bm2.data[i] {
|
|
|
|
t.Fatal("original bitmuncher isn't equal to copied")
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestBitmuncherSetOffset(t *testing.T) {
|
2021-03-23 22:30:57 -04:00
|
|
|
bm := CreateBitMuncher(testData, 0)
|
2021-02-08 02:23:43 -05:00
|
|
|
bm.SetOffset(5)
|
|
|
|
|
|
|
|
if bm.Offset() != 5 {
|
|
|
|
t.Fatal("Set Offset method didn't set offset to expected number")
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestBitmuncherSteBitsRead(t *testing.T) {
|
2021-03-23 22:30:57 -04:00
|
|
|
bm := CreateBitMuncher(testData, 0)
|
2021-02-08 02:23:43 -05:00
|
|
|
bm.SetBitsRead(8)
|
|
|
|
|
|
|
|
if bm.BitsRead() != 8 {
|
|
|
|
t.Fatal("Set bits read method didn't set bits read to expected value")
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestBitmuncherReadBit(t *testing.T) {
|
2021-03-23 22:30:57 -04:00
|
|
|
bm := CreateBitMuncher(testData, 0)
|
2021-02-08 03:55:02 -05:00
|
|
|
|
2021-02-08 02:23:43 -05:00
|
|
|
var result byte
|
|
|
|
|
|
|
|
for i := 0; i < bitsPerByte; i++ {
|
|
|
|
v := bm.GetBit()
|
|
|
|
result |= byte(v) << byte(i)
|
|
|
|
}
|
|
|
|
|
2021-03-23 22:30:57 -04:00
|
|
|
if result != testData[0] {
|
2021-02-08 02:23:43 -05:00
|
|
|
t.Fatal("result of rpeated 8 times get bit didn't return expected byte")
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestBitmuncherGetBits(t *testing.T) {
|
2021-03-23 22:30:57 -04:00
|
|
|
bm := CreateBitMuncher(testData, 0)
|
2021-02-08 02:23:43 -05:00
|
|
|
|
2021-03-23 22:30:57 -04:00
|
|
|
if byte(bm.GetBits(bitsPerByte)) != testData[0] {
|
2021-02-08 02:23:43 -05:00
|
|
|
t.Fatal("get bits didn't return expected value")
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2021-03-23 22:30:57 -04:00
|
|
|
func TestBitmuncherGetNoBits(t *testing.T) {
|
|
|
|
bm := CreateBitMuncher(testData, 0)
|
|
|
|
|
|
|
|
if bm.GetBits(0) != 0 {
|
|
|
|
t.Fatal("get bits didn't return expected value: 0")
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestBitmuncherGetSignedBits(t *testing.T) {
|
|
|
|
bm := CreateBitMuncher(testData, 0)
|
|
|
|
|
|
|
|
result := bm.GetSignedBits(6)
|
|
|
|
expected := -31
|
|
|
|
if result != expected {
|
|
|
|
t.Fatal("get signed bits didn't return expected value", result, expected)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestBitmuncherGetNoSignedBits(t *testing.T) {
|
|
|
|
bm := CreateBitMuncher(testData, 0)
|
|
|
|
|
|
|
|
result := bm.GetSignedBits(0)
|
|
|
|
expected := 0
|
|
|
|
if result != expected {
|
|
|
|
t.Fatal("get signed bits didn't return expected value", result, expected)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestBitmuncherGetOneSignedBit(t *testing.T) {
|
|
|
|
bm := CreateBitMuncher(testData, 0)
|
|
|
|
|
|
|
|
result := bm.GetSignedBits(1)
|
|
|
|
expected := -1
|
|
|
|
if result != expected {
|
|
|
|
t.Fatal("get signed bits didn't return expected value", result, expected)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2021-02-08 02:23:43 -05:00
|
|
|
func TestBitmuncherSkipBits(t *testing.T) {
|
2021-03-23 22:30:57 -04:00
|
|
|
bm := CreateBitMuncher(testData, 0)
|
2021-02-08 02:23:43 -05:00
|
|
|
|
|
|
|
bm.SkipBits(bitsPerByte)
|
|
|
|
|
2021-03-23 22:30:57 -04:00
|
|
|
if bm.GetByte() != testData[1] {
|
2021-02-08 02:23:43 -05:00
|
|
|
t.Fatal("skipping 8 bits didn't moved bit muncher's position into next byte")
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestBitmuncherGetInt32(t *testing.T) {
|
2021-03-23 22:30:57 -04:00
|
|
|
bm := CreateBitMuncher(testData, 0)
|
2021-02-08 02:23:43 -05:00
|
|
|
|
|
|
|
var testInt int32
|
|
|
|
|
|
|
|
for i := 0; i < bytesPerint32; i++ {
|
2021-03-23 22:30:57 -04:00
|
|
|
testInt |= int32(testData[i]) << int32(bitsPerByte*i)
|
2021-02-08 02:23:43 -05:00
|
|
|
}
|
|
|
|
|
2021-03-23 22:30:57 -04:00
|
|
|
if bm.GetInt32() != testInt {
|
2021-02-08 02:23:43 -05:00
|
|
|
t.Fatal("int32 value wasn't returned properly")
|
|
|
|
}
|
|
|
|
}
|
2021-03-23 22:30:57 -04:00
|
|
|
|
|
|
|
func TestBitmuncherGetUint32(t *testing.T) {
|
|
|
|
bm := CreateBitMuncher(testData, 0)
|
|
|
|
|
|
|
|
var testUint uint32
|
|
|
|
|
|
|
|
for i := 0; i < bytesPerint32; i++ {
|
|
|
|
testUint |= uint32(testData[i]) << uint32(bitsPerByte*i)
|
|
|
|
}
|
|
|
|
|
|
|
|
if bm.GetUInt32() != testUint {
|
|
|
|
t.Fatal("uint32 value wasn't returned properly")
|
|
|
|
}
|
|
|
|
}
|