257 lines
8.0 KiB
Go
257 lines
8.0 KiB
Go
package graphvent
|
|
|
|
import (
|
|
"testing"
|
|
"reflect"
|
|
"fmt"
|
|
"time"
|
|
)
|
|
|
|
func TestSerializeTest(t *testing.T) {
|
|
ctx := logTestContext(t, []string{"test", "serialize", "deserialize_types"})
|
|
testSerialize(t, ctx, map[string][]NodeID{"test_group": {RandID(), RandID(), RandID()}})
|
|
testSerialize(t, ctx, map[NodeID]ReqState{
|
|
RandID(): Locked,
|
|
RandID(): Unlocked,
|
|
})
|
|
}
|
|
|
|
func TestSerializeBasic(t *testing.T) {
|
|
ctx := logTestContext(t, []string{"test", "serialize"})
|
|
testSerializeComparable[bool](t, ctx, true)
|
|
|
|
type bool_wrapped bool
|
|
err := ctx.RegisterType(reflect.TypeOf(bool_wrapped(true)), NewSerializedType("BOOL_WRAPPED"), nil, nil, nil, DeserializeBool[bool_wrapped])
|
|
fatalErr(t, err)
|
|
testSerializeComparable[bool_wrapped](t, ctx, true)
|
|
|
|
testSerializeSlice[[]bool](t, ctx, []bool{false, false, true, false})
|
|
testSerializeComparable[string](t, ctx, "test")
|
|
testSerializeComparable[float32](t, ctx, 0.05)
|
|
testSerializeComparable[float64](t, ctx, 0.05)
|
|
testSerializeComparable[uint](t, ctx, uint(1234))
|
|
testSerializeComparable[uint8] (t, ctx, uint8(123))
|
|
testSerializeComparable[uint16](t, ctx, uint16(1234))
|
|
testSerializeComparable[uint32](t, ctx, uint32(12345))
|
|
testSerializeComparable[uint64](t, ctx, uint64(123456))
|
|
testSerializeComparable[int](t, ctx, 1234)
|
|
testSerializeComparable[int8] (t, ctx, int8(-123))
|
|
testSerializeComparable[int16](t, ctx, int16(-1234))
|
|
testSerializeComparable[int32](t, ctx, int32(-12345))
|
|
testSerializeComparable[int64](t, ctx, int64(-123456))
|
|
testSerializeComparable[time.Duration](t, ctx, time.Duration(100))
|
|
testSerializeComparable[time.Time](t, ctx, time.Now().Truncate(0))
|
|
testSerializeSlice[[]int](t, ctx, []int{123, 456, 789, 101112})
|
|
testSerializeSlice[[]int](t, ctx, ([]int)(nil))
|
|
testSerializeSliceSlice[[][]int](t, ctx, [][]int{{123, 456, 789, 101112}, {3253, 2341, 735, 212}, {123, 51}, nil})
|
|
testSerializeSliceSlice[[][]string](t, ctx, [][]string{{"123", "456", "789", "101112"}, {"3253", "2341", "735", "212"}, {"123", "51"}, nil})
|
|
|
|
testSerialize(t, ctx, map[int8]map[*int8]string{})
|
|
testSerialize(t, ctx, map[int8]time.Time{
|
|
1: time.Now(),
|
|
3: time.Now().Add(time.Second),
|
|
0: time.Now().Add(time.Second*2),
|
|
4: time.Now().Add(time.Second*3),
|
|
})
|
|
|
|
testSerialize(t, ctx, Tree{
|
|
NodeTypeSerialized: nil,
|
|
SerializedTypeSerialized: Tree{
|
|
NodeTypeSerialized: Tree{},
|
|
},
|
|
})
|
|
|
|
var i interface{} = nil
|
|
testSerialize(t, ctx, i)
|
|
|
|
testSerializeMap(t, ctx, map[int8]interface{}{
|
|
0: "abcd",
|
|
1: uint32(12345678),
|
|
2: i,
|
|
3: 123,
|
|
})
|
|
|
|
testSerializeMap(t, ctx, map[int8]int32{
|
|
0: 1234,
|
|
2: 5678,
|
|
4: 9101,
|
|
6: 1121,
|
|
})
|
|
|
|
type test_struct struct {
|
|
Int int `gv:"int"`
|
|
String string `gv:"string"`
|
|
}
|
|
|
|
test_struct_type := reflect.TypeOf(test_struct{})
|
|
test_struct_info, err := GetStructInfo(ctx, test_struct_type)
|
|
fatalErr(t, err)
|
|
|
|
err = ctx.RegisterType(test_struct_type, NewSerializedType("TEST_STRUCT"), nil, SerializeStruct(test_struct_info), nil, DeserializeStruct(test_struct_info))
|
|
fatalErr(t, err)
|
|
|
|
testSerialize(t, ctx, test_struct{
|
|
12345,
|
|
"test_string",
|
|
})
|
|
|
|
testSerialize(t, ctx, Tree{
|
|
MapType: nil,
|
|
StringType: nil,
|
|
})
|
|
|
|
testSerialize(t, ctx, Tree{
|
|
TreeType: nil,
|
|
})
|
|
|
|
testSerialize(t, ctx, Tree{
|
|
TreeType: {
|
|
ErrorType: Tree{},
|
|
MapType: nil,
|
|
},
|
|
StringType: nil,
|
|
})
|
|
|
|
type test_slice []string
|
|
test_slice_type := reflect.TypeOf(test_slice{})
|
|
err = ctx.RegisterType(test_slice_type, NewSerializedType("TEST_SLICE"), SerializeTypeStub, SerializeSlice, DeserializeTypeStub[test_slice], DeserializeSlice)
|
|
fatalErr(t, err)
|
|
|
|
testSerialize[[]string](t, ctx, []string{"test_1", "test_2", "test_3"})
|
|
testSerialize[test_slice](t, ctx, test_slice{"test_1", "test_2", "test_3"})
|
|
}
|
|
|
|
type test struct {
|
|
Int int `gv:"int"`
|
|
Str string `gv:"string"`
|
|
}
|
|
|
|
func (s test) String() string {
|
|
return fmt.Sprintf("%d:%s", s.Int, s.Str)
|
|
}
|
|
|
|
func TestSerializeStructTags(t *testing.T) {
|
|
ctx := logTestContext(t, []string{"test"})
|
|
|
|
test_type := NewSerializedType("TEST_STRUCT")
|
|
test_struct_type := reflect.TypeOf(test{})
|
|
ctx.Log.Logf("test", "TEST_TYPE: %+v", test_type)
|
|
test_struct_info, err := GetStructInfo(ctx, test_struct_type)
|
|
fatalErr(t, err)
|
|
ctx.RegisterType(test_struct_type, test_type, nil, SerializeStruct(test_struct_info), nil, DeserializeStruct(test_struct_info))
|
|
|
|
test_int := 10
|
|
test_string := "test"
|
|
|
|
ret := testSerialize(t, ctx, test{
|
|
test_int,
|
|
test_string,
|
|
})
|
|
if ret.Int != test_int {
|
|
t.Fatalf("Deserialized int %d does not equal test %d", ret.Int, test_int)
|
|
} else if ret.Str != test_string {
|
|
t.Fatalf("Deserialized string %s does not equal test %s", ret.Str, test_string)
|
|
}
|
|
|
|
testSerialize(t, ctx, []test{
|
|
{
|
|
test_int,
|
|
test_string,
|
|
},
|
|
{
|
|
test_int * 2,
|
|
fmt.Sprintf("%s%s", test_string, test_string),
|
|
},
|
|
{
|
|
test_int * 4,
|
|
fmt.Sprintf("%s%s%s", test_string, test_string, test_string),
|
|
},
|
|
})
|
|
}
|
|
|
|
func testSerializeMap[M map[T]R, T, R comparable](t *testing.T, ctx *Context, val M) {
|
|
v := testSerialize(t, ctx, val)
|
|
for key, value := range(val) {
|
|
recreated, exists := v[key]
|
|
if exists == false {
|
|
t.Fatalf("DeserializeValue returned wrong value %+v != %+v", v, val)
|
|
} else if recreated != value {
|
|
t.Fatalf("DeserializeValue returned wrong value %+v != %+v", v, val)
|
|
}
|
|
}
|
|
if len(v) != len(val) {
|
|
t.Fatalf("DeserializeValue returned wrong value %+v != %+v", v, val)
|
|
}
|
|
}
|
|
|
|
func testSerializeSliceSlice[S [][]T, T comparable](t *testing.T, ctx *Context, val S) {
|
|
v := testSerialize(t, ctx, val)
|
|
for i, original := range(val) {
|
|
if (original == nil && v[i] != nil) || (original != nil && v[i] == nil) {
|
|
t.Fatalf("DeserializeValue returned wrong value %+v != %+v", v, val)
|
|
}
|
|
for j, o := range(original) {
|
|
if v[i][j] != o {
|
|
t.Fatalf("DeserializeValue returned wrong value %+v != %+v", v, val)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func testSerializeSlice[S []T, T comparable](t *testing.T, ctx *Context, val S) {
|
|
v := testSerialize(t, ctx, val)
|
|
for i, original := range(val) {
|
|
if v[i] != original {
|
|
t.Fatal(fmt.Sprintf("DeserializeValue returned wrong value %+v != %+v", v, val))
|
|
}
|
|
}
|
|
}
|
|
|
|
func testSerializeComparable[T comparable](t *testing.T, ctx *Context, val T) {
|
|
v := testSerialize(t, ctx, val)
|
|
if v != val {
|
|
t.Fatal(fmt.Sprintf("DeserializeValue returned wrong value %+v != %+v", v, val))
|
|
}
|
|
}
|
|
|
|
func testSerialize[T any](t *testing.T, ctx *Context, val T) T {
|
|
value := reflect.ValueOf(&val).Elem()
|
|
type_stack, err := SerializeType(ctx, value.Type())
|
|
chunks, err := SerializeValue(ctx, value)
|
|
value_serialized := SerializedValue{type_stack, chunks.Slice()}
|
|
fatalErr(t, err)
|
|
ctx.Log.Logf("test", "Serialized %+v to %+v(%d)", val, value_serialized, len(value_serialized.Data))
|
|
|
|
value_chunks, err := value_serialized.Chunks()
|
|
fatalErr(t, err)
|
|
ctx.Log.Logf("test", "Binary: %+v", value_chunks.Slice())
|
|
|
|
val_parsed, remaining_parse, err := ParseSerializedValue(value_chunks.Slice())
|
|
fatalErr(t, err)
|
|
ctx.Log.Logf("test", "Parsed: %+v", val_parsed)
|
|
|
|
if len(remaining_parse) != 0 {
|
|
t.Fatal("Data remaining after deserializing value")
|
|
}
|
|
|
|
val_type, remaining_types, err := DeserializeType(ctx, val_parsed.TypeStack)
|
|
deserialized_value, remaining_deserialize, err := DeserializeValue(ctx, val_type, val_parsed.Data)
|
|
fatalErr(t, err)
|
|
|
|
if len(remaining_deserialize) != 0 {
|
|
t.Fatal("Data remaining after deserializing value")
|
|
} else if len(remaining_types) != 0 {
|
|
t.Fatal("TypeStack remaining after deserializing value")
|
|
} else if val_type != value.Type() {
|
|
t.Fatal(fmt.Sprintf("DeserializeValue returned wrong reflect.Type %+v - %+v", val_type, reflect.TypeOf(val)))
|
|
} else if deserialized_value.CanConvert(val_type) == false {
|
|
t.Fatal("DeserializeValue returned value that can't convert to original value")
|
|
}
|
|
ctx.Log.Logf("test", "Value: %+v", deserialized_value.Interface())
|
|
if val_type.Kind() == reflect.Interface && deserialized_value.Interface() == nil {
|
|
var zero T
|
|
return zero
|
|
}
|
|
return deserialized_value.Interface().(T)
|
|
}
|