Signed-off-by: Chun-Hung Tseng <henrybear327@gmail.com>
This commit is contained in:
Chun-Hung Tseng 2024-09-25 14:36:34 +02:00
parent 85f7a9206b
commit 65c1213c7c
No known key found for this signature in database
GPG Key ID: EF93C20F55FB48BB

View File

@ -17,7 +17,6 @@ package proxy
import ( import (
"bytes" "bytes"
"context" "context"
"fmt"
"io" "io"
"log" "log"
"net" "net"
@ -33,13 +32,13 @@ import (
"go.etcd.io/etcd/client/pkg/v3/transport" "go.etcd.io/etcd/client/pkg/v3/transport"
) )
/* Helper functions */ /* dummyServerHandler is a helper struct */
type dummyServerHandler struct { type dummyServerHandler struct {
t *testing.T t *testing.T
output chan<- []byte output chan<- []byte
} }
// reads the request body and write back to the response object // ServeHTTP read the request body and write back to the response object
func (sh *dummyServerHandler) ServeHTTP(resp http.ResponseWriter, req *http.Request) { func (sh *dummyServerHandler) ServeHTTP(resp http.ResponseWriter, req *http.Request) {
defer req.Body.Close() defer req.Body.Close()
resp.WriteHeader(200) resp.WriteHeader(200)
@ -181,7 +180,7 @@ func startHTTPServer(scheme, addr string, tlsInfo transport.TLSInfo, httpServer
log.Println("HTTP Server started on", addr) log.Println("HTTP Server started on", addr)
if err := httpServer.ServeTLS(ln, tlsInfo.CertFile, tlsInfo.KeyFile); err != http.ErrServerClosed { if err := httpServer.ServeTLS(ln, tlsInfo.CertFile, tlsInfo.KeyFile); err != http.ErrServerClosed {
// always returns error. ErrServerClosed on graceful close // always returns error. ErrServerClosed on graceful close
log.Fatalf(fmt.Sprintf("startHTTPServer ServeTLS(): %v", err)) log.Fatalf("startHTTPServer ServeTLS(): %v", err)
} }
} }
@ -248,7 +247,6 @@ func waitForServer(t *testing.T, s Server) {
} }
} }
/* Unit tests */
func TestServer_TCP(t *testing.T) { testServer(t, false) } func TestServer_TCP(t *testing.T) { testServer(t, false) }
func TestServer_TCP_DelayTx(t *testing.T) { testServer(t, true) } func TestServer_TCP_DelayTx(t *testing.T) { testServer(t, true) }
func testServer(t *testing.T, delayTx bool) { func testServer(t *testing.T, delayTx bool) {
@ -398,7 +396,6 @@ func TestServer_Shutdown(t *testing.T) {
if err := s.Close(); err != nil { if err := s.Close(); err != nil {
t.Fatal(err) t.Fatal(err)
} }
p = nil
time.Sleep(200 * time.Millisecond) time.Sleep(200 * time.Millisecond)
data := []byte("Hello World!") data := []byte("Hello World!")