[#397] refs: Fix OIDs marshalling

Signed-off-by: Pavel Karpy <carpawell@nspcc.ru>
pull/1/head
Pavel Karpy 2022-04-25 21:24:19 +03:00 committed by LeL
parent 3d3283411c
commit 870c1ffc0a
2 changed files with 4 additions and 5 deletions

View File

@ -118,11 +118,11 @@ func (o *ObjectID) StableSize() int {
func ObjectIDNestedListMarshal(fNum int64, buf []byte, ids []ObjectID) (off int) {
prefix, _ := proto.NestedStructurePrefix(fNum)
for i := range ids {
off += binary.PutUvarint(buf, prefix)
off += binary.PutUvarint(buf[off:], prefix)
n := ids[i].StableSize()
off += binary.PutUvarint(buf[off:], uint64(n))
off += proto.BytesMarshal(objectIDValField, buf, ids[i].val)
off += proto.BytesMarshal(objectIDValField, buf[off:], ids[i].val)
}
return

View File

@ -16,7 +16,7 @@ type jsonMessage interface {
}
type binaryMessage interface {
StableMarshal([]byte) ([]byte, error)
StableMarshal([]byte) []byte
Unmarshal([]byte) error
}
@ -54,8 +54,7 @@ func TestRPCMessage(t *testing.T, msgGens ...func(empty bool) message.Message) {
if bm, ok := msg.(binaryMessage); ok {
t.Run(fmt.Sprintf("Binary_%T", msg), func(t *testing.T) {
data, err := bm.StableMarshal(nil)
require.NoError(t, err)
data := bm.StableMarshal(nil)
bm2 := msgGen(true).(binaryMessage)
require.NoError(t, bm2.Unmarshal(data))