createUserLogic_mock_test.go 1.6 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849
  1. package user
  2. import (
  3. "context"
  4. "errors"
  5. "testing"
  6. "perms-system-server/internal/response"
  7. "perms-system-server/internal/testutil/ctxhelper"
  8. "perms-system-server/internal/testutil/mocks"
  9. "perms-system-server/internal/types"
  10. "github.com/go-sql-driver/mysql"
  11. "github.com/stretchr/testify/assert"
  12. "github.com/stretchr/testify/require"
  13. "github.com/zeromicro/go-zero/core/stores/sqlx"
  14. "go.uber.org/mock/gomock"
  15. )
  16. // TC-0144: 唯一索引冲突消息(Insert层1062错误)——H-R17-1 后 CreateUser 把 Insert
  17. // 收进 SysUserModel.TransactCtx 闭包,mock 路径必须同步切到 TransactCtx + InsertWithTx,
  18. // 否则 gomock 在 "Unexpected call to Insert" 上就把契约保真性覆盖掉。
  19. func TestCreateUser_Mock_InsertDuplicate1062(t *testing.T) {
  20. ctrl := gomock.NewController(t)
  21. defer ctrl.Finish()
  22. mockUser := mocks.NewMockSysUserModel(ctrl)
  23. mockUser.EXPECT().TransactCtx(gomock.Any(), gomock.Any()).DoAndReturn(
  24. func(ctx context.Context, fn func(context.Context, sqlx.Session) error) error {
  25. return fn(ctx, nil)
  26. })
  27. mockUser.EXPECT().InsertWithTx(gomock.Any(), gomock.Any(), gomock.Any()).
  28. Return(nil, &mysql.MySQLError{Number: 1062, Message: "Duplicate entry 'dupuser' for key 'username'"})
  29. svcCtx := mocks.NewMockServiceContext(mocks.MockModels{
  30. User: mockUser,
  31. })
  32. logic := NewCreateUserLogic(ctxhelper.SuperAdminCtx(), svcCtx)
  33. _, err := logic.CreateUser(&types.CreateUserReq{
  34. Username: "dupuser",
  35. })
  36. require.Error(t, err)
  37. var codeErr *response.CodeError
  38. require.True(t, errors.As(err, &codeErr))
  39. assert.Equal(t, 409, codeErr.Code())
  40. assert.Equal(t, "用户名已存在", codeErr.Error())
  41. }