homelab/traefik
1
0
Fork 0
Code Activity

Feature: add udp timeout configuration

Browse source
This commit is contained in:
Linden Krouse 2021-01-07 11:16:03 -05:00 • committed by GitHub
parent e5a01c7cc8
commit fc7ec17905
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
13 changed files with 98 additions and 22 deletions
Download patch file Download diff file Expand all files Collapse all files

18
pkg/udp/conn_test.go
View file

@ -15,7 +15,7 @@ func TestConsecutiveWrites(t *testing.T) {
addr, err := net.ResolveUDPAddr("udp", ":0")
require.NoError(t, err)
ln, err := Listen("udp", addr)
ln, err := Listen("udp", addr, 3*time.Second)
require.NoError(t, err)
defer func() {
err := ln.Close()
@ -77,7 +77,7 @@ func TestListenNotBlocking(t *testing.T) {
require.NoError(t, err)
ln, err := Listen("udp", addr)
ln, err := Listen("udp", addr, 3*time.Second)
require.NoError(t, err)
defer func() {
err := ln.Close()
@ -162,6 +162,14 @@ func TestListenNotBlocking(t *testing.T) {
}
}
func TestListenWithZeroTimeout(t *testing.T) {
addr, err := net.ResolveUDPAddr("udp", ":0")
require.NoError(t, err)
_, err = Listen("udp", addr, 0)
assert.Error(t, err)
}
func TestTimeoutWithRead(t *testing.T) {
testTimeout(t, true)
}
@ -176,7 +184,7 @@ func testTimeout(t *testing.T, withRead bool) {
addr, err := net.ResolveUDPAddr("udp", ":0")
require.NoError(t, err)
ln, err := Listen("udp", addr)
ln, err := Listen("udp", addr, 3*time.Second)
require.NoError(t, err)
defer func() {
err := ln.Close()
@ -212,7 +220,7 @@ func testTimeout(t *testing.T, withRead bool) {
assert.Equal(t, 10, len(ln.conns))
time.Sleep(4 * time.Second)
time.Sleep(ln.timeout + time.Second)
assert.Equal(t, 0, len(ln.conns))
}
@ -220,7 +228,7 @@ func TestShutdown(t *testing.T) {
addr, err := net.ResolveUDPAddr("udp", ":0")
require.NoError(t, err)
l, err := Listen("udp", addr)
l, err := Listen("udp", addr, 3*time.Second)
require.NoError(t, err)
go func() {