[#302] pkg/netmap: Convert nil Replica to nil message

Document that `Replica.ToV2` method return `nil`
when called on `nil`. Document that `ReplicaFromV2`
function return `nil` when called on `nil`. Write
corresponding unit tests.

Signed-off-by: Pavel Karpy <carpawell@nspcc.ru>
This commit is contained in:
Pavel Karpy 2021-06-08 18:26:20 +03:00 committed by Alex Vanin
parent 6fb7c79219
commit fb591f5fac
2 changed files with 24 additions and 5 deletions

View file

@ -13,11 +13,15 @@ func NewReplica() *Replica {
} }
// NewReplicaFromV2 converts v2 Replica to Replica. // NewReplicaFromV2 converts v2 Replica to Replica.
//
// Nil netmap.Replica converts to nil.
func NewReplicaFromV2(f *netmap.Replica) *Replica { func NewReplicaFromV2(f *netmap.Replica) *Replica {
return (*Replica)(f) return (*Replica)(f)
} }
// ToV2 converts Replica to v2 Replica. // ToV2 converts Replica to v2 Replica.
//
// Nil Replica converts to nil.
func (r *Replica) ToV2() *netmap.Replica { func (r *Replica) ToV2() *netmap.Replica {
return (*netmap.Replica)(r) return (*netmap.Replica)(r)
} }

View file

@ -4,6 +4,7 @@ import (
"testing" "testing"
"github.com/nspcc-dev/neofs-api-go/v2/netmap" "github.com/nspcc-dev/neofs-api-go/v2/netmap"
testv2 "github.com/nspcc-dev/neofs-api-go/v2/netmap/test"
"github.com/stretchr/testify/require" "github.com/stretchr/testify/require"
) )
@ -16,13 +17,19 @@ func testReplica() *Replica {
} }
func TestReplicaFromV2(t *testing.T) { func TestReplicaFromV2(t *testing.T) {
rV2 := new(netmap.Replica) t.Run("from nil", func(t *testing.T) {
rV2.SetCount(3) var x *netmap.Replica
rV2.SetSelector("selector")
r := NewReplicaFromV2(rV2) require.Nil(t, NewReplicaFromV2(x))
})
require.Equal(t, rV2, r.ToV2()) t.Run("from non-nil", func(t *testing.T) {
rV2 := testv2.GenerateReplica(false)
r := NewReplicaFromV2(rV2)
require.Equal(t, rV2, r.ToV2())
})
} }
func TestReplica_Count(t *testing.T) { func TestReplica_Count(t *testing.T) {
@ -66,3 +73,11 @@ func TestReplicaEncoding(t *testing.T) {
require.Equal(t, r, r2) require.Equal(t, r, r2)
}) })
} }
func TestReplica_ToV2(t *testing.T) {
t.Run("nil", func(t *testing.T) {
var x *Replica
require.Nil(t, x.ToV2())
})
}