chore: fix the URL on swagger UI after refactoring (#23)

swaggo · May 26, 2022 · 476cf42 · 476cf42
1 parent 60605cf
commit 476cf42
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/swagger.go b/swagger.go
@@ -219,7 +219,7 @@ const swaggerIndexTpl = `<!-- HTML for static distribution bundle build -->
 window.onload = function() {
   // Build a system
   const ui = SwaggerUIBundle({
-    url: "{{.Host}}",
+    url: "{{.URL}}",
     dom_id: '#swagger-ui',
     validatorUrl: null,
 	defaultModelRendering: "model",

diff --git a/swagger_test.go b/swagger_test.go
@@ -1,6 +1,7 @@
 package buffaloSwagger
 
 import (
+	"io/ioutil"
 	"net/http"
 	"net/http/httptest"
 	"testing"
@@ -30,6 +31,8 @@ func TestWrapHandler(t *testing.T) {
 	w1 := performRequest(http.MethodGet, "/swagger/index.html", router)
 	assert.Equal(t, http.StatusOK, w1.Code)
 	assert.Equal(t, w1.Header()["Content-Type"][0], "text/html; charset=utf-8")
+	w1BodyBytes, _ := ioutil.ReadAll(w1.Body)
+	assert.Contains(t, string(w1BodyBytes), "doc.json")
 
 	assert.Equal(t, http.StatusInternalServerError, performRequest(http.MethodGet, "/swagger/doc.json", router).Code)