huma/docs_test.go

41 lines
878 B
Go
Raw Permalink Normal View History

2020-08-30 23:58:12 -07:00
package huma
import (
"net/http"
"net/http/httptest"
"testing"
"github.com/stretchr/testify/assert"
)
var handlers = []struct {
name string
handler http.Handler
}{
2020-09-10 22:28:29 -07:00
{"RapiDoc", RapiDocHandler(New("Test API", "1.0.0"))},
{"ReDoc", ReDocHandler(New("Test API", "1.0.0"))},
{"SwaggerUI", SwaggerUIHandler(New("Test API", "1.0.0"))},
2020-08-30 23:58:12 -07:00
}
func TestDocHandlers(outer *testing.T) {
for _, tt := range handlers {
local := tt
outer.Run(local.name, func(t *testing.T) {
app := newTestRouter()
app.DocsHandler(local.handler)
w := httptest.NewRecorder()
req, _ := http.NewRequest(http.MethodGet, "/docs", nil)
app.ServeHTTP(w, req)
assert.Equal(t, http.StatusOK, w.Code)
})
}
}
func TestSplitDocs(t *testing.T) {
title, desc := splitDocs("One two\nthree\nfour five")
assert.Equal(t, "One two", title)
assert.Equal(t, "three\nfour five", desc)
}