TrueCloudLab/frostfs-http-gw
17
1
Fork 18
Code Issues 10 Pull requests 4 Projects Releases 16 Activity Actions

[#216] Rework http2 test to be tls test

Browse source 
Signed-off-by: Nikita Zinkevich <n.zinkevich@yadro.com>
This commit is contained in:
Nikita Zinkevich 2025-02-18 13:24:20 +03:00
parent 22ee93ab5e
commit 07f997b3ec
1 changed files with 13 additions and 20 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

33
cmd/http-gw/server_test.go
View file

@ -18,7 +18,7 @@ import (
"time" "time"
"github.com/stretchr/testify/require" "github.com/stretchr/testify/require"
"golang.org/x/net/http2" "github.com/valyala/fasthttp"
) )
const ( const (
@ -26,14 +26,10 @@ const (
expHeaderValue = "Bar" expHeaderValue = "Bar"
) )
func TestHTTP2TLS(t *testing.T) { func TestHTTP_TLS(t *testing.T) {
ctx := context.Background() ctx := context.Background()
certPath, keyPath := prepareTestCerts(t) certPath, keyPath := prepareTestCerts(t)
srv := &http.Server{
Handler: http.HandlerFunc(testHandler),
}
tlsListener, err := newServer(ctx, ServerInfo{ tlsListener, err := newServer(ctx, ServerInfo{
Address: ":0", Address: ":0",
TLS: ServerTLSInfo{ TLS: ServerTLSInfo{
@ -47,37 +43,34 @@ func TestHTTP2TLS(t *testing.T) {
addr := fmt.Sprintf("https://localhost:%d", port) addr := fmt.Sprintf("https://localhost:%d", port)
go func() { go func() {
_ = srv.Serve(tlsListener.Listener()) _ = fasthttp.Serve(tlsListener.Listener(), testHandler)
}() }()
// Server is running, now send HTTP/2 request
tlsClientConfig := &tls.Config{ tlsClientConfig := &tls.Config{
InsecureSkipVerify: true, InsecureSkipVerify: true,
} }
cliHTTP1 := http.Client{Transport: &http.Transport{TLSClientConfig: tlsClientConfig}} cliHTTP := http.Client{Transport: &http.Transport{}}
cliHTTP2 := http.Client{Transport: &http2.Transport{TLSClientConfig: tlsClientConfig}} cliHTTPS := http.Client{Transport: &http.Transport{TLSClientConfig: tlsClientConfig}}
req, err := http.NewRequest("GET", addr, nil) req, err := http.NewRequest("GET", addr, nil)
require.NoError(t, err) require.NoError(t, err)
req.Header[expHeaderKey] = []string{expHeaderValue} req.Header[expHeaderKey] = []string{expHeaderValue}
resp, err := cliHTTP1.Do(req) resp, err := cliHTTPS.Do(req)
require.NoError(t, err) require.NoError(t, err)
require.Equal(t, http.StatusOK, resp.StatusCode) require.Equal(t, http.StatusOK, resp.StatusCode)
resp, err = cliHTTP2.Do(req) _, err = cliHTTP.Do(req)
require.NoError(t, err) require.ErrorContains(t, err, "failed to verify certificate")
require.Equal(t, http.StatusOK, resp.StatusCode)
} }
func testHandler(resp http.ResponseWriter, req *http.Request) { func testHandler(ctx *fasthttp.RequestCtx) {
hdr, ok := req.Header[expHeaderKey] hdr := ctx.Request.Header.Peek(expHeaderKey)
if !ok || len(hdr) != 1 || hdr[0] != expHeaderValue { if len(hdr) == 0 || string(hdr) != expHeaderValue {
resp.WriteHeader(http.StatusBadRequest) ctx.Response.SetStatusCode(http.StatusBadRequest)
} else { } else {
resp.WriteHeader(http.StatusOK) ctx.Response.SetStatusCode(http.StatusOK)
} }
} }