Function TestAddExtension

lib/encoding/messagepack_test.go:179–213

(t *testing.T)

}

func TestAddExtension(t *testing.T) {

	type CustomType struct {
		Name string
	}

	// global extention info.
	const extensionID byte = 1

	encoder := func(rv reflect.Value) ([]byte, error) {
		ct := rv.Interface().(CustomType)
		return []byte(ct.Name), nil
	}
	decoder := func(rv reflect.Value, b []byte) error {
		rv.Set(reflect.ValueOf(CustomType{Name: string(b)}))
		return nil
	}

	m := NewMessagePackEncoder()
	err := m.AddExtension(reflect.TypeOf(CustomType{}), extensionID, encoder, decoder)
	if err != nil {
		t.Fatalf("Failed adding extension: %v", err)
	}
	data := CustomType{Name: "test"}
	dataVerify := CustomType{}
	err = m.enc.Encode(&data)
	assert.NoError(t, err)
	assert.NotNil(t, m.Bytes())
	d := NewMessagePackDecoder(m.Bytes())
	err = d.AddExtension(reflect.TypeOf(CustomType{}), extensionID, encoder, decoder)
	assert.NoError(t, err)
	err = d.Decode(&dataVerify)
	assert.NoError(t, err)
	assert.EqualValues(t, dataVerify, data)
}