haibei/test/test.go
2025-06-19 10:33:58 +08:00

56 lines
950 B
Go

package test
import (
"context"
"net/http"
"os"
"testing"
"github.com/gavv/httpexpect/v2"
"github.com/gin-gonic/gin"
"github.guxuan/haibei/internal/config"
"github.guxuan/haibei/internal/wirex"
)
const (
baseAPI = "/api/v1"
)
var (
app *gin.Engine
)
func init() {
config.MustLoad("")
_ = os.RemoveAll(config.C.Storage.DB.DSN)
ctx := context.Background()
injector, _, err := wirex.BuildInjector(ctx)
if err != nil {
panic(err)
}
if err := injector.M.Init(ctx); err != nil {
panic(err)
}
app = gin.New()
err = injector.M.RegisterRouters(ctx, app)
if err != nil {
panic(err)
}
}
func tester(t *testing.T) *httpexpect.Expect {
return httpexpect.WithConfig(httpexpect.Config{
Client: &http.Client{
Transport: httpexpect.NewBinder(app),
Jar: httpexpect.NewCookieJar(),
},
Reporter: httpexpect.NewAssertReporter(t),
Printers: []httpexpect.Printer{
httpexpect.NewDebugPrinter(t, true),
},
})
}