forked from TrueCloudLab/frostfs-api-go
[#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:
parent
6fb7c79219
commit
fb591f5fac
2 changed files with 24 additions and 5 deletions
|
@ -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)
|
||||||
}
|
}
|
||||||
|
|
|
@ -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())
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
Loading…
Reference in a new issue