forked from TrueCloudLab/frostfs-node
62 lines
1.2 KiB
Go
62 lines
1.2 KiB
Go
|
package transport
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"net"
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
manet "github.com/multiformats/go-multiaddr-net"
|
||
|
|
circuit "github.com/rubyist/circuitbreaker"
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestTransport(t *testing.T) {
|
||
|
|
var (
|
||
|
|
attempts = int64(5)
|
||
|
|
lc net.ListenConfig
|
||
|
|
tr = New(attempts, time.Second)
|
||
|
|
ctx, cancel = context.WithCancel(context.TODO())
|
||
|
|
)
|
||
|
|
|
||
|
|
defer cancel()
|
||
|
|
|
||
|
|
lis1, err := lc.Listen(ctx, "tcp", ":0")
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
addr1, err := manet.FromNetAddr(lis1.Addr())
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
_, err = tr.Dial(ctx, addr1, false)
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
lis2, err := lc.Listen(ctx, "tcp", ":0")
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
addr2, err := manet.FromNetAddr(lis2.Addr())
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
_, err = tr.Dial(ctx, addr1, false)
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
require.NoError(t, lis1.Close())
|
||
|
|
|
||
|
|
for i := int64(0); i < 10; i++ {
|
||
|
|
_, err = tr.Dial(ctx, addr1, false)
|
||
|
|
require.Error(t, err)
|
||
|
|
|
||
|
|
if i >= attempts {
|
||
|
|
require.EqualError(t, err, circuit.ErrBreakerOpen.Error())
|
||
|
|
}
|
||
|
|
|
||
|
|
_, err = tr.Dial(ctx, addr2, false)
|
||
|
|
require.NoError(t, err)
|
||
|
|
}
|
||
|
|
|
||
|
|
time.Sleep(time.Second)
|
||
|
|
|
||
|
|
_, err = tr.Dial(ctx, addr1, false)
|
||
|
|
require.Error(t, err)
|
||
|
|
require.NotContains(t, err.Error(), circuit.ErrBreakerOpen.Error())
|
||
|
|
}
|