embed_folder_test.go

package static

import (
	"embed"
	"fmt"
	"testing"

	"github.com/gin-gonic/gin"
	"github.com/stretchr/testify/assert"
)

//go:embed test/data/server
var server embed.FS

var embedTests = []struct {
	targetURL string // input
	httpCode  int    // expected http code
	httpBody  string // expected http body
	name      string // test name
}{
	{"/404.html", 301, "<a href=\"/\">Moved Permanently</a>.\n\n", "Unknown file"},
	{"/", 200, "<h1>Hello Embed</h1>", "Root"},
	{"/index.html", 301, "", "Root by file name automatic redirect"},
	{"/static.html", 200, "<h1>Hello Gin Static</h1>", "Other file"},
}

func TestEmbedFolder(t *testing.T) {
	router := gin.New()
	router.Use(Serve("/", EmbedFolder(server, "test/data/server")))
	router.NoRoute(func(c *gin.Context) {
		fmt.Printf("%s doesn't exists, redirect on /\n", c.Request.URL.Path)
		c.Redirect(301, "/")
	})

	for _, tt := range embedTests {
		w := PerformRequest(router, "GET", tt.targetURL)
		assert.Equal(t, tt.httpCode, w.Code, tt.name)
		assert.Equal(t, tt.httpBody, w.Body.String(), tt.name)
	}
}