perms-system-server/internal/logic/role/roleListLogic_test.go

package role

import (
	"testing"
	"time"

	roleModel "perms-system-server/internal/model/role"
	"perms-system-server/internal/svc"
	"perms-system-server/internal/testutil"
	"perms-system-server/internal/testutil/ctxhelper"
	"perms-system-server/internal/types"

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

// TC-0122: 正常查询
func TestRoleList_Normal(t *testing.T) {
	ctx := ctxhelper.SuperAdminCtx()
	svcCtx := svc.NewServiceContext(testutil.GetTestConfig())
	conn := testutil.GetTestSqlConn()
	now := time.Now().Unix()
	pc := testutil.UniqueId()

	var roleIds []int64
	for i := 0; i < 3; i++ {
		res, err := svcCtx.SysRoleModel.Insert(ctx, &roleModel.SysRole{
			ProductCode: pc, Name: testutil.UniqueId(), Status: 1,
			PermsLevel: int64(i + 1), CreateTime: now, UpdateTime: now,
		})
		require.NoError(t, err)
		id, _ := res.LastInsertId()
		roleIds = append(roleIds, id)
	}

	t.Cleanup(func() {
		testutil.CleanTable(ctx, conn, "`sys_role`", roleIds...)
	})

	logic := NewRoleListLogic(ctx, svcCtx)
	resp, err := logic.RoleList(&types.RoleListReq{
		ProductCode: pc, Page: 1, PageSize: 10,
	})
	require.NoError(t, err)
	assert.Equal(t, int64(3), resp.Total)

	items := resp.List.([]types.RoleItem)
	assert.Len(t, items, 3)
}

// TC-0122: 正常查询
func TestRoleList_DefaultPagination(t *testing.T) {
	ctx := ctxhelper.SuperAdminCtx()
	svcCtx := svc.NewServiceContext(testutil.GetTestConfig())
	conn := testutil.GetTestSqlConn()
	now := time.Now().Unix()
	pc := testutil.UniqueId()

	res, err := svcCtx.SysRoleModel.Insert(ctx, &roleModel.SysRole{
		ProductCode: pc, Name: testutil.UniqueId(), Status: 1, PermsLevel: 1,
		CreateTime: now, UpdateTime: now,
	})
	require.NoError(t, err)
	roleId, _ := res.LastInsertId()

	t.Cleanup(func() {
		testutil.CleanTable(ctx, conn, "`sys_role`", roleId)
	})

	logic := NewRoleListLogic(ctx, svcCtx)
	resp, err := logic.RoleList(&types.RoleListReq{
		ProductCode: pc,
	})
	require.NoError(t, err)
	assert.Equal(t, int64(1), resp.Total)
}

// TC-0123: pageSize超过上限
func TestRoleList_PageSizeExceedsLimit(t *testing.T) {
	ctx := ctxhelper.SuperAdminCtx()
	svcCtx := svc.NewServiceContext(testutil.GetTestConfig())
	conn := testutil.GetTestSqlConn()
	now := time.Now().Unix()
	pc := testutil.UniqueId()

	res, err := svcCtx.SysRoleModel.Insert(ctx, &roleModel.SysRole{
		ProductCode: pc, Name: testutil.UniqueId(), Status: 1, PermsLevel: 1,
		CreateTime: now, UpdateTime: now,
	})
	require.NoError(t, err)
	roleId, _ := res.LastInsertId()

	t.Cleanup(func() {
		testutil.CleanTable(ctx, conn, "`sys_role`", roleId)
	})

	logic := NewRoleListLogic(ctx, svcCtx)
	resp, err := logic.RoleList(&types.RoleListReq{
		ProductCode: pc, Page: 1, PageSize: 200,
	})
	require.NoError(t, err)
	assert.Equal(t, int64(1), resp.Total)
}

// TC-1308: 非超管不传productCode时从JWT获取
func TestRoleList_NonSuperAdminUsesJWTProductCode(t *testing.T) {
	svcCtx := svc.NewServiceContext(testutil.GetTestConfig())
	conn := testutil.GetTestSqlConn()
	now := time.Now().Unix()
	pc := testutil.UniqueId()

	res, err := svcCtx.SysRoleModel.Insert(ctxhelper.SuperAdminCtx(), &roleModel.SysRole{
		ProductCode: pc, Name: testutil.UniqueId(), Status: 1, PermsLevel: 5,
		CreateTime: now, UpdateTime: now,
	})
	require.NoError(t, err)
	roleId, _ := res.LastInsertId()

	t.Cleanup(func() {
		testutil.CleanTable(ctxhelper.SuperAdminCtx(), conn, "`sys_role`", roleId)
	})

	ctx := ctxhelper.AdminCtx(pc)
	logic := NewRoleListLogic(ctx, svcCtx)
	resp, err := logic.RoleList(&types.RoleListReq{
		Page:     1,
		PageSize: 10,
	})
	require.NoError(t, err)
	assert.Equal(t, int64(1), resp.Total)
}

// TC-1309: 非超管传了其他产品code时被忽略
func TestRoleList_NonSuperAdminIgnoresReqProductCode(t *testing.T) {
	svcCtx := svc.NewServiceContext(testutil.GetTestConfig())
	conn := testutil.GetTestSqlConn()
	now := time.Now().Unix()
	pc := testutil.UniqueId()

	res, err := svcCtx.SysRoleModel.Insert(ctxhelper.SuperAdminCtx(), &roleModel.SysRole{
		ProductCode: pc, Name: testutil.UniqueId(), Status: 1, PermsLevel: 5,
		CreateTime: now, UpdateTime: now,
	})
	require.NoError(t, err)
	roleId, _ := res.LastInsertId()

	t.Cleanup(func() {
		testutil.CleanTable(ctxhelper.SuperAdminCtx(), conn, "`sys_role`", roleId)
	})

	ctx := ctxhelper.AdminCtx(pc)
	logic := NewRoleListLogic(ctx, svcCtx)
	resp, err := logic.RoleList(&types.RoleListReq{
		ProductCode: "other_product",
		Page:        1,
		PageSize:    10,
	})
	require.NoError(t, err)
	assert.Equal(t, int64(1), resp.Total)
}

// TC-1310: 超管不传productCode时返回全量数据
func TestRoleList_SuperAdminNoProductCodeReturnsAll(t *testing.T) {
	ctx := ctxhelper.SuperAdminCtx()
	svcCtx := svc.NewServiceContext(testutil.GetTestConfig())
	conn := testutil.GetTestSqlConn()
	now := time.Now().Unix()

	pc1 := testutil.UniqueId()
	pc2 := testutil.UniqueId()

	r1Res, err := svcCtx.SysRoleModel.Insert(ctx, &roleModel.SysRole{
		ProductCode: pc1, Name: testutil.UniqueId(), Status: 1, PermsLevel: 1,
		CreateTime: now, UpdateTime: now,
	})
	require.NoError(t, err)
	r1Id, _ := r1Res.LastInsertId()

	r2Res, err := svcCtx.SysRoleModel.Insert(ctx, &roleModel.SysRole{
		ProductCode: pc2, Name: testutil.UniqueId(), Status: 1, PermsLevel: 2,
		CreateTime: now, UpdateTime: now,
	})
	require.NoError(t, err)
	r2Id, _ := r2Res.LastInsertId()

	t.Cleanup(func() {
		testutil.CleanTable(ctx, conn, "`sys_role`", r1Id, r2Id)
	})

	logic := NewRoleListLogic(ctx, svcCtx)
	resp, err := logic.RoleList(&types.RoleListReq{
		Page:     1,
		PageSize: 10000,
	})
	require.NoError(t, err)
	assert.GreaterOrEqual(t, resp.Total, int64(2))
}