Merge pull request #2503 from oasislabs/pro-wh/feature/xferwl

go staking: add isTransferPermitted tests

.changelog/2503.trivial.md

@@ -0,0 +1,3 @@
+go staking: add isTransferPermitted tests
+
+it was desired that we have tests for this transfer permissions engine

go/consensus/tendermint/apps/staking/transactions_test.go

@@ -0,0 +1,58 @@
+package staking
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/require"
+
+	"github.com/oasislabs/oasis-core/go/common/crypto/signature"
+	staking "github.com/oasislabs/oasis-core/go/staking/api"
+)
+
+func TestIsTransferPermitted(t *testing.T) {
+	for _, tt := range []struct {
+		msg       string
+		params    *staking.ConsensusParameters
+		fromID    signature.PublicKey
+		permitted bool
+	}{
+		{
+			"no disablement",
+			&staking.ConsensusParameters{},
+			signature.PublicKey{},
+			true,
+		},
+		{
+			"all disabled",
+			&staking.ConsensusParameters{
+				DisableTransfers: true,
+			},
+			signature.PublicKey{},
+			false,
+		},
+		{
+			"not whitelisted",
+			&staking.ConsensusParameters{
+				DisableTransfers: true,
+				UndisableTransfersFrom: map[signature.PublicKey]bool{
+					signature.PublicKey{1}: true,
+				},
+			},
+			signature.PublicKey{},
+			false,
+		},
+		{
+			"whitelisted",
+			&staking.ConsensusParameters{
+				DisableTransfers: true,
+				UndisableTransfersFrom: map[signature.PublicKey]bool{
+					signature.PublicKey{}: true,
+				},
+			},
+			signature.PublicKey{},
+			true,
+		},
+	} {
+		require.Equal(t, tt.permitted, isTransferPermitted(tt.params, tt.fromID), tt.msg)
+	}
+}