98 lines
1.8 KiB
Go
98 lines
1.8 KiB
Go
package object
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func testHeaders(num uint32) []ExtendedHeader {
|
|
res := make([]ExtendedHeader, num)
|
|
|
|
for i := uint32(0); i < num; i++ {
|
|
res[i].SetType(TypeFromUint32(i))
|
|
res[i].SetValue(i)
|
|
}
|
|
|
|
return res
|
|
}
|
|
|
|
func TestObject_ExtendedHeaders(t *testing.T) {
|
|
h := new(Header)
|
|
|
|
hs := testHeaders(2)
|
|
|
|
h.SetExtendedHeaders(hs)
|
|
|
|
require.Equal(t, hs, h.ExtendedHeaders())
|
|
}
|
|
|
|
func TestCopyExtendedHeaders(t *testing.T) {
|
|
require.Nil(t, CopyExtendedHeaders(nil))
|
|
|
|
h := new(Header)
|
|
|
|
// set initial headers
|
|
initHs := testHeaders(2)
|
|
h.SetExtendedHeaders(initHs)
|
|
|
|
// get extended headers copy
|
|
hsCopy := CopyExtendedHeaders(h)
|
|
|
|
// change the copy
|
|
hsCopy[0] = hsCopy[1]
|
|
|
|
// check that extended headers have not changed
|
|
require.Equal(t, initHs, h.ExtendedHeaders())
|
|
}
|
|
|
|
func TestSetExtendedHeadersCopy(t *testing.T) {
|
|
require.NotPanics(t, func() {
|
|
SetExtendedHeadersCopy(nil, nil)
|
|
})
|
|
|
|
h := new(Header)
|
|
|
|
// create source headers
|
|
srcHs := testHeaders(2)
|
|
|
|
// copy and set headers
|
|
SetExtendedHeadersCopy(h, srcHs)
|
|
|
|
// get extended headers
|
|
objHs := h.ExtendedHeaders()
|
|
|
|
// change the source headers
|
|
srcHs[0] = srcHs[1]
|
|
|
|
// check that headeres have not changed
|
|
require.Equal(t, objHs, h.ExtendedHeaders())
|
|
}
|
|
|
|
func TestHeaderRelations(t *testing.T) {
|
|
items := []struct {
|
|
relFn func(ExtendedHeaderType, ExtendedHeaderType) bool
|
|
|
|
base, ok, fail uint32
|
|
}{
|
|
{relFn: TypesEQ, base: 1, ok: 1, fail: 2},
|
|
{relFn: TypesLT, base: 1, ok: 2, fail: 0},
|
|
{relFn: TypesGT, base: 1, ok: 0, fail: 2},
|
|
}
|
|
|
|
for _, item := range items {
|
|
require.True(t,
|
|
item.relFn(
|
|
TypeFromUint32(item.base),
|
|
TypeFromUint32(item.ok),
|
|
),
|
|
)
|
|
|
|
require.False(t,
|
|
item.relFn(
|
|
TypeFromUint32(item.base),
|
|
TypeFromUint32(item.fail),
|
|
),
|
|
)
|
|
}
|
|
}
|