forked from TrueCloudLab/frostfs-sdk-go
[#240] container: Support disabling homomorphic hashing
Signed-off-by: Pavel Karpy <carpawell@nspcc.ru>
This commit is contained in:
parent
0cd790cfe0
commit
9996b3be01
2 changed files with 53 additions and 0 deletions
20
container/network.go
Normal file
20
container/network.go
Normal file
|
@ -0,0 +1,20 @@
|
|||
package container
|
||||
|
||||
import (
|
||||
"github.com/nspcc-dev/neofs-sdk-go/netmap"
|
||||
)
|
||||
|
||||
// ApplyNetworkConfig applies network configuration to the
|
||||
// container. Changes the container if it does not satisfy
|
||||
// network configuration.
|
||||
func ApplyNetworkConfig(cnr *Container, cfg netmap.NetworkInfo) {
|
||||
if cfg.HomomorphicHashingDisabled() {
|
||||
DisableHomomorphicHashing(cnr)
|
||||
}
|
||||
}
|
||||
|
||||
// AssertNetworkConfig checks if a container matches passed
|
||||
// network configuration.
|
||||
func AssertNetworkConfig(cnr Container, cfg netmap.NetworkInfo) bool {
|
||||
return IsHomomorphicHashingDisabled(cnr) == cfg.HomomorphicHashingDisabled()
|
||||
}
|
33
container/network_test.go
Normal file
33
container/network_test.go
Normal file
|
@ -0,0 +1,33 @@
|
|||
package container_test
|
||||
|
||||
import (
|
||||
"testing"
|
||||
|
||||
"github.com/nspcc-dev/neofs-sdk-go/container"
|
||||
containertest "github.com/nspcc-dev/neofs-sdk-go/container/test"
|
||||
netmaptest "github.com/nspcc-dev/neofs-sdk-go/netmap/test"
|
||||
"github.com/stretchr/testify/require"
|
||||
)
|
||||
|
||||
func TestContainer_NetworkConfig(t *testing.T) {
|
||||
c := containertest.Container()
|
||||
nc := netmaptest.NetworkInfo()
|
||||
|
||||
t.Run("default", func(t *testing.T) {
|
||||
require.False(t, container.IsHomomorphicHashingDisabled(c))
|
||||
|
||||
res := container.AssertNetworkConfig(c, nc)
|
||||
|
||||
require.True(t, res)
|
||||
})
|
||||
|
||||
nc.DisableHomomorphicHashing()
|
||||
|
||||
t.Run("apply", func(t *testing.T) {
|
||||
require.False(t, container.IsHomomorphicHashingDisabled(c))
|
||||
|
||||
container.ApplyNetworkConfig(&c, nc)
|
||||
|
||||
require.True(t, container.IsHomomorphicHashingDisabled(c))
|
||||
})
|
||||
}
|
Loading…
Reference in a new issue