119 Zeilen
2.1 KiB
Go
119 Zeilen
2.1 KiB
Go
|
package funcs_test
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"strings"
|
||
|
|
"testing"
|
||
|
|
"testing/fstest"
|
||
|
|
|
||
|
|
"sync"
|
||
|
|
|
||
|
|
"io/fs"
|
||
|
|
|
||
|
|
"github.com/gin-gonic/gin"
|
||
|
|
"go.sebtobie.de/httpserver/funcs"
|
||
|
|
)
|
||
|
|
|
||
|
|
const al = "abcdefghijklmnopqrstunwxyz"
|
||
|
|
|
||
|
|
type Node struct {
|
||
|
|
name string
|
||
|
|
children []*Node
|
||
|
|
}
|
||
|
|
|
||
|
|
func (n *Node) getchildren(dir string, m *sync.Map, wg *sync.WaitGroup) {
|
||
|
|
var p = n.name
|
||
|
|
if dir != "" {
|
||
|
|
p = strings.Join([]string{dir, n.name}, "/")
|
||
|
|
}
|
||
|
|
if len(n.children) != 0 {
|
||
|
|
m.Store(p, &fstest.MapFile{Mode: fs.ModeDir | fs.ModePerm})
|
||
|
|
for _, c := range n.children {
|
||
|
|
wg.Add(1)
|
||
|
|
go c.getchildren(p, m, wg)
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
m.Store(p, &fstest.MapFile{Data: []byte(p), Mode: fs.ModePerm})
|
||
|
|
}
|
||
|
|
wg.Done()
|
||
|
|
}
|
||
|
|
|
||
|
|
func getFS(depth, width int) fstest.MapFS {
|
||
|
|
var nodes = &Node{
|
||
|
|
"",
|
||
|
|
createNodes(depth, width),
|
||
|
|
}
|
||
|
|
var m = sync.Map{}
|
||
|
|
var wg = sync.WaitGroup{}
|
||
|
|
wg.Add(1)
|
||
|
|
go nodes.getchildren("", &m, &wg)
|
||
|
|
wg.Wait()
|
||
|
|
var fs = make(fstest.MapFS)
|
||
|
|
m.Range(func(key, value interface{}) bool {
|
||
|
|
fs[key.(string)] = value.(*fstest.MapFile)
|
||
|
|
return true
|
||
|
|
})
|
||
|
|
return fs
|
||
|
|
}
|
||
|
|
|
||
|
|
func name(i int) string {
|
||
|
|
var output string
|
||
|
|
for {
|
||
|
|
if i < 26 {
|
||
|
|
output = output + string(al[i])
|
||
|
|
break
|
||
|
|
}
|
||
|
|
output += string(al[i%26])
|
||
|
|
i = (i - (i % 26)) / 26
|
||
|
|
}
|
||
|
|
return output
|
||
|
|
}
|
||
|
|
|
||
|
|
func createNodes(depth, width int) (n []*Node) {
|
||
|
|
if depth != 0 {
|
||
|
|
for i := 0; i < width; i++ {
|
||
|
|
n = append(n, &Node{string(name(i)), createNodes(depth-1, width)})
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
func toMap(routes gin.RoutesInfo) (m map[string]gin.RouteInfo) {
|
||
|
|
m = make(map[string]gin.RouteInfo)
|
||
|
|
for _, r := range routes {
|
||
|
|
m[r.Path] = r
|
||
|
|
}
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestStaticFS(t *testing.T) {
|
||
|
|
t.Parallel()
|
||
|
|
var (
|
||
|
|
r = gin.New()
|
||
|
|
rg = r.Group("")
|
||
|
|
fs fstest.MapFS = getFS(5, 10)
|
||
|
|
)
|
||
|
|
funcs.StaticFS(rg, &fs)
|
||
|
|
routes := toMap(r.Routes())
|
||
|
|
t.Log(routes)
|
||
|
|
for path, node := range fs {
|
||
|
|
if node.Mode.IsDir() {
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
_, found := routes["/"+path]
|
||
|
|
if !found {
|
||
|
|
t.Errorf("Path %s was not handled", path)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func BenchmarkStaticFS(b *testing.B) {
|
||
|
|
ro := gin.New()
|
||
|
|
fs := getFS(52, 2)
|
||
|
|
b.ResetTimer()
|
||
|
|
for i := 0; i < b.N; i++ {
|
||
|
|
g := ro.Group(fmt.Sprint(i))
|
||
|
|
funcs.StaticFS(g, fs)
|
||
|
|
}
|
||
|
|
}
|