net_basic_test.go

package net

import (
	"io"
	"net"
	"net/http"
	"strings"
	"testing"

	"github.com/go-chi/chi/v5"
)

func TestImpl_ListenAndServe(t *testing.T) {
	const addr = "127.0.0.1:8080"
	const uri = "http" + "://" + addr + "/"

	// Сервер.
	nut := New().
		Handler(func(l net.Listener) (err error) {
			var (
				route *chi.Mux
				srv   *http.Server
			)

			route = chi.NewRouter()
			route.Get("/", func(wr http.ResponseWriter, rq *http.Request) {
				wr.Header().Set("Content-Type", "text/plain")
				_, _ = io.WriteString(wr, "Hello, World!")
			})
			srv = &http.Server{
				Addr:    addr,
				Handler: route,
			}
			err = srv.Serve(l)

			return
		}).
		ListenAndServe(addr)
	if nut.Error() != nil {
		t.Errorf("запуск сервера прерван ошибкой: %s", nut.Error())
		return
	}
	// Клиент.
	go func() {
		defer func() { nut.Stop() }() // Остановка сервера.
		rq, e := http.Get(uri)
		if e != nil {
			t.Errorf("запрос к %q прерван ошибкой: %s", uri, e)
		}
		defer func() { _ = rq.Body.Close() }()
		buf, e := io.ReadAll(rq.Body)
		if e != nil {
			t.Errorf("чтение ответа на запрос к %q прервано ошибкой: %s", uri, e)
		}
		if rq.StatusCode != 200 || !strings.Contains(string(buf), "Hello, World") {
			t.Errorf("получен не корректный ответ сервера: %q, ожидался: %q", string(buf), "Hello, World")
		}
	}()
	// Ожидание завершения сервера.
	if err := nut.Wait().
		Error(); err != nil {
		t.Errorf("сервер завершился с ошибкой: %s", err)
		return
	}
	nut.Stop()
}