123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209 |
- // Copyright 2019 The go-ethereum Authors
- // This file is part of go-ethereum.
- //
- // go-ethereum is free software: you can redistribute it and/or modify
- // it under the terms of the GNU General Public License as published by
- // the Free Software Foundation, either version 3 of the License, or
- // (at your option) any later version.
- //
- // go-ethereum is distributed in the hope that it will be useful,
- // but WITHOUT ANY WARRANTY; without even the implied warranty of
- // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- // GNU General Public License for more details.
- //
- // You should have received a copy of the GNU General Public License
- // along with go-ethereum. If not, see <http://www.gnu.org/licenses/>.
- // Package utils contains internal helper functions for go-ethereum commands.
- package utils
- import (
- "flag"
- "io/ioutil"
- "os"
- "path"
- "reflect"
- "strconv"
- "testing"
- "time"
- "github.com/ethereum/go-ethereum/common"
- "github.com/ethereum/go-ethereum/eth"
- "github.com/ethereum/go-ethereum/node"
- "github.com/stretchr/testify/assert"
- "github.com/stretchr/testify/require"
- "gopkg.in/urfave/cli.v1"
- )
- func TestAuthorizationList(t *testing.T) {
- value := "1=" + common.HexToHash("0xfa").Hex() + ",2=" + common.HexToHash("0x12").Hex()
- result := map[uint64]common.Hash{
- 1: common.HexToHash("0xfa"),
- 2: common.HexToHash("0x12"),
- }
- arbitraryNodeConfig := ð.Config{}
- fs := &flag.FlagSet{}
- fs.String(AuthorizationListFlag.Name, value, "")
- arbitraryCLIContext := cli.NewContext(nil, fs, nil)
- arbitraryCLIContext.GlobalSet(AuthorizationListFlag.Name, value)
- setAuthorizationList(arbitraryCLIContext, arbitraryNodeConfig)
- assert.Equal(t, result, arbitraryNodeConfig.AuthorizationList)
- fs = &flag.FlagSet{}
- fs.String(AuthorizationListFlag.Name, value, "")
- arbitraryCLIContext = cli.NewContext(nil, fs, nil)
- arbitraryCLIContext.GlobalSet(DeprecatedAuthorizationListFlag.Name, value) // old wlist flag
- setAuthorizationList(arbitraryCLIContext, arbitraryNodeConfig)
- assert.Equal(t, result, arbitraryNodeConfig.AuthorizationList)
- }
- func TestSetPlugins_whenPluginsNotEnabled(t *testing.T) {
- arbitraryNodeConfig := &node.Config{}
- arbitraryCLIContext := cli.NewContext(nil, &flag.FlagSet{}, nil)
- assert.NoError(t, SetPlugins(arbitraryCLIContext, arbitraryNodeConfig))
- assert.Nil(t, arbitraryNodeConfig.Plugins)
- }
- func TestSetPlugins_whenInvalidFlagsCombination(t *testing.T) {
- arbitraryNodeConfig := &node.Config{}
- fs := &flag.FlagSet{}
- fs.String(PluginSettingsFlag.Name, "", "")
- fs.Bool(PluginSkipVerifyFlag.Name, true, "")
- fs.Bool(PluginLocalVerifyFlag.Name, true, "")
- fs.String(PluginPublicKeyFlag.Name, "", "")
- arbitraryCLIContext := cli.NewContext(nil, fs, nil)
- assert.NoError(t, arbitraryCLIContext.GlobalSet(PluginSettingsFlag.Name, "arbitrary value"))
- verifyErrorMessage(t, arbitraryCLIContext, arbitraryNodeConfig, "only --plugins.skipverify or --plugins.localverify must be set")
- assert.NoError(t, arbitraryCLIContext.GlobalSet(PluginSkipVerifyFlag.Name, "false"))
- assert.NoError(t, arbitraryCLIContext.GlobalSet(PluginLocalVerifyFlag.Name, "false"))
- assert.NoError(t, arbitraryCLIContext.GlobalSet(PluginPublicKeyFlag.Name, "arbitrary value"))
- verifyErrorMessage(t, arbitraryCLIContext, arbitraryNodeConfig, "--plugins.localverify is required for setting --plugins.publickey")
- }
- func TestSetPlugins_whenInvalidPluginSettingsURL(t *testing.T) {
- arbitraryNodeConfig := &node.Config{}
- fs := &flag.FlagSet{}
- fs.String(PluginSettingsFlag.Name, "", "")
- arbitraryCLIContext := cli.NewContext(nil, fs, nil)
- assert.NoError(t, arbitraryCLIContext.GlobalSet(PluginSettingsFlag.Name, "arbitrary value"))
- verifyErrorMessage(t, arbitraryCLIContext, arbitraryNodeConfig, "plugins: unable to create reader due to unsupported scheme ")
- }
- func TestSetImmutabilityThreshold(t *testing.T) {
- fs := &flag.FlagSet{}
- fs.Int(QuorumImmutabilityThreshold.Name, 0, "")
- arbitraryCLIContext := cli.NewContext(nil, fs, nil)
- assert.NoError(t, arbitraryCLIContext.GlobalSet(QuorumImmutabilityThreshold.Name, strconv.Itoa(100000)))
- assert.True(t, arbitraryCLIContext.GlobalIsSet(QuorumImmutabilityThreshold.Name), "immutability threshold flag not set")
- assert.Equal(t, 100000, arbitraryCLIContext.GlobalInt(QuorumImmutabilityThreshold.Name), "immutability threshold value not set")
- }
- func TestSetPlugins_whenTypical(t *testing.T) {
- tmpDir, err := ioutil.TempDir("", "q-")
- if err != nil {
- t.Fatal(err)
- }
- defer func() {
- _ = os.RemoveAll(tmpDir)
- }()
- arbitraryJSONFile := path.Join(tmpDir, "arbitary.json")
- if err := ioutil.WriteFile(arbitraryJSONFile, []byte("{}"), 0644); err != nil {
- t.Fatal(err)
- }
- arbitraryNodeConfig := &node.Config{}
- fs := &flag.FlagSet{}
- fs.String(PluginSettingsFlag.Name, "", "")
- arbitraryCLIContext := cli.NewContext(nil, fs, nil)
- assert.NoError(t, arbitraryCLIContext.GlobalSet(PluginSettingsFlag.Name, "file://"+arbitraryJSONFile))
- assert.NoError(t, SetPlugins(arbitraryCLIContext, arbitraryNodeConfig))
- assert.NotNil(t, arbitraryNodeConfig.Plugins)
- }
- func verifyErrorMessage(t *testing.T, ctx *cli.Context, cfg *node.Config, expectedMsg string) {
- err := SetPlugins(ctx, cfg)
- assert.EqualError(t, err, expectedMsg)
- }
- func Test_SplitTagsFlag(t *testing.T) {
- tests := []struct {
- name string
- args string
- want map[string]string
- }{
- {
- "2 tags case",
- "host=localhost,bzzkey=123",
- map[string]string{
- "host": "localhost",
- "bzzkey": "123",
- },
- },
- {
- "1 tag case",
- "host=localhost123",
- map[string]string{
- "host": "localhost123",
- },
- },
- {
- "empty case",
- "",
- map[string]string{},
- },
- {
- "garbage",
- "smth=smthelse=123",
- map[string]string{},
- },
- }
- for _, tt := range tests {
- t.Run(tt.name, func(t *testing.T) {
- if got := SplitTagsFlag(tt.args); !reflect.DeepEqual(got, tt.want) {
- t.Errorf("splitTagsFlag() = %v, want %v", got, tt.want)
- }
- })
- }
- }
- func TestQuorumConfigFlags(t *testing.T) {
- fs := &flag.FlagSet{}
- arbitraryCLIContext := cli.NewContext(nil, fs, nil)
- arbitraryEthConfig := ð.Config{}
- fs.Int(EVMCallTimeOutFlag.Name, 0, "")
- assert.NoError(t, arbitraryCLIContext.GlobalSet(EVMCallTimeOutFlag.Name, strconv.Itoa(12)))
- fs.Bool(MultitenancyFlag.Name, false, "")
- assert.NoError(t, arbitraryCLIContext.GlobalSet(MultitenancyFlag.Name, "true"))
- fs.Bool(QuorumEnablePrivacyMarker.Name, true, "")
- assert.NoError(t, arbitraryCLIContext.GlobalSet(QuorumEnablePrivacyMarker.Name, "true"))
- fs.Uint64(IstanbulRequestTimeoutFlag.Name, 0, "")
- assert.NoError(t, arbitraryCLIContext.GlobalSet(IstanbulRequestTimeoutFlag.Name, "23"))
- fs.Uint64(IstanbulBlockPeriodFlag.Name, 0, "")
- assert.NoError(t, arbitraryCLIContext.GlobalSet(IstanbulBlockPeriodFlag.Name, "34"))
- fs.Bool(RaftModeFlag.Name, false, "")
- assert.NoError(t, arbitraryCLIContext.GlobalSet(RaftModeFlag.Name, "true"))
- require.NoError(t, setQuorumConfig(arbitraryCLIContext, arbitraryEthConfig))
- assert.True(t, arbitraryCLIContext.GlobalIsSet(EVMCallTimeOutFlag.Name), "EVMCallTimeOutFlag not set")
- assert.True(t, arbitraryCLIContext.GlobalIsSet(MultitenancyFlag.Name), "MultitenancyFlag not set")
- assert.True(t, arbitraryCLIContext.GlobalIsSet(RaftModeFlag.Name), "RaftModeFlag not set")
- assert.Equal(t, 12*time.Second, arbitraryEthConfig.EVMCallTimeOut, "EVMCallTimeOut value is incorrect")
- assert.Equal(t, true, arbitraryEthConfig.QuorumChainConfig.MultiTenantEnabled(), "MultitenancyFlag value is incorrect")
- assert.Equal(t, true, arbitraryEthConfig.QuorumChainConfig.PrivacyMarkerEnabled(), "QuorumEnablePrivacyMarker value is incorrect")
- config := arbitraryEthConfig.Istanbul.GetConfig(nil)
- assert.Equal(t, uint64(23), config.RequestTimeout, "IstanbulRequestTimeoutFlag value is incorrect")
- assert.Equal(t, uint64(34), config.BlockPeriod, "IstanbulBlockPeriodFlag value is incorrect")
- assert.Equal(t, uint64(0), config.EmptyBlockPeriod, "IstanbulEmptyBlockPeriodFlag value is incorrect")
- assert.Equal(t, true, arbitraryEthConfig.RaftMode, "RaftModeFlag value is incorrect")
- }
|