package aero_test import ( "errors" "io" "net/http" "net/http/httptest" "testing" "git.akyoto.dev/go/aero" "git.akyoto.dev/go/assert" ) func TestServer(t *testing.T) { server := aero.New() server.Get("/", func(ctx aero.Context) error { return ctx.Bytes([]byte("Hello")) }) server.Get("/blog/:post", func(ctx aero.Context) error { return ctx.Bytes([]byte("Hello")) }) server.Get("/error", func(ctx aero.Context) error { return ctx.Error(http.StatusUnauthorized, errors.New("Not logged in")) }) tests := []struct { URL string Status int Body string }{ {URL: "/", Status: http.StatusOK, Body: "Hello"}, {URL: "/blog/post", Status: http.StatusOK, Body: "Hello"}, {URL: "/error", Status: http.StatusUnauthorized, Body: "Not logged in"}, {URL: "/not-found", Status: http.StatusNotFound, Body: http.StatusText(http.StatusNotFound)}, } for _, test := range tests { t.Run("example.com"+test.URL, func(t *testing.T) { request := httptest.NewRequest(http.MethodGet, test.URL, nil) response := httptest.NewRecorder() server.ServeHTTP(response, request) result := response.Result() assert.Equal(t, result.StatusCode, test.Status) body, err := io.ReadAll(result.Body) assert.Nil(t, err) assert.DeepEqual(t, string(body), test.Body) }) } }