tmp

Author: bigbes

arrow/arrow.go

@@ -28,10 +28,12 @@ func (a Arrow) Raw() []byte {
 	return a.data
 }
 
+// MarshalMsgpack implements a custom msgpack marshaler for extension type.
 func (a Arrow) MarshalMsgpack() ([]byte, error) {
 	return a.data, nil
 }
 
+// UnmarshalMsgpack implements a custom msgpack unmarshaler for extension type.
 func (a *Arrow) UnmarshalMsgpack(data []byte) error {
 	a.data = data
 	return nil

arrow/arrow_gen_test.go

@@ -0,0 +1,116 @@
+package arrow
+
+import (
+	"bytes"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+	"github.com/vmihailenco/msgpack/v5"
+)
+
+func TestSomeOptionalArrow(t *testing.T) {
+	val, err := MakeArrow([]byte{1, 2, 3})
+	assert.NoError(t, err)
+	opt := SomeOptionalArrow(val)
+
+	assert.True(t, opt.IsSome())
+	assert.False(t, opt.IsZero())
+
+	v, ok := opt.Get()
+	assert.True(t, ok)
+	assert.Equal(t, val, v)
+}
+
+func TestNoneOptionalArrow(t *testing.T) {
+	opt := NoneOptionalArrow()
+
+	assert.False(t, opt.IsSome())
+	assert.True(t, opt.IsZero())
+
+	_, ok := opt.Get()
+	assert.False(t, ok)
+}
+
+func TestOptionalArrow_MustGet(t *testing.T) {
+	val, err := MakeArrow([]byte{1, 2, 3})
+	assert.NoError(t, err)
+	optSome := SomeOptionalArrow(val)
+	optNone := NoneOptionalArrow()
+
+	assert.Equal(t, val, optSome.MustGet())
+	assert.Panics(t, func() { optNone.MustGet() })
+}
+
+func TestOptionalArrow_Unwrap(t *testing.T) {
+	val, err := MakeArrow([]byte{1, 2, 3})
+	assert.NoError(t, err)
+	optSome := SomeOptionalArrow(val)
+	optNone := NoneOptionalArrow()
+
+	assert.Equal(t, val, optSome.Unwrap())
+	assert.Equal(t, Arrow{}, optNone.Unwrap())
+}
+
+func TestOptionalArrow_UnwrapOr(t *testing.T) {
+	val, err := MakeArrow([]byte{1, 2, 3})
+	assert.NoError(t, err)
+	def, err := MakeArrow([]byte{4, 5, 6})
+	assert.NoError(t, err)
+	optSome := SomeOptionalArrow(val)
+	optNone := NoneOptionalArrow()
+
+	assert.Equal(t, val, optSome.UnwrapOr(def))
+	assert.Equal(t, def, optNone.UnwrapOr(def))
+}
+
+func TestOptionalArrow_UnwrapOrElse(t *testing.T) {
+	val, err := MakeArrow([]byte{1, 2, 3})
+	assert.NoError(t, err)
+	def, err := MakeArrow([]byte{4, 5, 6})
+	assert.NoError(t, err)
+	optSome := SomeOptionalArrow(val)
+	optNone := NoneOptionalArrow()
+
+	assert.Equal(t, val, optSome.UnwrapOrElse(func() Arrow { return def }))
+	assert.Equal(t, def, optNone.UnwrapOrElse(func() Arrow { return def }))
+}
+
+func TestOptionalArrow_EncodeDecodeMsgpack(t *testing.T) {
+	val, err := MakeArrow([]byte{1, 2, 3})
+	assert.NoError(t, err)
+	some := SomeOptionalArrow(val)
+	none := NoneOptionalArrow()
+
+	var buf bytes.Buffer
+	enc := msgpack.NewEncoder(&buf)
+	dec := msgpack.NewDecoder(&buf)
+
+	// Test Some
+	err = enc.Encode(some)
+	assert.NoError(t, err)
+
+	var decodedSome OptionalArrow
+	err = dec.Decode(&decodedSome)
+	assert.NoError(t, err)
+	assert.True(t, decodedSome.IsSome())
+	assert.Equal(t, val, decodedSome.Unwrap())
+
+	// Test None
+	buf.Reset()
+	err = enc.Encode(none)
+	assert.NoError(t, err)
+
+	var decodedNone OptionalArrow
+	err = dec.Decode(&decodedNone)
+	assert.NoError(t, err)
+	assert.True(t, decodedNone.IsZero())
+
+	// Test invalid type
+	buf.Reset()
+	err = enc.Encode(123)
+	assert.NoError(t, err)
+
+	var decodedInvalid OptionalArrow
+	err = dec.Decode(&decodedInvalid)
+	assert.Error(t, err)
+}

box_error.go renamed to boxerror.go

@@ -9,6 +9,8 @@ import (
 
 const errorExtID = 3
 
+//go:generate go tool gentypes -ext-code 3 BoxError
+
 const (
 	keyErrorStack   = 0x00
 	keyErrorType    = 0x00