gf/encoding/gbinary/gbinary_z_test.go
2020-03-20 08:56:17 +08:00

138 lines
3.6 KiB
Go

// Copyright 2017 gf Author(https://github.com/gogf/gf). All Rights Reserved.
//
// This Source Code Form is subject to the terms of the MIT License.
// If a copy of the MIT was not distributed with this file,
// You can obtain one at https://github.com/gogf/gf.
package gbinary_test
import (
"math"
"testing"
"github.com/gogf/gf/encoding/gbinary"
"github.com/gogf/gf/test/gtest"
)
type User struct {
Name string
Age int
Url string
}
var testData = map[string]interface{}{
//"nil": nil,
"int": int(123),
"int8": int8(-99),
"int8.max": math.MaxInt8,
"int16": int16(123),
"int16.max": math.MaxInt16,
"int32": int32(-199),
"int32.max": math.MaxInt32,
"int64": int64(123),
"uint": uint(123),
"uint8": uint8(123),
"uint8.max": math.MaxUint8,
"uint16": uint16(9999),
"uint16.max": math.MaxUint16,
"uint32": uint32(123),
"uint64": uint64(123),
"bool.true": true,
"bool.false": false,
"string": "hehe haha",
"byte": []byte("hehe haha"),
"float32": float32(123.456),
"float32.max": math.MaxFloat32,
"float64": float64(123.456),
}
var testBitData = []int{0, 99, 122, 129, 222, 999, 22322}
func Test_EncodeAndDecode(t *testing.T) {
gtest.C(t, func(t *gtest.T) {
for k, v := range testData {
ve := gbinary.Encode(v)
ve1 := gbinary.EncodeByLength(len(ve), v)
//t.Logf("%s:%v, encoded:%v\n", k, v, ve)
switch v.(type) {
case int:
t.Assert(gbinary.DecodeToInt(ve), v)
t.Assert(gbinary.DecodeToInt(ve1), v)
case int8:
t.Assert(gbinary.DecodeToInt8(ve), v)
t.Assert(gbinary.DecodeToInt8(ve1), v)
case int16:
t.Assert(gbinary.DecodeToInt16(ve), v)
t.Assert(gbinary.DecodeToInt16(ve1), v)
case int32:
t.Assert(gbinary.DecodeToInt32(ve), v)
t.Assert(gbinary.DecodeToInt32(ve1), v)
case int64:
t.Assert(gbinary.DecodeToInt64(ve), v)
t.Assert(gbinary.DecodeToInt64(ve1), v)
case uint:
t.Assert(gbinary.DecodeToUint(ve), v)
t.Assert(gbinary.DecodeToUint(ve1), v)
case uint8:
t.Assert(gbinary.DecodeToUint8(ve), v)
t.Assert(gbinary.DecodeToUint8(ve1), v)
case uint16:
t.Assert(gbinary.DecodeToUint16(ve1), v)
t.Assert(gbinary.DecodeToUint16(ve), v)
case uint32:
t.Assert(gbinary.DecodeToUint32(ve1), v)
t.Assert(gbinary.DecodeToUint32(ve), v)
case uint64:
t.Assert(gbinary.DecodeToUint64(ve), v)
t.Assert(gbinary.DecodeToUint64(ve1), v)
case bool:
t.Assert(gbinary.DecodeToBool(ve), v)
t.Assert(gbinary.DecodeToBool(ve1), v)
case string:
t.Assert(gbinary.DecodeToString(ve), v)
t.Assert(gbinary.DecodeToString(ve1), v)
case float32:
t.Assert(gbinary.DecodeToFloat32(ve), v)
t.Assert(gbinary.DecodeToFloat32(ve1), v)
case float64:
t.Assert(gbinary.DecodeToFloat64(ve), v)
t.Assert(gbinary.DecodeToFloat64(ve1), v)
default:
if v == nil {
continue
}
res := make([]byte, len(ve))
err := gbinary.Decode(ve, res)
if err != nil {
t.Errorf("test data: %s, %v, error:%v", k, v, err)
}
t.Assert(res, v)
}
}
})
}
func Test_EncodeStruct(t *testing.T) {
gtest.C(t, func(t *gtest.T) {
user := User{"wenzi1", 999, "www.baidu.com"}
ve := gbinary.Encode(user)
s := gbinary.DecodeToString(ve)
t.Assert(s, s)
})
}
func Test_Bits(t *testing.T) {
gtest.C(t, func(t *gtest.T) {
for i := range testBitData {
bits := make([]gbinary.Bit, 0)
res := gbinary.EncodeBits(bits, testBitData[i], 64)
t.Assert(gbinary.DecodeBits(res), testBitData[i])
t.Assert(gbinary.DecodeBitsToUint(res), uint(testBitData[i]))
t.Assert(gbinary.DecodeBytesToBits(gbinary.EncodeBitsToBytes(res)), res)
}
})
}