reset backend
This commit is contained in:
@@ -1,228 +0,0 @@
|
||||
package services
|
||||
|
||||
import (
|
||||
"database/sql"
|
||||
"fmt"
|
||||
"testing"
|
||||
|
||||
"quyun/v2/app/commands/testx"
|
||||
"quyun/v2/app/http/super/dto"
|
||||
"quyun/v2/database"
|
||||
"quyun/v2/database/models"
|
||||
"quyun/v2/pkg/consts"
|
||||
"quyun/v2/pkg/utils"
|
||||
|
||||
"github.com/samber/lo"
|
||||
. "github.com/smartystreets/goconvey/convey"
|
||||
"github.com/stretchr/testify/suite"
|
||||
|
||||
_ "go.ipao.vip/atom"
|
||||
"go.ipao.vip/atom/contracts"
|
||||
"go.ipao.vip/gen/types"
|
||||
"go.uber.org/dig"
|
||||
)
|
||||
|
||||
type UserTestSuiteInjectParams struct {
|
||||
dig.In
|
||||
|
||||
DB *sql.DB
|
||||
Initials []contracts.Initial `group:"initials"` // nolint:structcheck
|
||||
}
|
||||
|
||||
type UserTestSuite struct {
|
||||
suite.Suite
|
||||
|
||||
UserTestSuiteInjectParams
|
||||
}
|
||||
|
||||
func Test_User(t *testing.T) {
|
||||
providers := testx.Default().With(Provide)
|
||||
|
||||
testx.Serve(providers, t, func(p UserTestSuiteInjectParams) {
|
||||
suite.Run(t, &UserTestSuite{UserTestSuiteInjectParams: p})
|
||||
})
|
||||
}
|
||||
|
||||
func (t *UserTestSuite) Test_Create() {
|
||||
Convey("test user create", t.T(), func() {
|
||||
database.Truncate(t.T().Context(), t.DB, models.TableNameUser)
|
||||
|
||||
m := &models.User{
|
||||
Username: "test-user",
|
||||
Password: "test-password",
|
||||
Roles: types.NewArray([]consts.Role{consts.RoleUser, consts.RoleSuperAdmin}),
|
||||
Status: consts.UserStatusPendingVerify,
|
||||
}
|
||||
|
||||
err := m.Create(t.T().Context())
|
||||
So(err, ShouldBeNil)
|
||||
|
||||
So(m.ID, ShouldBeGreaterThan, 0)
|
||||
|
||||
same := m.ComparePassword(t.T().Context(), "test-password")
|
||||
So(same, ShouldBeTrue)
|
||||
|
||||
same = m.ComparePassword(t.T().Context(), "test-password1")
|
||||
So(same, ShouldBeFalse)
|
||||
})
|
||||
}
|
||||
|
||||
// FindByUsername
|
||||
func (t *UserTestSuite) Test_FindByUsername() {
|
||||
Convey("test user FindByUsername", t.T(), func() {
|
||||
database.Truncate(t.T().Context(), t.DB, models.TableNameUser)
|
||||
|
||||
Convey("user table is empty", func() {
|
||||
m, err := User.FindByUsername(t.T().Context(), "test-user")
|
||||
So(err, ShouldNotBeNil)
|
||||
So(m, ShouldBeNil)
|
||||
})
|
||||
|
||||
Convey("insert one record", func() {
|
||||
username := "test-user"
|
||||
m := &models.User{
|
||||
Username: username,
|
||||
Password: "test-password",
|
||||
Roles: types.NewArray([]consts.Role{consts.RoleUser, consts.RoleSuperAdmin}),
|
||||
Status: consts.UserStatusPendingVerify,
|
||||
}
|
||||
|
||||
err := m.Create(t.T().Context())
|
||||
So(err, ShouldBeNil)
|
||||
|
||||
Convey("user table is not empty", func() {
|
||||
m, err := User.FindByUsername(t.T().Context(), username)
|
||||
So(err, ShouldBeNil)
|
||||
So(m, ShouldNotBeNil)
|
||||
})
|
||||
})
|
||||
})
|
||||
}
|
||||
|
||||
// Test_Page
|
||||
func (t *UserTestSuite) Test_Page() {
|
||||
Convey("test page", t.T(), func() {
|
||||
Convey("filter username", func() {
|
||||
database.Truncate(t.T().Context(), t.DB, models.TableNameUser)
|
||||
|
||||
username := "test-user"
|
||||
m := &models.User{
|
||||
Username: username,
|
||||
Password: "test-password",
|
||||
Roles: types.NewArray([]consts.Role{consts.RoleUser, consts.RoleSuperAdmin}),
|
||||
Status: consts.UserStatusPendingVerify,
|
||||
}
|
||||
|
||||
err := m.Create(t.T().Context())
|
||||
So(err, ShouldBeNil)
|
||||
|
||||
pager, err := User.Page(t.T().Context(), &dto.UserPageFilter{
|
||||
Username: &username,
|
||||
})
|
||||
|
||||
So(err, ShouldBeNil)
|
||||
So(pager.Total, ShouldEqual, 1)
|
||||
})
|
||||
|
||||
Convey("filter tenant users", func() {
|
||||
database.Truncate(
|
||||
t.T().Context(),
|
||||
t.DB,
|
||||
models.TableNameUser,
|
||||
models.TableNameTenant,
|
||||
models.TableNameTenantUser,
|
||||
)
|
||||
|
||||
username := "test-user"
|
||||
m := &models.User{
|
||||
Username: username,
|
||||
Password: "test-password",
|
||||
Roles: types.NewArray([]consts.Role{consts.RoleUser, consts.RoleSuperAdmin}),
|
||||
Status: consts.UserStatusPendingVerify,
|
||||
}
|
||||
|
||||
err := m.Create(t.T().Context())
|
||||
So(err, ShouldBeNil)
|
||||
|
||||
m = &models.User{
|
||||
Username: username + "02",
|
||||
Password: "test-password",
|
||||
Roles: types.NewArray([]consts.Role{consts.RoleUser}),
|
||||
Status: consts.UserStatusPendingVerify,
|
||||
}
|
||||
|
||||
err = m.Create(t.T().Context())
|
||||
So(err, ShouldBeNil)
|
||||
|
||||
tenantModel := &models.Tenant{
|
||||
UserID: 1,
|
||||
Code: "abc",
|
||||
UUID: types.NewUUIDv4(),
|
||||
Name: "T01",
|
||||
Status: consts.TenantStatusVerified,
|
||||
}
|
||||
|
||||
err = tenantModel.Create(t.T().Context())
|
||||
So(err, ShouldBeNil)
|
||||
|
||||
tenantModel = &models.Tenant{
|
||||
UserID: 2,
|
||||
Code: "abc01",
|
||||
UUID: types.NewUUIDv4(),
|
||||
Name: "T02",
|
||||
Status: consts.TenantStatusVerified,
|
||||
}
|
||||
|
||||
err = tenantModel.Create(t.T().Context())
|
||||
So(err, ShouldBeNil)
|
||||
|
||||
count := 12
|
||||
for i := 0; i < count; i++ {
|
||||
m = &models.User{
|
||||
Username: fmt.Sprintf("user_%d", i),
|
||||
Password: "test-password",
|
||||
Roles: types.NewArray([]consts.Role{consts.RoleUser}),
|
||||
Status: consts.UserStatusPendingVerify,
|
||||
}
|
||||
|
||||
err = m.Create(t.T().Context())
|
||||
So(err, ShouldBeNil)
|
||||
|
||||
// create tenant user
|
||||
err = Tenant.AddUser(t.T().Context(), int64(i%2+1), m.ID)
|
||||
So(err, ShouldBeNil)
|
||||
}
|
||||
|
||||
pager, err := User.Page(t.T().Context(), &dto.UserPageFilter{
|
||||
TenantID: lo.ToPtr(int64(1)),
|
||||
})
|
||||
|
||||
So(err, ShouldBeNil)
|
||||
So(pager.Total, ShouldEqual, 6)
|
||||
})
|
||||
})
|
||||
}
|
||||
|
||||
func (t *UserTestSuite) Test_Relations() {
|
||||
Convey("test page", t.T(), func() {
|
||||
Convey("filter tenant users", func() {
|
||||
m, err := User.FindByID(t.T().Context(), 1)
|
||||
So(err, ShouldBeNil)
|
||||
// So(m.OwnedTenant, ShouldNotBeNil)
|
||||
// So(m.Tenants, ShouldHaveLength, 10)
|
||||
t.T().Logf("%s", utils.MustJsonString(m))
|
||||
})
|
||||
})
|
||||
}
|
||||
|
||||
func (t *UserTestSuite) Test_Statistics() {
|
||||
Convey("test page", t.T(), func() {
|
||||
Convey("filter tenant users", func() {
|
||||
m, err := User.Statistics(t.T().Context())
|
||||
So(err, ShouldBeNil)
|
||||
// So(m.OwnedTenant, ShouldNotBeNil)
|
||||
// So(m.Tenants, ShouldHaveLength, 10)
|
||||
t.T().Logf("%s", utils.MustJsonString(m))
|
||||
})
|
||||
})
|
||||
}
|
||||
Reference in New Issue
Block a user