forked from TrueCloudLab/frostfs-api-go
[#164] sdk/container: Implement Equal method on ID
Signed-off-by: Leonard Lyubich <leonard@nspcc.ru>
This commit is contained in:
parent
1305bc750e
commit
cf70026c7e
3 changed files with 34 additions and 2 deletions
|
@ -1,6 +1,7 @@
|
||||||
package container
|
package container
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"bytes"
|
||||||
"crypto/sha256"
|
"crypto/sha256"
|
||||||
|
|
||||||
"github.com/nspcc-dev/neofs-api-go/v2/refs"
|
"github.com/nspcc-dev/neofs-api-go/v2/refs"
|
||||||
|
@ -30,3 +31,11 @@ func (id *ID) SetSHA256(v [sha256.Size]byte) {
|
||||||
func (id *ID) ToV2() *refs.ContainerID {
|
func (id *ID) ToV2() *refs.ContainerID {
|
||||||
return (*refs.ContainerID)(id)
|
return (*refs.ContainerID)(id)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Equal returns true if identifiers are identical.
|
||||||
|
func (id *ID) Equal(id2 *ID) bool {
|
||||||
|
return bytes.Equal(
|
||||||
|
(*refs.ContainerID)(id).GetValue(),
|
||||||
|
(*refs.ContainerID)(id2).GetValue(),
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
|
@ -22,3 +22,26 @@ func TestIDV2_0(t *testing.T) {
|
||||||
|
|
||||||
require.Equal(t, checksum[:], cidV2.GetValue())
|
require.Equal(t, checksum[:], cidV2.GetValue())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func randSHA256Checksum(t *testing.T) (cs [sha256.Size]byte) {
|
||||||
|
_, err := rand.Read(cs[:])
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestID_Equal(t *testing.T) {
|
||||||
|
cs := randSHA256Checksum(t)
|
||||||
|
|
||||||
|
id1 := NewID()
|
||||||
|
id1.SetSHA256(cs)
|
||||||
|
|
||||||
|
id2 := NewID()
|
||||||
|
id2.SetSHA256(cs)
|
||||||
|
|
||||||
|
id3 := NewID()
|
||||||
|
id3.SetSHA256(randSHA256Checksum(t))
|
||||||
|
|
||||||
|
require.True(t, id1.Equal(id2))
|
||||||
|
require.False(t, id1.Equal(id3))
|
||||||
|
}
|
||||||
|
|
|
@ -30,8 +30,8 @@ func (id *ID) SetSHA256(v [sha256.Size]byte) {
|
||||||
// Equal returns true if identifiers are identical.
|
// Equal returns true if identifiers are identical.
|
||||||
func (id *ID) Equal(id2 *ID) bool {
|
func (id *ID) Equal(id2 *ID) bool {
|
||||||
return bytes.Equal(
|
return bytes.Equal(
|
||||||
(*ID)(id).ToV2().GetValue(),
|
(*refs.ObjectID)(id).GetValue(),
|
||||||
(*ID)(id2).ToV2().GetValue(),
|
(*refs.ObjectID)(id2).GetValue(),
|
||||||
)
|
)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue