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

package role

import (
	"context"
	"errors"
	"testing"

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

	"github.com/stretchr/testify/assert"
	"github.com/zeromicro/go-zero/core/stores/sqlx"
	"go.uber.org/mock/gomock"
)

// TC-0100: 事务回滚
func TestBindRolePerms_Mock_BatchInsertFail(t *testing.T) {
	ctrl := gomock.NewController(t)
	defer ctrl.Finish()

	dbErr := errors.New("db error")

	mockRole := mocks.NewMockSysRoleModel(ctrl)
	mockRole.EXPECT().FindOne(gomock.Any(), int64(1)).
		Return(&roleModel.SysRole{Id: 1}, nil)

	mockRP := mocks.NewMockSysRolePermModel(ctrl)
	mockRP.EXPECT().TransactCtx(gomock.Any(), gomock.Any()).
		DoAndReturn(func(ctx context.Context, fn func(context.Context, sqlx.Session) error) error {
			return fn(ctx, nil)
		})
	mockRP.EXPECT().DeleteByRoleIdTx(gomock.Any(), nil, int64(1)).Return(nil)
	mockRP.EXPECT().BatchInsertWithTx(gomock.Any(), nil, gomock.Any()).Return(dbErr)

	svcCtx := mocks.NewMockServiceContext(mocks.MockModels{
		Role:     mockRole,
		RolePerm: mockRP,
	})

	logic := NewBindRolePermsLogic(ctxhelper.SuperAdminCtx(), svcCtx)
	err := logic.BindRolePerms(&types.BindPermsReq{
		RoleId:  1,
		PermIds: []int64{10, 20},
	})

	assert.Error(t, err)
	assert.ErrorIs(t, err, dbErr)
}