142 lines
3.8 KiB
Go
142 lines
3.8 KiB
Go
package server
|
|
|
|
import (
|
|
"bufio"
|
|
"context"
|
|
"net"
|
|
"net/http"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/containous/traefik/v2/pkg/config/static"
|
|
"github.com/containous/traefik/v2/pkg/tcp"
|
|
"github.com/containous/traefik/v2/pkg/types"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestShutdownHTTP(t *testing.T) {
|
|
entryPoint, err := NewTCPEntryPoint(context.Background(), &static.EntryPoint{
|
|
Address: ":0",
|
|
Transport: &static.EntryPointsTransport{
|
|
LifeCycle: &static.LifeCycle{
|
|
RequestAcceptGraceTimeout: 0,
|
|
GraceTimeOut: types.Duration(5 * time.Second),
|
|
},
|
|
},
|
|
ForwardedHeaders: &static.ForwardedHeaders{},
|
|
})
|
|
require.NoError(t, err)
|
|
|
|
go entryPoint.StartTCP(context.Background())
|
|
|
|
router := &tcp.Router{}
|
|
router.HTTPHandler(http.HandlerFunc(func(rw http.ResponseWriter, req *http.Request) {
|
|
time.Sleep(1 * time.Second)
|
|
rw.WriteHeader(http.StatusOK)
|
|
}))
|
|
entryPoint.SwitchRouter(router)
|
|
|
|
conn, err := net.Dial("tcp", entryPoint.listener.Addr().String())
|
|
require.NoError(t, err)
|
|
|
|
go entryPoint.Shutdown(context.Background())
|
|
|
|
request, err := http.NewRequest(http.MethodGet, "http://127.0.0.1:8082", nil)
|
|
require.NoError(t, err)
|
|
|
|
err = request.Write(conn)
|
|
require.NoError(t, err)
|
|
|
|
resp, err := http.ReadResponse(bufio.NewReader(conn), request)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, resp.StatusCode, http.StatusOK)
|
|
}
|
|
|
|
func TestShutdownHTTPHijacked(t *testing.T) {
|
|
entryPoint, err := NewTCPEntryPoint(context.Background(), &static.EntryPoint{
|
|
Address: ":0",
|
|
Transport: &static.EntryPointsTransport{
|
|
LifeCycle: &static.LifeCycle{
|
|
RequestAcceptGraceTimeout: 0,
|
|
GraceTimeOut: types.Duration(5 * time.Second),
|
|
},
|
|
},
|
|
ForwardedHeaders: &static.ForwardedHeaders{},
|
|
})
|
|
require.NoError(t, err)
|
|
|
|
go entryPoint.StartTCP(context.Background())
|
|
|
|
router := &tcp.Router{}
|
|
router.HTTPHandler(http.HandlerFunc(func(rw http.ResponseWriter, req *http.Request) {
|
|
conn, _, err := rw.(http.Hijacker).Hijack()
|
|
require.NoError(t, err)
|
|
time.Sleep(1 * time.Second)
|
|
|
|
resp := http.Response{StatusCode: http.StatusOK}
|
|
err = resp.Write(conn)
|
|
require.NoError(t, err)
|
|
}))
|
|
|
|
entryPoint.SwitchRouter(router)
|
|
|
|
conn, err := net.Dial("tcp", entryPoint.listener.Addr().String())
|
|
require.NoError(t, err)
|
|
|
|
go entryPoint.Shutdown(context.Background())
|
|
|
|
request, err := http.NewRequest(http.MethodGet, "http://127.0.0.1:8082", nil)
|
|
require.NoError(t, err)
|
|
|
|
err = request.Write(conn)
|
|
require.NoError(t, err)
|
|
|
|
resp, err := http.ReadResponse(bufio.NewReader(conn), request)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, resp.StatusCode, http.StatusOK)
|
|
}
|
|
|
|
func TestShutdownTCPConn(t *testing.T) {
|
|
entryPoint, err := NewTCPEntryPoint(context.Background(), &static.EntryPoint{
|
|
Address: ":0",
|
|
Transport: &static.EntryPointsTransport{
|
|
LifeCycle: &static.LifeCycle{
|
|
RequestAcceptGraceTimeout: 0,
|
|
GraceTimeOut: types.Duration(5 * time.Second),
|
|
},
|
|
},
|
|
ForwardedHeaders: &static.ForwardedHeaders{},
|
|
})
|
|
require.NoError(t, err)
|
|
|
|
go entryPoint.StartTCP(context.Background())
|
|
|
|
router := &tcp.Router{}
|
|
router.AddCatchAllNoTLS(tcp.HandlerFunc(func(conn tcp.WriteCloser) {
|
|
_, err := http.ReadRequest(bufio.NewReader(conn))
|
|
require.NoError(t, err)
|
|
time.Sleep(1 * time.Second)
|
|
|
|
resp := http.Response{StatusCode: http.StatusOK}
|
|
err = resp.Write(conn)
|
|
require.NoError(t, err)
|
|
}))
|
|
|
|
entryPoint.SwitchRouter(router)
|
|
|
|
conn, err := net.Dial("tcp", entryPoint.listener.Addr().String())
|
|
require.NoError(t, err)
|
|
|
|
go entryPoint.Shutdown(context.Background())
|
|
|
|
request, err := http.NewRequest(http.MethodGet, "http://127.0.0.1:8082", nil)
|
|
require.NoError(t, err)
|
|
|
|
err = request.Write(conn)
|
|
require.NoError(t, err)
|
|
|
|
resp, err := http.ReadResponse(bufio.NewReader(conn), request)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, resp.StatusCode, http.StatusOK)
|
|
}
|