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