mirror of
https://github.com/crawlab-team/crawlab.git
synced 2026-01-22 17:31:03 +01:00
105 lines
2.0 KiB
Go
105 lines
2.0 KiB
Go
package test
|
|
|
|
import (
|
|
"github.com/crawlab-team/crawlab/core/controllers"
|
|
"github.com/crawlab-team/crawlab/core/models/service"
|
|
"github.com/crawlab-team/crawlab/core/routes"
|
|
"github.com/crawlab-team/go-trace"
|
|
"github.com/gavv/httpexpect/v2"
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/stretchr/testify/require"
|
|
"go.uber.org/dig"
|
|
"net/http/httptest"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func init() {
|
|
var err error
|
|
T, err = NewTest()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
type Test struct {
|
|
// dependencies
|
|
modelSvc service.ModelService
|
|
|
|
// internals
|
|
app *gin.Engine
|
|
svr *httptest.Server
|
|
|
|
// test data
|
|
TestUsername string
|
|
TestPassword string
|
|
TestToken string
|
|
}
|
|
|
|
func (t *Test) Setup(t2 *testing.T) {
|
|
//if err := controllers.InitControllers(); err != nil {
|
|
// panic(err)
|
|
//}
|
|
//t2.Cleanup(t.Cleanup)
|
|
}
|
|
|
|
func (t *Test) Cleanup() {
|
|
_ = t.modelSvc.DropAll()
|
|
time.Sleep(200 * time.Millisecond)
|
|
}
|
|
|
|
func (t *Test) NewExpect(t2 *testing.T) (e *httpexpect.Expect) {
|
|
e = httpexpect.New(t2, t.svr.URL)
|
|
res := e.POST("/login").WithJSON(map[string]string{
|
|
"username": t.TestUsername,
|
|
"password": t.TestPassword,
|
|
}).Expect().JSON().Object()
|
|
t.TestToken = res.Path("$.data").String().Raw()
|
|
require.NotEmpty(t2, t.TestToken)
|
|
return e
|
|
}
|
|
|
|
func (t *Test) WithAuth(req *httpexpect.Request) *httpexpect.Request {
|
|
return req.WithHeader("Authorization", t.TestToken)
|
|
}
|
|
|
|
var T *Test
|
|
|
|
func NewTest() (res *Test, err error) {
|
|
// test
|
|
t := &Test{}
|
|
|
|
// gin app
|
|
t.app = gin.New()
|
|
|
|
// http test server
|
|
t.svr = httptest.NewServer(t.app)
|
|
|
|
// init controllers
|
|
if err := controllers.InitControllers(); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// init routes
|
|
if err := routes.InitRoutes(t.app); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// dependency injection
|
|
c := dig.New()
|
|
if err := c.Provide(service.NewService); err != nil {
|
|
return nil, trace.TraceError(err)
|
|
}
|
|
if err := c.Invoke(func(modelSvc service.ModelService) {
|
|
t.modelSvc = modelSvc
|
|
}); err != nil {
|
|
return nil, trace.TraceError(err)
|
|
}
|
|
|
|
// test data
|
|
t.TestUsername = "admin"
|
|
t.TestPassword = "admin"
|
|
|
|
return t, nil
|
|
}
|