forked from quic-go/quic-go
72c6180ce5
* fix race condition in version negotiation tests * versionnegotiation: migrate tests away from Ginkgo
52 lines
1.4 KiB
Go
52 lines
1.4 KiB
Go
package versionnegotiation
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/quic-go/quic-go"
|
|
quicproxy "github.com/quic-go/quic-go/integrationtests/tools/proxy"
|
|
"github.com/quic-go/quic-go/internal/protocol"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
const rtt = 400 * time.Millisecond
|
|
|
|
func expectDurationInRTTs(t *testing.T, startTime time.Time, num int) {
|
|
t.Helper()
|
|
testDuration := time.Since(startTime)
|
|
rtts := float32(testDuration) / float32(rtt)
|
|
require.GreaterOrEqual(t, rtts, float32(num))
|
|
require.Less(t, rtts, float32(num+1))
|
|
}
|
|
|
|
func TestVersionNegotiationFailure(t *testing.T) {
|
|
if len(protocol.SupportedVersions) == 1 {
|
|
t.Fatal("Test requires at least 2 supported versions.")
|
|
}
|
|
|
|
serverConfig := &quic.Config{}
|
|
serverConfig.Versions = protocol.SupportedVersions[:1]
|
|
ln, err := quic.ListenAddr("localhost:0", getTLSConfig(), serverConfig)
|
|
require.NoError(t, err)
|
|
defer ln.Close()
|
|
|
|
// start the proxy
|
|
proxy, err := quicproxy.NewQuicProxy("localhost:0", &quicproxy.Opts{
|
|
RemoteAddr: ln.Addr().String(),
|
|
DelayPacket: func(_ quicproxy.Direction, _ []byte) time.Duration { return rtt / 2 },
|
|
})
|
|
require.NoError(t, err)
|
|
|
|
startTime := time.Now()
|
|
_, err = quic.DialAddr(
|
|
context.Background(),
|
|
proxy.LocalAddr().String(),
|
|
getTLSClientConfig(),
|
|
maybeAddQLOGTracer(&quic.Config{Versions: protocol.SupportedVersions[1:2]}),
|
|
)
|
|
require.Error(t, err)
|
|
expectDurationInRTTs(t, startTime, 1)
|
|
}
|