96 lines
2.1 KiB
Go
96 lines
2.1 KiB
Go
package state_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/innerring/processors/netmap/nodevalidation/state"
|
|
"git.frostfs.info/TrueCloudLab/frostfs-sdk-go/netmap"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
// implements state.NetworkSettings for testing.
|
|
type testNetworkSettings struct {
|
|
disallowed bool
|
|
}
|
|
|
|
func (x testNetworkSettings) MaintenanceModeAllowed() error {
|
|
if x.disallowed {
|
|
return state.ErrMaintenanceModeDisallowed
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func TestValidator_VerifyAndUpdate(t *testing.T) {
|
|
var vDefault state.NetMapCandidateValidator
|
|
var s testNetworkSettings
|
|
|
|
vDefault.SetNetworkSettings(s)
|
|
|
|
for _, testCase := range []struct {
|
|
name string
|
|
preparer func(*netmap.NodeInfo) // modifies zero instance
|
|
valid bool // is node valid after preparation
|
|
|
|
validatorPreparer func(*state.NetMapCandidateValidator) // optionally modifies default validator
|
|
}{
|
|
{
|
|
name: "UNDEFINED",
|
|
preparer: func(info *netmap.NodeInfo) {},
|
|
valid: false,
|
|
},
|
|
{
|
|
name: "ONLINE",
|
|
preparer: (*netmap.NodeInfo).SetOnline,
|
|
valid: true,
|
|
},
|
|
{
|
|
name: "OFFLINE",
|
|
preparer: (*netmap.NodeInfo).SetOffline,
|
|
valid: false,
|
|
},
|
|
{
|
|
name: "MAINTENANCE/allowed",
|
|
preparer: (*netmap.NodeInfo).SetMaintenance,
|
|
valid: true,
|
|
},
|
|
{
|
|
name: "MAINTENANCE/disallowed",
|
|
preparer: (*netmap.NodeInfo).SetMaintenance,
|
|
valid: false,
|
|
validatorPreparer: func(v *state.NetMapCandidateValidator) {
|
|
var s testNetworkSettings
|
|
s.disallowed = true
|
|
|
|
v.SetNetworkSettings(s)
|
|
},
|
|
},
|
|
} {
|
|
var node netmap.NodeInfo
|
|
|
|
// prepare node
|
|
testCase.preparer(&node)
|
|
|
|
// save binary representation for mutation check
|
|
binNode := node.Marshal()
|
|
|
|
var v state.NetMapCandidateValidator
|
|
if testCase.validatorPreparer == nil {
|
|
v = vDefault
|
|
} else {
|
|
testCase.validatorPreparer(&v)
|
|
}
|
|
|
|
err := v.VerifyAndUpdate(&node)
|
|
|
|
if testCase.valid {
|
|
require.NoError(t, err, testCase.name)
|
|
} else {
|
|
require.Error(t, err, testCase.name)
|
|
}
|
|
|
|
// check mutation
|
|
require.Equal(t, binNode, node.Marshal(), testCase.name)
|
|
}
|
|
}
|