[#227] netmap: Add more encoding methods for PlacementPolicy
Signed-off-by: Leonard Lyubich <leonard@nspcc.ru>
This commit is contained in:
parent
e999fb00c3
commit
e986f47807
2 changed files with 66 additions and 2 deletions
|
@ -57,7 +57,48 @@ func (p *PlacementPolicy) readFromV2(m netmap.PlacementPolicy, checkFieldPresenc
|
|||
return nil
|
||||
}
|
||||
|
||||
// UnmarshalJSON decodes PlacementPolicy from protobuf JSON format.
|
||||
// Marshal encodes PlacementPolicy into a binary format of the NeoFS API
|
||||
// protocol (Protocol Buffers with direct field order).
|
||||
//
|
||||
// See also Unmarshal.
|
||||
func (p PlacementPolicy) Marshal() []byte {
|
||||
var m netmap.PlacementPolicy
|
||||
p.WriteToV2(&m)
|
||||
|
||||
return m.StableMarshal(nil)
|
||||
}
|
||||
|
||||
// Unmarshal decodes NeoFS API protocol binary format into the PlacementPolicy
|
||||
// (Protocol Buffers with direct field order). Returns an error describing
|
||||
// a format violation.
|
||||
//
|
||||
// See also Marshal.
|
||||
func (p *PlacementPolicy) Unmarshal(data []byte) error {
|
||||
var m netmap.PlacementPolicy
|
||||
|
||||
err := m.Unmarshal(data)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return p.readFromV2(m, false)
|
||||
}
|
||||
|
||||
// MarshalJSON encodes PlacementPolicy into a JSON format of the NeoFS API
|
||||
// protocol (Protocol Buffers JSON).
|
||||
//
|
||||
// See also UnmarshalJSON.
|
||||
func (p PlacementPolicy) MarshalJSON() ([]byte, error) {
|
||||
var m netmap.PlacementPolicy
|
||||
p.WriteToV2(&m)
|
||||
|
||||
return m.MarshalJSON()
|
||||
}
|
||||
|
||||
// UnmarshalJSON decodes NeoFS API protocol JSON format into the PlacementPolicy
|
||||
// (Protocol Buffers JSON). Returns an error describing a format violation.
|
||||
//
|
||||
// See also MarshalJSON.
|
||||
func (p *PlacementPolicy) UnmarshalJSON(data []byte) error {
|
||||
var m netmap.PlacementPolicy
|
||||
|
||||
|
|
|
@ -1,9 +1,11 @@
|
|||
package netmap
|
||||
package netmap_test
|
||||
|
||||
import (
|
||||
"strings"
|
||||
"testing"
|
||||
|
||||
. "github.com/nspcc-dev/neofs-sdk-go/netmap"
|
||||
netmaptest "github.com/nspcc-dev/neofs-sdk-go/netmap/test"
|
||||
"github.com/stretchr/testify/require"
|
||||
)
|
||||
|
||||
|
@ -34,3 +36,24 @@ FILTER City EQ SPB AND SSD EQ true OR City EQ SPB AND Rating GE 5 AS SPBSSD`,
|
|||
require.Equal(t, testCase, b.String())
|
||||
}
|
||||
}
|
||||
|
||||
func TestPlacementPolicyEncoding(t *testing.T) {
|
||||
v := netmaptest.PlacementPolicy()
|
||||
|
||||
t.Run("binary", func(t *testing.T) {
|
||||
var v2 PlacementPolicy
|
||||
require.NoError(t, v2.Unmarshal(v.Marshal()))
|
||||
|
||||
require.Equal(t, v, v2)
|
||||
})
|
||||
|
||||
t.Run("json", func(t *testing.T) {
|
||||
data, err := v.MarshalJSON()
|
||||
require.NoError(t, err)
|
||||
|
||||
var v2 PlacementPolicy
|
||||
require.NoError(t, v2.UnmarshalJSON(data))
|
||||
|
||||
require.Equal(t, v, v2)
|
||||
})
|
||||
}
|
||||
|
|
Loading…
Reference in a new issue