tenant: move admin member queries into service

This commit is contained in:
2025-12-18 18:03:32 +08:00
parent eedb940799
commit 462bde351d
8 changed files with 331 additions and 26 deletions

View File

@@ -5,6 +5,7 @@ import (
"testing"
"quyun/v2/app/commands/testx"
tenantdto "quyun/v2/app/http/tenant/dto"
"quyun/v2/database"
"quyun/v2/database/models"
"quyun/v2/pkg/consts"
@@ -15,6 +16,7 @@ import (
_ "go.ipao.vip/atom"
"go.ipao.vip/atom/contracts"
"go.ipao.vip/gen/types"
"go.uber.org/dig"
)
@@ -96,5 +98,85 @@ func (t *TenantTestSuite) Test_SetUserRole() {
So(len(m.Role), ShouldEqual, 1)
So(m.Role[0], ShouldEqual, consts.TenantUserRoleTenantAdmin)
})
Convey("设置为 member 成功", func() {
err := Tenant.SetUserRole(ctx, tenantID, userID, consts.TenantUserRoleMember)
So(err, ShouldBeNil)
m, err := Tenant.FindTenantUser(ctx, tenantID, userID)
So(err, ShouldBeNil)
So(m, ShouldNotBeNil)
So(len(m.Role), ShouldEqual, 1)
So(m.Role[0], ShouldEqual, consts.TenantUserRoleMember)
})
})
}
func (t *TenantTestSuite) Test_RemoveUser() {
Convey("Tenant.RemoveUser", t.T(), func() {
ctx := t.T().Context()
tenantID := int64(1)
userID := int64(2)
database.Truncate(ctx, t.DB, models.TableNameTenantUser)
Convey("移除不存在成员应幂等返回成功", func() {
So(Tenant.RemoveUser(ctx, tenantID, userID), ShouldBeNil)
})
Convey("移除已存在成员成功", func() {
So(Tenant.AddUser(ctx, tenantID, userID), ShouldBeNil)
So(Tenant.RemoveUser(ctx, tenantID, userID), ShouldBeNil)
_, err := Tenant.FindTenantUser(ctx, tenantID, userID)
So(err, ShouldNotBeNil)
})
})
}
func (t *TenantTestSuite) Test_AdminTenantUsersPage() {
Convey("Tenant.AdminTenantUsersPage", t.T(), func() {
ctx := t.T().Context()
tenantID := int64(1)
database.Truncate(ctx, t.DB, models.TableNameTenantUser, models.TableNameUser)
u1 := &models.User{
Username: "u1",
Password: "pw",
Roles: types.NewArray([]consts.Role{consts.RoleUser}),
Status: consts.UserStatusVerified,
}
So(u1.Create(ctx), ShouldBeNil)
u2 := &models.User{
Username: "u2",
Password: "pw",
Roles: types.NewArray([]consts.Role{consts.RoleUser}),
Status: consts.UserStatusVerified,
}
So(u2.Create(ctx), ShouldBeNil)
So(Tenant.AddUser(ctx, tenantID, u1.ID), ShouldBeNil)
So(Tenant.AddUser(ctx, tenantID, u2.ID), ShouldBeNil)
So(Tenant.SetUserRole(ctx, tenantID, u2.ID, consts.TenantUserRoleTenantAdmin), ShouldBeNil)
Convey("不加过滤应返回用户信息", func() {
pager, err := Tenant.AdminTenantUsersPage(ctx, tenantID, &tenantdto.AdminTenantUserListFilter{})
So(err, ShouldBeNil)
So(pager.Total, ShouldEqual, 2)
items, ok := pager.Items.([]*tenantdto.AdminTenantUserItem)
So(ok, ShouldBeTrue)
So(len(items), ShouldEqual, 2)
So(items[0].User, ShouldNotBeNil)
})
Convey("按 role=tenant_admin 过滤", func() {
role := consts.TenantUserRoleTenantAdmin
pager, err := Tenant.AdminTenantUsersPage(ctx, tenantID, &tenantdto.AdminTenantUserListFilter{Role: &role})
So(err, ShouldBeNil)
So(pager.Total, ShouldEqual, 1)
})
})
}