150 lines
3.0 KiB
Go
150 lines
3.0 KiB
Go
package app
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
_ "github.com/xiaoqidun/entps"
|
|
|
|
"git.dotya.ml/mirre-mt/pcmt/app/settings"
|
|
"git.dotya.ml/mirre-mt/pcmt/config"
|
|
"git.dotya.ml/mirre-mt/pcmt/ent/enttest"
|
|
"git.dotya.ml/mirre-mt/pcmt/slogging"
|
|
)
|
|
|
|
const connstr = "file:ent_tests?mode=memory&_fk=1"
|
|
|
|
var conf = &config.Config{
|
|
Port: 3005,
|
|
AppName: "pcmt-test",
|
|
LiveMode: true,
|
|
DevelMode: false,
|
|
SessionCookieName: "sessionz",
|
|
SessionCookieSecret: "secret",
|
|
}
|
|
|
|
var (
|
|
host = new(string)
|
|
port = new(int)
|
|
devel = new(bool)
|
|
)
|
|
|
|
func TestStaticRoute(t *testing.T) {
|
|
setting := settings.New()
|
|
a := &App{
|
|
templatesPath: "../templates",
|
|
assetsPath: "../assets",
|
|
}
|
|
log := slogging.Init(false)
|
|
tstRoute := "/static/"
|
|
db := enttest.Open(t, "sqlite3", connstr)
|
|
|
|
defer db.Close()
|
|
|
|
if err := db.Schema.Create(context.Background()); err != nil {
|
|
t.Fatalf("failed creating schema resources: %v", err)
|
|
}
|
|
|
|
*host = ""
|
|
*port = 3500
|
|
*devel = true
|
|
|
|
setting.Consolidate(conf, host, port, devel, "test")
|
|
|
|
err := a.Init(setting, log, db)
|
|
if err != nil {
|
|
t.Errorf("failed to initialise app: %v", a)
|
|
}
|
|
|
|
a.SetupRoutes()
|
|
|
|
req := httptest.NewRequest(http.MethodGet, tstRoute, nil)
|
|
rec := httptest.NewRecorder()
|
|
|
|
a.E().ServeHTTP(rec, req)
|
|
|
|
want := http.StatusMovedPermanently
|
|
got := rec.Code
|
|
|
|
if want != got {
|
|
t.Fatalf("unexpected status code when testing %s, want: %d, got %d",
|
|
tstRoute, want, got,
|
|
)
|
|
}
|
|
|
|
uri, err := rec.Result().Location()
|
|
if err != nil {
|
|
t.Fatal("error getting response location")
|
|
}
|
|
|
|
if uri.Path != "/assets/" {
|
|
t.Errorf("unexpected location, want: %s, got: %s", "/assets/", uri.Path)
|
|
}
|
|
}
|
|
|
|
func BenchmarkStatic(b *testing.B) {
|
|
setting := settings.New()
|
|
a := &App{
|
|
templatesPath: "../templates",
|
|
assetsPath: "../assets",
|
|
}
|
|
log := slogging.Init(false)
|
|
db := enttest.Open(b, "sqlite3", connstr)
|
|
|
|
defer db.Close()
|
|
|
|
if err := db.Schema.Create(context.Background()); err != nil {
|
|
b.Fatalf("failed creating schema resources: %v", err)
|
|
}
|
|
|
|
setting.Consolidate(conf, host, port, devel, "bench")
|
|
|
|
err := a.Init(setting, log, db)
|
|
if err != nil {
|
|
b.Errorf("failed to initialise app: %v", a)
|
|
}
|
|
|
|
a.SetupRoutes()
|
|
|
|
for i := 0; i < b.N; i++ {
|
|
req := httptest.NewRequest(http.MethodGet, "/static/", nil)
|
|
rec := httptest.NewRecorder()
|
|
|
|
a.E().ServeHTTP(rec, req)
|
|
}
|
|
}
|
|
|
|
func BenchmarkStatic2(b *testing.B) {
|
|
setting := settings.New()
|
|
a := &App{
|
|
templatesPath: "../templates",
|
|
assetsPath: "../assets",
|
|
}
|
|
log := slogging.Init(false)
|
|
db := enttest.Open(b, "sqlite3", connstr)
|
|
|
|
defer db.Close()
|
|
|
|
if err := db.Schema.Create(context.Background()); err != nil {
|
|
b.Fatalf("failed creating schema resources: %v", err)
|
|
}
|
|
|
|
setting.Consolidate(conf, host, port, devel, "bench")
|
|
|
|
err := a.Init(setting, log, db)
|
|
if err != nil {
|
|
b.Errorf("failed to initialise app: %v", a)
|
|
}
|
|
|
|
a.SetupRoutes()
|
|
|
|
for i := 0; i < b.N; i++ {
|
|
req := httptest.NewRequest(http.MethodGet, "/static2/", nil)
|
|
rec := httptest.NewRecorder()
|
|
|
|
a.E().ServeHTTP(rec, req)
|
|
}
|
|
}
|