1f8fb17be7
This commit syncs ENT changes to the OSS repo.
Original commit details in ENT:
```
commit 569d25f7f4578981c3801e6e067295668210f748
Author: FFMMM <FFMMM@users.noreply.github.com>
Date: Thu Feb 10 10:23:33 2022 -0800
Vendor fork net rpc (#1538)
* replace net/rpc w consul-net-rpc/net/rpc
Signed-off-by: FFMMM <FFMMM@users.noreply.github.com>
* replace msgpackrpc and go-msgpack with fork from mono repo
Signed-off-by: FFMMM <FFMMM@users.noreply.github.com>
* gofmt all files touched
Signed-off-by: FFMMM <FFMMM@users.noreply.github.com>
```
Signed-off-by: FFMMM <FFMMM@users.noreply.github.com>
77 lines
2.2 KiB
Go
77 lines
2.2 KiB
Go
package structs
|
|
|
|
import (
|
|
"bytes"
|
|
"reflect"
|
|
"testing"
|
|
|
|
"github.com/hashicorp/consul-net-rpc/go-msgpack/codec"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
// TestingOldPre1dot7MsgpackHandle is the common configuration pre-1.7.0
|
|
var TestingOldPre1dot7MsgpackHandle = &codec.MsgpackHandle{}
|
|
|
|
// TestMsgpackEncodeDecode is a test helper to easily write a test to verify
|
|
// msgpack encoding and decoding using two handles is identical.
|
|
func TestMsgpackEncodeDecode(t *testing.T, in interface{}, requireEncoderEquality bool) {
|
|
t.Helper()
|
|
var (
|
|
// This is the common configuration pre-1.7.0
|
|
handle1 = TestingOldPre1dot7MsgpackHandle
|
|
// This is the common configuration post-1.7.0
|
|
handle2 = MsgpackHandle
|
|
)
|
|
|
|
// Verify the 3 interface{} args are all pointers to the same kind of type.
|
|
inType := reflect.TypeOf(in)
|
|
require.Equal(t, reflect.Ptr, inType.Kind())
|
|
|
|
// Encode using both handles.
|
|
var b1 []byte
|
|
{
|
|
var buf bytes.Buffer
|
|
enc := codec.NewEncoder(&buf, handle1)
|
|
require.NoError(t, enc.Encode(in))
|
|
b1 = buf.Bytes()
|
|
}
|
|
var b2 []byte
|
|
{
|
|
var buf bytes.Buffer
|
|
enc := codec.NewEncoder(&buf, handle2)
|
|
require.NoError(t, enc.Encode(in))
|
|
b2 = buf.Bytes()
|
|
}
|
|
|
|
if requireEncoderEquality {
|
|
// The resulting bytes should be identical.
|
|
require.Equal(t, b1, b2)
|
|
}
|
|
|
|
// Decode both outputs using both handles.
|
|
t.Run("old encoder and old decoder", func(t *testing.T) {
|
|
out1 := reflect.New(inType.Elem()).Interface()
|
|
dec := codec.NewDecoderBytes(b1, handle1)
|
|
require.NoError(t, dec.Decode(out1))
|
|
require.Equal(t, in, out1)
|
|
})
|
|
t.Run("old encoder and new decoder", func(t *testing.T) {
|
|
out1 := reflect.New(inType.Elem()).Interface()
|
|
dec := codec.NewDecoderBytes(b1, handle2)
|
|
require.NoError(t, dec.Decode(out1))
|
|
require.Equal(t, in, out1)
|
|
})
|
|
t.Run("new encoder and old decoder", func(t *testing.T) {
|
|
out2 := reflect.New(inType.Elem()).Interface()
|
|
dec := codec.NewDecoderBytes(b2, handle1)
|
|
require.NoError(t, dec.Decode(out2))
|
|
require.Equal(t, in, out2)
|
|
})
|
|
t.Run("new encoder and new decoder", func(t *testing.T) {
|
|
out2 := reflect.New(inType.Elem()).Interface()
|
|
dec := codec.NewDecoderBytes(b2, handle2)
|
|
require.NoError(t, dec.Decode(out2))
|
|
require.Equal(t, in, out2)
|
|
})
|
|
}
|