mirror of
https://github.com/refraction-networking/uquic.git
synced 2025-04-04 20:57:36 +03:00
9950b4c687
It is invalid to use a net.PacketConn in multiple Transports. However, the validation logic is causing pain when using wrapped net.PacketConns. It was introduce to guard against incorrect uses of the API when the Transport was introduced, but this is probably less relevant now than it was back then.
97 lines
2.2 KiB
Go
97 lines
2.2 KiB
Go
package quic
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"io"
|
|
"log"
|
|
"net"
|
|
"os"
|
|
"runtime/pprof"
|
|
"strconv"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
|
|
. "github.com/onsi/ginkgo/v2"
|
|
. "github.com/onsi/gomega"
|
|
"github.com/stretchr/testify/require"
|
|
"go.uber.org/mock/gomock"
|
|
)
|
|
|
|
func TestQuicGo(t *testing.T) {
|
|
RegisterFailHandler(Fail)
|
|
RunSpecs(t, "QUIC Suite")
|
|
}
|
|
|
|
var mockCtrl *gomock.Controller
|
|
|
|
var _ = BeforeEach(func() {
|
|
mockCtrl = gomock.NewController(GinkgoT())
|
|
})
|
|
|
|
var _ = BeforeSuite(func() {
|
|
log.SetOutput(io.Discard)
|
|
})
|
|
|
|
// in the tests for the stream deadlines we set a deadline
|
|
// and wait to make an assertion when Read / Write was unblocked
|
|
// on the CIs, the timing is a lot less precise, so scale every duration by this factor
|
|
func scaleDuration(t time.Duration) time.Duration {
|
|
scaleFactor := 1
|
|
if f, err := strconv.Atoi(os.Getenv("TIMESCALE_FACTOR")); err == nil { // parsing "" errors, so this works fine if the env is not set
|
|
scaleFactor = f
|
|
}
|
|
if scaleFactor == 0 {
|
|
panic("TIMESCALE_FACTOR is 0")
|
|
}
|
|
return time.Duration(scaleFactor) * t
|
|
}
|
|
|
|
func newUPDConnLocalhost(t testing.TB) *net.UDPConn {
|
|
t.Helper()
|
|
conn, err := net.ListenUDP("udp", &net.UDPAddr{IP: net.IPv4(127, 0, 0, 1), Port: 0})
|
|
require.NoError(t, err)
|
|
t.Cleanup(func() { conn.Close() })
|
|
return conn
|
|
}
|
|
|
|
func areConnsRunning() bool {
|
|
var b bytes.Buffer
|
|
pprof.Lookup("goroutine").WriteTo(&b, 1)
|
|
return strings.Contains(b.String(), "quic-go.(*connection).run")
|
|
}
|
|
|
|
func areServersRunning() bool {
|
|
var b bytes.Buffer
|
|
pprof.Lookup("goroutine").WriteTo(&b, 1)
|
|
return strings.Contains(b.String(), "quic-go.(*baseServer).run")
|
|
}
|
|
|
|
func areTransportsRunning() bool {
|
|
var b bytes.Buffer
|
|
pprof.Lookup("goroutine").WriteTo(&b, 1)
|
|
return strings.Contains(b.String(), "quic-go.(*Transport).listen")
|
|
}
|
|
|
|
var _ = AfterEach(func() {
|
|
mockCtrl.Finish()
|
|
Eventually(areServersRunning).Should(BeFalse())
|
|
Eventually(areTransportsRunning()).Should(BeFalse())
|
|
})
|
|
|
|
func TestMain(m *testing.M) {
|
|
status := m.Run()
|
|
if status != 0 {
|
|
os.Exit(status)
|
|
}
|
|
if areConnsRunning() {
|
|
fmt.Println("stray connection goroutines found")
|
|
os.Exit(1)
|
|
}
|
|
if areTransportsRunning() {
|
|
fmt.Println("stray transport goroutines found")
|
|
os.Exit(1)
|
|
}
|
|
os.Exit(status)
|
|
}
|