Append tests

planetscale · Nov 20, 2023 · c13c5eb · c13c5eb
1 parent 7894a28
commit c13c5eb
Show file tree

Hide file tree

Showing 4 changed files with 445 additions and 20 deletions.
diff --git a/testproto/pool/pool.pb.go b/testproto/pool/pool.pb.go
diff --git a/testproto/pool/pool.proto b/testproto/pool/pool.proto
@@ -3,8 +3,13 @@ option go_package = "testproto/pool";
 
 import "github.com/planetscale/vtprotobuf/vtproto/ext.proto";
 
+message OptionalMessage{
+  option (vtproto.mempool) = true;
+}
+
 message MemoryPoolExtension {
   option (vtproto.mempool) = true;
   string foo1 = 1;
   uint64 foo2 = 2;
+  optional OptionalMessage foo3 = 3;
 }
diff --git a/testproto/pool/pool_test.go b/testproto/pool/pool_test.go
@@ -1,6 +1,7 @@
 package pool
 
 import (
+	"google.golang.org/protobuf/proto"
 	"log"
 	"testing"
 
@@ -163,3 +164,24 @@ func Test_Pool_Oneof(t *testing.T) {
 	require.NoError(t, t8.UnmarshalVT(t4Bytes))
 	require.Equal(t, &t4, &t8)
 }
+
+func Test_Pool_Optional(t *testing.T) {
+	m := &MemoryPoolExtension{
+		Foo1: "foo1",
+		Foo2: 123,
+		Foo3: &OptionalMessage{},
+	}
+
+	mBytes, err := m.MarshalVT()
+	require.NoError(t, err)
+
+	mUnmarshal := &MemoryPoolExtension{}
+	err = proto.Unmarshal(mBytes, mUnmarshal)
+	require.NoError(t, err)
+
+	require.True(t, m.EqualVT(mUnmarshal))
+
+	m.ReturnToVTPool()
+	mFromPool := MemoryPoolExtensionFromVTPool()
+	require.True(t, mFromPool.EqualVT(&MemoryPoolExtension{}))
+}