mirror of
https://github.com/kataras/iris.git
synced 2025-01-23 18:51:03 +01:00
62 lines
997 B
Go
62 lines
997 B
Go
package main
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/kataras/iris/v12/httptest"
|
|
)
|
|
|
|
type resource string
|
|
|
|
func (r resource) String() string {
|
|
return string(r)
|
|
}
|
|
|
|
func (r resource) strip(strip string) string {
|
|
s := r.String()
|
|
return strings.TrimPrefix(s, strip)
|
|
}
|
|
|
|
func (r resource) loadFromBase(dir string) string {
|
|
filename := r.String()
|
|
|
|
if filename == "/" {
|
|
filename = "/index.html"
|
|
}
|
|
|
|
fullpath := filepath.Join(dir, filename)
|
|
|
|
b, err := os.ReadFile(fullpath)
|
|
if err != nil {
|
|
panic(fullpath + " failed with error: " + err.Error())
|
|
}
|
|
|
|
return string(b)
|
|
}
|
|
|
|
var urls = []resource{
|
|
"/",
|
|
"/index.html",
|
|
"/app.js",
|
|
"/css/main.css",
|
|
}
|
|
|
|
func TestCustomWrapper(t *testing.T) {
|
|
app := newApp()
|
|
e := httptest.New(t, app)
|
|
|
|
for _, u := range urls {
|
|
url := u.String()
|
|
contents := u.loadFromBase("./public")
|
|
|
|
e.GET(url).Expect().
|
|
Status(httptest.StatusOK).
|
|
Body().Equal(contents)
|
|
}
|
|
|
|
e.GET("/other/something").Expect().Status(httptest.StatusOK)
|
|
}
|