package middleware import ( "context" "fmt" "net/http" "net/http/httptest" "testing" "github.com/stretchr/testify/assert" "github.com/stretchr/testify/require" ) func TestRedocMiddleware(t *testing.T) { t.Run("with defaults", func(t *testing.T) { redoc := Redoc(RedocOpts{}, nil) req, err := http.NewRequestWithContext(context.Background(), http.MethodGet, "/docs", nil) require.NoError(t, err) recorder := httptest.NewRecorder() redoc.ServeHTTP(recorder, req) assert.Equal(t, http.StatusOK, recorder.Code) assert.Equal(t, "text/html; charset=utf-8", recorder.Header().Get(contentTypeHeader)) var o RedocOpts o.EnsureDefaults() assert.Contains(t, recorder.Body.String(), fmt.Sprintf("", o.Title)) assert.Contains(t, recorder.Body.String(), fmt.Sprintf("", o.SpecURL)) assert.Contains(t, recorder.Body.String(), redocLatest) }) t.Run("with alternate path and spec URL", func(t *testing.T) { redoc := Redoc(RedocOpts{ BasePath: "/base", Path: "ui", SpecURL: "/ui/swagger.json", }, nil) req, err := http.NewRequestWithContext(context.Background(), http.MethodGet, "/base/ui", nil) require.NoError(t, err) recorder := httptest.NewRecorder() redoc.ServeHTTP(recorder, req) assert.Equal(t, http.StatusOK, recorder.Code) assert.Contains(t, recorder.Body.String(), "") }) t.Run("with custom template", func(t *testing.T) { redoc := Redoc(RedocOpts{ Template: `
{{ .Title }}