github.com/okex/exchain@v1.8.0/libs/tendermint/rpc/core/net_test.go (about)

     1  package core
     2  
     3  import (
     4  	"testing"
     5  
     6  	"github.com/stretchr/testify/assert"
     7  	"github.com/stretchr/testify/require"
     8  
     9  	cfg "github.com/okex/exchain/libs/tendermint/config"
    10  	"github.com/okex/exchain/libs/tendermint/libs/log"
    11  	"github.com/okex/exchain/libs/tendermint/p2p"
    12  	rpctypes "github.com/okex/exchain/libs/tendermint/rpc/jsonrpc/types"
    13  )
    14  
    15  func TestUnsafeDialSeeds(t *testing.T) {
    16  	sw := p2p.MakeSwitch(cfg.DefaultP2PConfig(), 1, "testing", "123.123.123",
    17  		func(n int, sw *p2p.Switch) *p2p.Switch { return sw })
    18  	err := sw.Start()
    19  	require.NoError(t, err)
    20  	defer sw.Stop()
    21  
    22  	env.Logger = log.TestingLogger()
    23  	env.P2PPeers = sw
    24  
    25  	testCases := []struct {
    26  		seeds []string
    27  		isErr bool
    28  	}{
    29  		{[]string{}, true},
    30  		{[]string{"d51fb70907db1c6c2d5237e78379b25cf1a37ab4@127.0.0.1:41198"}, false},
    31  		{[]string{"127.0.0.1:41198"}, true},
    32  	}
    33  
    34  	for _, tc := range testCases {
    35  		res, err := UnsafeDialSeeds(&rpctypes.Context{}, tc.seeds)
    36  		if tc.isErr {
    37  			assert.Error(t, err)
    38  		} else {
    39  			assert.NoError(t, err)
    40  			assert.NotNil(t, res)
    41  		}
    42  	}
    43  }
    44  
    45  func TestUnsafeDialPeers(t *testing.T) {
    46  	sw := p2p.MakeSwitch(cfg.DefaultP2PConfig(), 1, "testing", "123.123.123",
    47  		func(n int, sw *p2p.Switch) *p2p.Switch { return sw })
    48  	err := sw.Start()
    49  	require.NoError(t, err)
    50  	defer sw.Stop()
    51  
    52  	env.Logger = log.TestingLogger()
    53  	env.P2PPeers = sw
    54  
    55  	testCases := []struct {
    56  		peers []string
    57  		isErr bool
    58  	}{
    59  		{[]string{}, true},
    60  		{[]string{"d51fb70907db1c6c2d5237e78379b25cf1a37ab4@127.0.0.1:41198"}, false},
    61  		{[]string{"127.0.0.1:41198"}, true},
    62  	}
    63  
    64  	for _, tc := range testCases {
    65  		res, err := UnsafeDialPeers(&rpctypes.Context{}, tc.peers, false)
    66  		if tc.isErr {
    67  			assert.Error(t, err)
    68  		} else {
    69  			assert.NoError(t, err)
    70  			assert.NotNil(t, res)
    71  		}
    72  	}
    73  }