test-http-inmemory.go

func TestHttpServer(t *testing.T) {
	mux := http.NewServeMux()
	mux.HandleFunc("/test", func(w http.ResponseWriter, r *http.Request) {
		w.Write([]byte("OK"))
	})

	srv := httptest.NewUnstartedServer(mux)
	srv.Start()

	time.Sleep(1 * time.Second)

	resp, _ := http.Get(fmt.Sprintf("http://%s/test", srv.Listener.Addr().String()))
	respBody, _ := io.ReadAll(resp.Body)
	assert.Equal(t, string(respBody), "OK")
}

func TestHttpServerInMemory(t *testing.T) {
	mux := http.NewServeMux()
	mux.HandleFunc("/test", func(w http.ResponseWriter, r *http.Request) {
		w.Write([]byte("OK"))
	})

	srv := httptest.NewUnstartedServer(mux)

	ln := fasthttputil.NewInmemoryListener()
	srv.Listener = ln
	srv.Start()

	c := http.Client{
		Transport: &http.Transport{
			DialContext: func(ctx context.Context, network, addr string) (net.Conn, error) {
				return ln.Dial()
			},
		},
	}
	resp, _ := c.Get("http://localhost/test")
	respBody, _ := io.ReadAll(resp.Body)
	assert.Equal(t, string(respBody), "OK")
}