github.com/gnolang/gno@v0.0.0-20240520182011-228e9d0192ce/tm2/pkg/bft/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  	rpctypes "github.com/gnolang/gno/tm2/pkg/bft/rpc/lib/types"
    10  	"github.com/gnolang/gno/tm2/pkg/log"
    11  	"github.com/gnolang/gno/tm2/pkg/p2p"
    12  	p2pcfg "github.com/gnolang/gno/tm2/pkg/p2p/config"
    13  )
    14  
    15  func TestUnsafeDialSeeds(t *testing.T) {
    16  	t.Parallel()
    17  
    18  	sw := p2p.MakeSwitch(p2pcfg.DefaultP2PConfig(), 1, "testing", "123.123.123",
    19  		func(n int, sw *p2p.Switch) *p2p.Switch { return sw })
    20  	err := sw.Start()
    21  	require.NoError(t, err)
    22  	defer sw.Stop()
    23  
    24  	logger = log.NewNoopLogger()
    25  	p2pPeers = sw
    26  
    27  	testCases := []struct {
    28  		seeds []string
    29  		isErr bool
    30  	}{
    31  		{[]string{}, true},
    32  		{[]string{"g1m6kmam774klwlh4dhmhaatd7al02m0h0jwnyc6@127.0.0.1:41198"}, false},
    33  		{[]string{"127.0.0.1:41198"}, true},
    34  	}
    35  
    36  	for _, tc := range testCases {
    37  		res, err := UnsafeDialSeeds(&rpctypes.Context{}, tc.seeds)
    38  		if tc.isErr {
    39  			assert.Error(t, err)
    40  		} else {
    41  			assert.NoError(t, err)
    42  			assert.NotNil(t, res)
    43  		}
    44  	}
    45  }
    46  
    47  func TestUnsafeDialPeers(t *testing.T) {
    48  	t.Parallel()
    49  
    50  	sw := p2p.MakeSwitch(p2pcfg.DefaultP2PConfig(), 1, "testing", "123.123.123",
    51  		func(n int, sw *p2p.Switch) *p2p.Switch { return sw })
    52  	err := sw.Start()
    53  	require.NoError(t, err)
    54  	defer sw.Stop()
    55  
    56  	logger = log.NewNoopLogger()
    57  	p2pPeers = sw
    58  
    59  	testCases := []struct {
    60  		peers []string
    61  		isErr bool
    62  	}{
    63  		{[]string{}, true},
    64  		{[]string{"g1m6kmam774klwlh4dhmhaatd7al02m0h0jwnyc6@127.0.0.1:41198"}, false},
    65  		{[]string{"127.0.0.1:41198"}, true},
    66  	}
    67  
    68  	for _, tc := range testCases {
    69  		res, err := UnsafeDialPeers(&rpctypes.Context{}, tc.peers, false)
    70  		if tc.isErr {
    71  			assert.Error(t, err)
    72  		} else {
    73  			assert.NoError(t, err)
    74  			assert.NotNil(t, res)
    75  		}
    76  	}
    77  }