Compare commits

...

4 Commits

Author SHA1 Message Date
5b4d2afc5a 修复验证码发送权限问题 2025-08-09 22:12:36 +08:00
3daffdf323 修复登录,注册的bug 2025-08-09 01:22:54 +08:00
927fab3bdd 修复some bug 2025-08-08 19:26:21 +08:00
f0f2b779f8 完成微信支付以外的所有功能 2025-08-08 16:55:37 +08:00
12 changed files with 458 additions and 378 deletions

View File

@ -109,6 +109,7 @@ public class CourseController {
public BaseResponse<List<CourseCardVO>> miniQueryCourseByKeyword(@Valid @RequestBody CommonStringRequest commonStringRequest) {
String keyword = commonStringRequest.getTemplateString();
LambdaQueryWrapper<Course> lambdaQueryWrapper = new LambdaQueryWrapper<>();
lambdaQueryWrapper.eq(Course::getIsShelves, true);
lambdaQueryWrapper.like(Course::getName, keyword);
List<Course> courseList = courseService.list(lambdaQueryWrapper);
List<CourseCardVO> courseCardVOS = commonService.convertList(courseList, CourseCardVO.class);

View File

@ -18,11 +18,9 @@ import com.greenorange.promotion.model.dto.CommonRequest;
import com.greenorange.promotion.model.dto.courseOrder.CourseOrderAddRequest;
import com.greenorange.promotion.model.dto.courseOrder.CourseOrderQueryRequest;
import com.greenorange.promotion.model.dto.courseOrder.CourseOrderUpdateRequest;
import com.greenorange.promotion.model.entity.Course;
import com.greenorange.promotion.model.entity.CourseOrder;
import com.greenorange.promotion.model.entity.CoursePromotionCommissionPending;
import com.greenorange.promotion.model.entity.UserPerformanceSummary;
import com.greenorange.promotion.model.entity.*;
import com.greenorange.promotion.model.enums.CommissionStatusEnum;
import com.greenorange.promotion.model.enums.UserRoleEnum;
import com.greenorange.promotion.model.vo.course.CourseCardVO;
import com.greenorange.promotion.model.vo.course.CourseVO;
import com.greenorange.promotion.model.vo.courseOrder.CourseOrderBaseInfoVO;
@ -93,6 +91,10 @@ public class CourseOrderController {
@SysLog(title = "课程订单管理", content = "小程序端用户生成课程订单")
public BaseResponse<Long> addCourseOrder(@Valid @RequestBody CourseOrderAddRequest courseOrderAddRequest, HttpServletRequest request) {
Long userId = (Long) request.getAttribute("userId");
UserInfo userInfo = userInfoService.getById(userId);
String userRole = userInfo.getUserRole();
UserRoleEnum userRoleEnum = UserRoleEnum.getEnumByValue(userRole);
ThrowUtils.throwIf(!UserRoleEnum.USER.equals(userRoleEnum), ErrorCode.NO_AUTH_ERROR, "只有普通用户才能创建订单");
Long courseId = courseOrderAddRequest.getCourseId();
Course course = courseService.getById(courseId);
ThrowUtils.throwIf(course == null, ErrorCode.OPERATION_ERROR, "该课程不存在");

View File

@ -1,72 +1,72 @@
package com.greenorange.promotion.controller.course;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.greenorange.promotion.annotation.RequiresPermission;
import com.greenorange.promotion.annotation.SysLog;
import com.greenorange.promotion.common.BaseResponse;
import com.greenorange.promotion.common.ResultUtils;
import com.greenorange.promotion.constant.UserConstant;
import com.greenorange.promotion.model.dto.CommonBatchRequest;
import com.greenorange.promotion.model.dto.refundRecord.RefundRecordAddRequest;
import com.greenorange.promotion.model.dto.refundRecord.RefundRecordQueryRequest;
import com.greenorange.promotion.model.dto.refundRecord.RefundRecordUpdateRequest;
import com.greenorange.promotion.model.entity.RefundRecord;
import com.greenorange.promotion.model.vo.refundRecord.RefundRecordVO;
import com.greenorange.promotion.service.common.CommonService;
import com.greenorange.promotion.service.refund.RefundRecordService;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.tags.Tag;
import jakarta.annotation.Resource;
import lombok.extern.slf4j.Slf4j;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.greenorange.promotion.model.dto.CommonRequest;
import jakarta.validation.Valid;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import java.util.List;
/**
* 退款记录 控制器
*/
@RestController
@RequestMapping("refundRecord")
@Slf4j
@Tag(name = "退款记录模块")
@Transactional
public class RefundRecordController {
@Resource
private RefundRecordService refundRecordService;
@Resource
private CommonService commonService;
/**
* Web端管理员分页查询退款记录
* @param refundRecordQueryRequest 退款记录查询请求体
* @return 退款记录列表
*/
@PostMapping("page")
@Operation(summary = "Web端管理员分页查询退款记录", description = "参数退款记录查询请求体权限管理员方法名listRefundRecordByPage")
@RequiresPermission(mustRole = UserConstant.ADMIN_ROLE)
@SysLog(title = "退款记录管理", content = "Web端管理员分页查询退款记录")
public BaseResponse<Page<RefundRecordVO>> listRefundRecordByPage(@Valid @RequestBody RefundRecordQueryRequest refundRecordQueryRequest) {
long current = refundRecordQueryRequest.getCurrent();
long pageSize = refundRecordQueryRequest.getPageSize();
QueryWrapper<RefundRecord> queryWrapper = refundRecordService.getQueryWrapper(refundRecordQueryRequest);
Page<RefundRecord> page = refundRecordService.page(new Page<>(current, pageSize), queryWrapper);
List<RefundRecord> refundRecordList = page.getRecords();
List<RefundRecordVO> refundRecordVOList = commonService.convertList(refundRecordList, RefundRecordVO.class);
Page<RefundRecordVO> voPage = new Page<>(current, pageSize);
voPage.setRecords(refundRecordVOList);
voPage.setPages(page.getPages());
voPage.setTotal(page.getTotal());
return ResultUtils.success(voPage);
}
}
//package com.greenorange.promotion.controller.course;
//
//import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
//import com.greenorange.promotion.annotation.RequiresPermission;
//import com.greenorange.promotion.annotation.SysLog;
//import com.greenorange.promotion.common.BaseResponse;
//import com.greenorange.promotion.common.ResultUtils;
//import com.greenorange.promotion.constant.UserConstant;
//import com.greenorange.promotion.model.dto.CommonBatchRequest;
//import com.greenorange.promotion.model.dto.refundRecord.RefundRecordAddRequest;
//import com.greenorange.promotion.model.dto.refundRecord.RefundRecordQueryRequest;
//import com.greenorange.promotion.model.dto.refundRecord.RefundRecordUpdateRequest;
//import com.greenorange.promotion.model.entity.RefundRecord;
//import com.greenorange.promotion.model.vo.refundRecord.RefundRecordVO;
//import com.greenorange.promotion.service.common.CommonService;
//import com.greenorange.promotion.service.refund.RefundRecordService;
//import io.swagger.v3.oas.annotations.Operation;
//import io.swagger.v3.oas.annotations.tags.Tag;
//import jakarta.annotation.Resource;
//import lombok.extern.slf4j.Slf4j;
//import org.springframework.transaction.annotation.Transactional;
//import org.springframework.web.bind.annotation.PostMapping;
//import org.springframework.web.bind.annotation.RequestBody;
//import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
//import com.greenorange.promotion.model.dto.CommonRequest;
//import jakarta.validation.Valid;
//import org.springframework.web.bind.annotation.RequestMapping;
//import org.springframework.web.bind.annotation.RestController;
//
//import java.util.List;
//
//
///**
// * 退款记录 控制器
// */
//@RestController
//@RequestMapping("refundRecord")
//@Slf4j
//@Tag(name = "退款记录模块")
//@Transactional
//public class RefundRecordController {
//
// @Resource
// private RefundRecordService refundRecordService;
//
// @Resource
// private CommonService commonService;
//
//
// /**
// * Web端管理员分页查询退款记录
// * @param refundRecordQueryRequest 退款记录查询请求体
// * @return 退款记录列表
// */
// @PostMapping("page")
// @Operation(summary = "Web端管理员分页查询退款记录", description = "参数退款记录查询请求体权限管理员方法名listRefundRecordByPage")
// @RequiresPermission(mustRole = UserConstant.ADMIN_ROLE)
// @SysLog(title = "退款记录管理", content = "Web端管理员分页查询退款记录")
// public BaseResponse<Page<RefundRecordVO>> listRefundRecordByPage(@Valid @RequestBody RefundRecordQueryRequest refundRecordQueryRequest) {
// long current = refundRecordQueryRequest.getCurrent();
// long pageSize = refundRecordQueryRequest.getPageSize();
// QueryWrapper<RefundRecord> queryWrapper = refundRecordService.getQueryWrapper(refundRecordQueryRequest);
// Page<RefundRecord> page = refundRecordService.page(new Page<>(current, pageSize), queryWrapper);
// List<RefundRecord> refundRecordList = page.getRecords();
// List<RefundRecordVO> refundRecordVOList = commonService.convertList(refundRecordList, RefundRecordVO.class);
// Page<RefundRecordVO> voPage = new Page<>(current, pageSize);
// voPage.setRecords(refundRecordVOList);
// voPage.setPages(page.getPages());
// voPage.setTotal(page.getTotal());
// return ResultUtils.success(voPage);
// }
//}

View File

@ -37,7 +37,7 @@ import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import java.util.List;
import java.util.UUID;
/**

View File

@ -5,6 +5,7 @@ import cn.binarywang.wx.miniapp.bean.WxMaQrcode;
import com.auth0.jwt.interfaces.DecodedJWT;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.greenorange.promotion.annotation.RequiresPermission;
import com.greenorange.promotion.annotation.SysLog;
@ -97,33 +98,48 @@ public class UserInfoController {
// }
/**
* 小程序端用户修改用户头像
* @param commonStringRequest 头像view值
* @return 是否修改成功
*/
@PostMapping("modify/avatar")
@Operation(summary = "小程序端用户修改用户头像", description = "参数头像view值权限管理员boss, admin)方法名modifyUserAvatar")
@RequiresPermission(mustRole = UserConstant.DEFAULT_ROLE)
public BaseResponse<Boolean> modifyUserAvatar(@Valid @RequestBody CommonStringRequest commonStringRequest, HttpServletRequest request) {
Long userId = (Long) request.getAttribute("userId");
String view = commonStringRequest.getTemplateString();
LambdaUpdateWrapper<UserInfo> updateWrapper = new LambdaUpdateWrapper<>();
updateWrapper.eq(UserInfo::getId, userId).set(UserInfo::getUserAvatar, view);
userInfoService.update(updateWrapper);
return ResultUtils.success(true);
}
/**
* 小程序端用户获取验证码(用于注册)
* @param commonStringRequest 手机号
* @param verificationCodeGetRequest 验证码获取请求体
* @return 验证码
*/
@PostMapping("code/register")
@Operation(summary = "小程序端用户获取验证码(用于注册)", description = "参数手机号权限管理员boss, admin)方法名getVerificationCodeForRegister")
public BaseResponse<String> getVerificationCodeForRegister(@Valid @RequestBody CommonStringRequest commonStringRequest) {
String phoneNumber = commonStringRequest.getTemplateString();
String verificationCode = userInfoService.getVerificationCodeForRegister(phoneNumber);
public BaseResponse<String> getVerificationCodeForRegister(@Valid @RequestBody VerificationCodeGetRequest verificationCodeGetRequest) {
String verificationCode = userInfoService.getVerificationCodeForRegister(verificationCodeGetRequest);
return ResultUtils.success(verificationCode);
}
/**
* 小程序端用户获取验证码(用于密码登录和忘记密码)
* @param commonStringRequest 手机号
* @param verificationCodeGetRequest 验证码获取请求体
* @return 验证码
*/
@PostMapping("code/pwd")
@Operation(summary = "小程序端用户获取验证码(用于密码登录和忘记密码)", description = "参数手机号权限管理员boss, admin)方法名getVerificationCode")
// @SysLog(title = "用户管理", content = "小程序端用户获取验证码")
public BaseResponse<String> getVerificationCode(@Valid @RequestBody CommonStringRequest commonStringRequest) {
String phoneNumber = commonStringRequest.getTemplateString();
String verificationCode = userInfoService.getVerificationCodeForPwdLogin(phoneNumber);
public BaseResponse<String> getVerificationCode(@Valid @RequestBody VerificationCodeGetRequest verificationCodeGetRequest) {
String verificationCode = userInfoService.getVerificationCodeForPwdLogin(verificationCodeGetRequest);
return ResultUtils.success(verificationCode);
}

View File

@ -0,0 +1,36 @@
package com.greenorange.promotion.model.dto.userInfo;
import com.greenorange.promotion.annotation.EnumValue;
import com.greenorange.promotion.model.enums.UserRoleEnum;
import io.swagger.v3.oas.annotations.media.Schema;
import jakarta.validation.constraints.NotBlank;
import lombok.Data;
import java.io.Serial;
import java.io.Serializable;
/**
* 验证码发送请求体
*/
@Data
@Schema(description = "验证码发送请求体", requiredProperties = {"phoneNumber", "userRole"})
public class VerificationCodeGetRequest implements Serializable {
/**
* 手机号
*/
@NotBlank(message = "手机号不能为空")
@Schema(description = "手机号", example = "15888610253")
private String phoneNumber;
/**
* 权限
*/
@NotBlank(message = "权限不能为空")
@EnumValue(enumClass = UserRoleEnum.class)
@Schema(description = "权限", example = "user")
private String userRole;
@Serial
private static final long serialVersionUID = 1L;
}

View File

@ -58,13 +58,13 @@ public interface UserInfoService extends IService<UserInfo> {
/**
* 小程序用户获取验证码(用于密码登录和忘记密码)
*/
String getVerificationCodeForPwdLogin(String phoneNumber);
String getVerificationCodeForPwdLogin(VerificationCodeGetRequest verificationCodeGetRequest);
/**
* 小程序用户获取验证码(用于注册)
*/
String getVerificationCodeForRegister(String phoneNumber);
String getVerificationCodeForRegister(VerificationCodeGetRequest verificationCodeGetRequest);
/**

View File

@ -162,12 +162,20 @@ public class UserInfoServiceImpl extends ServiceImpl<UserInfoMapper, UserInfo>
@Override
@Transactional(rollbackFor = Exception.class)
public void userInfoMiniRegister(UserInfoRegisterRequest userInfoRegisterRequest) {
String nickName = userInfoRegisterRequest.getNickName();
String phoneNumber = userInfoRegisterRequest.getPhoneNumber();
String verificationCode = userInfoRegisterRequest.getVerificationCode();
String userRole = userInfoRegisterRequest.getUserRole();
UserRoleEnum userRoleEnum = UserRoleEnum.getEnumByValue(userRole);
// 校验用户手机号和验证码
checkPhoneAndVerificationCode(phoneNumber, verificationCode, userRoleEnum);
// 如果注册主管,校验昵称是否重复
if (UserRoleEnum.SUPERVISOR.equals(userRoleEnum)) {
LambdaQueryWrapper<UserInfo> lambdaQueryWrapper = new LambdaQueryWrapper<>();
lambdaQueryWrapper.eq(UserInfo::getNickName, nickName).eq(UserInfo::getUserRole, UserConstant.SUPERVISOR_ROLE);
UserInfo userInfo = this.getOne(lambdaQueryWrapper);
ThrowUtils.throwIf(userInfo != null, ErrorCode.OPERATION_ERROR, "昵称重复");
}
// 根据邀请码获得上级用户信息
String invitationCode = userInfoRegisterRequest.getInvitationCode();
@ -220,15 +228,26 @@ public class UserInfoServiceImpl extends ServiceImpl<UserInfoMapper, UserInfo>
ThrowUtils.throwIf(RegexUtils.isPhoneInvalid(phoneNumber), ErrorCode.PARAMS_ERROR, "手机号格式无效");
String userPassword = userInfoMiniPasswordLoginRequest.getUserPassword();
String userRole = userInfoMiniPasswordLoginRequest.getUserRole();
UserRoleEnum userRoleEnum = UserRoleEnum.getEnumByValue(userRole);
LambdaQueryWrapper<UserInfo> lambdaQueryWrapper = new LambdaQueryWrapper<>();
lambdaQueryWrapper.eq(UserInfo::getPhoneNumber, phoneNumber);
lambdaQueryWrapper.eq(UserInfo::getUserRole, userRole);
UserInfo userInfo = this.getOne(lambdaQueryWrapper);
ThrowUtils.throwIf(userInfo == null, ErrorCode.OPERATION_ERROR, "手机号未注册");
if (UserRoleEnum.USER.equals(userRoleEnum)) {
lambdaQueryWrapper.eq(UserInfo::getUserRole, userRole);
lambdaQueryWrapper.eq(UserInfo::getPhoneNumber, phoneNumber);
UserInfo userInfo = this.getOne(lambdaQueryWrapper);
ThrowUtils.throwIf(userInfo == null, ErrorCode.OPERATION_ERROR, "手机号未注册");
} else {
lambdaQueryWrapper.eq(UserInfo::getPhoneNumber, phoneNumber);
lambdaQueryWrapper.in(UserInfo::getUserRole, UserRoleEnum.STAFF.getValue(), UserRoleEnum.SUPERVISOR.getValue(), UserRoleEnum.MANAGER.getValue());
UserInfo userInfo = this.getOne(lambdaQueryWrapper);
ThrowUtils.throwIf(userInfo == null, ErrorCode.OPERATION_ERROR, "手机号未注册");
UserRoleEnum currentUserRoleEnum = UserRoleEnum.getEnumByValue(userInfo.getUserRole());
ThrowUtils.throwIf(!userRoleEnum.equals(currentUserRoleEnum), ErrorCode.OPERATION_ERROR, "该手机号为" + currentUserRoleEnum.getText() + "账号");
}
lambdaQueryWrapper.eq(UserInfo::getUserPassword, userPassword);
userInfo = this.getOne(lambdaQueryWrapper);
UserInfo userInfo = this.getOne(lambdaQueryWrapper);
ThrowUtils.throwIf(userInfo == null, ErrorCode.OPERATION_ERROR, "密码不正确");
Map<String, String> payload = new HashMap<>();
payload.put("userAccount", phoneNumber);
payload.put("userPassword", userPassword);
@ -301,12 +320,14 @@ public class UserInfoServiceImpl extends ServiceImpl<UserInfoMapper, UserInfo>
* 小程序用户获取验证码(用于密码登录和忘记密码)
*/
@Override
public String getVerificationCodeForPwdLogin(String phoneNumber) {
public String getVerificationCodeForPwdLogin(VerificationCodeGetRequest verificationCodeGetRequest) {
String phoneNumber = verificationCodeGetRequest.getPhoneNumber();
String userRole = verificationCodeGetRequest.getUserRole();
ThrowUtils.throwIf(RegexUtils.isPhoneInvalid(phoneNumber), ErrorCode.PARAMS_ERROR, "手机号格式错误");
// 判断手机号是否已注册
LambdaQueryWrapper<UserInfo> lambdaQueryWrapper = new LambdaQueryWrapper<>();
lambdaQueryWrapper.eq(UserInfo::getPhoneNumber, phoneNumber);
lambdaQueryWrapper.eq(UserInfo::getPhoneNumber, phoneNumber).eq(UserInfo::getUserRole, userRole);
UserInfo userInfo = this.getOne(lambdaQueryWrapper);
ThrowUtils.throwIf(userInfo == null, ErrorCode.OPERATION_ERROR, "手机号未注册");
@ -321,12 +342,14 @@ public class UserInfoServiceImpl extends ServiceImpl<UserInfoMapper, UserInfo>
* 小程序用户获取验证码(用于注册)
*/
@Override
public String getVerificationCodeForRegister(String phoneNumber) {
public String getVerificationCodeForRegister(VerificationCodeGetRequest verificationCodeGetRequest) {
String phoneNumber = verificationCodeGetRequest.getPhoneNumber();
String userRole = verificationCodeGetRequest.getUserRole();
ThrowUtils.throwIf(RegexUtils.isPhoneInvalid(phoneNumber), ErrorCode.PARAMS_ERROR, "手机号格式错误");
// 判断手机号是否已注册
LambdaQueryWrapper<UserInfo> lambdaQueryWrapper = new LambdaQueryWrapper<>();
lambdaQueryWrapper.eq(UserInfo::getPhoneNumber, phoneNumber);
lambdaQueryWrapper.eq(UserInfo::getPhoneNumber, phoneNumber).eq(UserInfo::getUserRole, userRole);
UserInfo userInfo = this.getOne(lambdaQueryWrapper);
ThrowUtils.throwIf(userInfo != null, ErrorCode.OPERATION_ERROR, "手机号已注册");
@ -348,7 +371,9 @@ public class UserInfoServiceImpl extends ServiceImpl<UserInfoMapper, UserInfo>
phoneNumberLambdaQueryWrapper.eq(UserInfo::getPhoneNumber, phoneNumber);
phoneNumberLambdaQueryWrapper = getQueryWrapperByUserRole(userRoleEnum, phoneNumberLambdaQueryWrapper);
UserInfo userInfo = this.getOne(phoneNumberLambdaQueryWrapper);
ThrowUtils.throwIf(userInfo != null, ErrorCode.OPERATION_ERROR, "手机号已注册");
String userRole = userInfo.getUserRole();
UserRoleEnum currentUserRoleEnum = UserRoleEnum.getEnumByValue(userRole);
ThrowUtils.throwIf(userInfo != null, ErrorCode.OPERATION_ERROR, "该手机号为"+ currentUserRoleEnum.getText() +"账号");
}
String code = redisTemplate.opsForValue().get(SystemConstant.VERIFICATION_CODE + ":" + verificationCode);
ThrowUtils.throwIf(code == null, ErrorCode.OPERATION_ERROR, "验证码已失效");
@ -408,7 +433,7 @@ public class UserInfoServiceImpl extends ServiceImpl<UserInfoMapper, UserInfo>
// 生成邀请二维码
String invitationQrcode = generateInvitationQrcode(myUserInfo.getInvitationCode(), UserRoleEnum.STAFF);
UserMainInfo userMainInfo = UserMainInfo.builder().userId(userId).inviteQrCode(invitationQrcode).build();
UserMainInfo userMainInfo = UserMainInfo.builder().userId(myUserInfo.getId()).inviteQrCode(invitationQrcode).build();
userMainInfoService.save(userMainInfo);
// // 批量保存当前用户的项目明细抽佣记录和下级用户项目明细抽佣记录

View File

@ -33,8 +33,8 @@ public class SendSmsUtil {
// 设置请求参数
List<NameValuePair> data = new ArrayList<>();
data.add(new BasicNameValuePair("account", "C08121984"));
data.add(new BasicNameValuePair("password", "84a27a879413ec629bf26c5d84a25271"));
data.add(new BasicNameValuePair("account", "C55991947"));
data.add(new BasicNameValuePair("password", "d3979496a8d9c7e9a322804b7ed187e2"));
data.add(new BasicNameValuePair("mobile", phoneNumber));
data.add(new BasicNameValuePair("content", content));

View File

@ -1,79 +1,79 @@
package com.greenorange.promotion.junit;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.greenorange.promotion.model.dto.CommonBatchRequest;
import com.greenorange.promotion.model.entity.PromoCode;
import com.greenorange.promotion.service.project.impl.PromoCodeServiceImpl;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.ExtendWith;
import org.mockito.InjectMocks;
import org.mockito.Spy;
import org.mockito.junit.jupiter.MockitoExtension;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;
import static org.junit.jupiter.api.Assertions.assertThrows;
import static org.junit.jupiter.api.Assertions.assertTrue;
import static org.mockito.Mockito.*;
@ExtendWith(MockitoExtension.class)
class PromoCodeServiceImplTest {
@Spy
@InjectMocks
private PromoCodeServiceImpl service;
// Spy + InjectMocks使用真实的 PromoCodeServiceImpl但可以 stub 它的 list(...) 和 removeByIds(...)
/**
* 当存在状态为“占用”的推广码时,应抛出异常并且不执行删除
*/
@Test
void delBatchPromoCode_whenCodesInUse_shouldThrow() {
// Arrange
List<Long> ids = Arrays.asList(1L, 2L, 3L);
CommonBatchRequest req = new CommonBatchRequest();
req.setIds(ids);
// 模拟 list(...) 返回一个非空列表,表示有正在使用的推广码
PromoCode inUse = new PromoCode();
inUse.setId(2L);
inUse.setPromoCodeStatus(true);
doReturn(Collections.singletonList(inUse))
.when(service).list(any(LambdaQueryWrapper.class));
// Act & Assert
RuntimeException ex = assertThrows(RuntimeException.class,
() -> service.delBatchPromoCode(req));
assertTrue(ex.getMessage().contains("当前推广码正在使用中"));
// 验证 removeByIds(...) 从未被调用
verify(service, never()).removeByIds(anyList());
}
/**
* 当所有推广码都未被占用时,应正常调用 removeByIds 删除
*/
@Test
void delBatchPromoCode_whenNoCodesInUse_shouldRemove() {
// Arrange
List<Long> ids = Arrays.asList(4L, 5L);
CommonBatchRequest req = new CommonBatchRequest();
req.setIds(ids);
// 模拟 list(...) 返回空列表,表示无任何占用的推广码
doReturn(Collections.emptyList())
.when(service).list(any(LambdaQueryWrapper.class));
// 模拟 removeByIds(...) 返回 true表示删除成功
doReturn(true).when(service).removeByIds(ids);
// Act
service.delBatchPromoCode(req);
// Assert
// 验证 removeByIds(...) 被调用一次,且参数正是传入的 ids
verify(service, times(1)).removeByIds(ids);
}
}
//package com.greenorange.promotion.junit;
//
//import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
//import com.greenorange.promotion.model.dto.CommonBatchRequest;
//import com.greenorange.promotion.model.entity.PromoCode;
//import com.greenorange.promotion.service.project.impl.PromoCodeServiceImpl;
//import org.junit.jupiter.api.Test;
//import org.junit.jupiter.api.extension.ExtendWith;
//import org.mockito.InjectMocks;
//import org.mockito.Spy;
//import org.mockito.junit.jupiter.MockitoExtension;
//
//import java.util.Arrays;
//import java.util.Collections;
//import java.util.List;
//
//import static org.junit.jupiter.api.Assertions.assertThrows;
//import static org.junit.jupiter.api.Assertions.assertTrue;
//import static org.mockito.Mockito.*;
//
//@ExtendWith(MockitoExtension.class)
//class PromoCodeServiceImplTest {
//
// @Spy
// @InjectMocks
// private PromoCodeServiceImpl service;
// // Spy + InjectMocks使用真实的 PromoCodeServiceImpl但可以 stub 它的 list(...) 和 removeByIds(...)
//
// /**
// * 当存在状态为“占用”的推广码时,应抛出异常并且不执行删除
// */
// @Test
// void delBatchPromoCode_whenCodesInUse_shouldThrow() {
// // Arrange
// List<Long> ids = Arrays.asList(1L, 2L, 3L);
// CommonBatchRequest req = new CommonBatchRequest();
// req.setIds(ids);
//
// // 模拟 list(...) 返回一个非空列表,表示有正在使用的推广码
// PromoCode inUse = new PromoCode();
// inUse.setId(2L);
// inUse.setPromoCodeStatus(true);
// doReturn(Collections.singletonList(inUse))
// .when(service).list(any(LambdaQueryWrapper.class));
//
// // Act & Assert
// RuntimeException ex = assertThrows(RuntimeException.class,
// () -> service.delBatchPromoCode(req));
// assertTrue(ex.getMessage().contains("当前推广码正在使用中"));
//
// // 验证 removeByIds(...) 从未被调用
// verify(service, never()).removeByIds(anyList());
// }
//
// /**
// * 当所有推广码都未被占用时,应正常调用 removeByIds 删除
// */
// @Test
// void delBatchPromoCode_whenNoCodesInUse_shouldRemove() {
// // Arrange
// List<Long> ids = Arrays.asList(4L, 5L);
// CommonBatchRequest req = new CommonBatchRequest();
// req.setIds(ids);
//
// // 模拟 list(...) 返回空列表,表示无任何占用的推广码
// doReturn(Collections.emptyList())
// .when(service).list(any(LambdaQueryWrapper.class));
//
// // 模拟 removeByIds(...) 返回 true表示删除成功
// doReturn(true).when(service).removeByIds(ids);
//
// // Act
// service.delBatchPromoCode(req);
//
// // Assert
// // 验证 removeByIds(...) 被调用一次,且参数正是传入的 ids
// verify(service, times(1)).removeByIds(ids);
// }
//}

View File

@ -1,96 +1,96 @@
package com.greenorange.promotion.junit;
import com.greenorange.promotion.model.dto.userAccount.UserAccountUpdateRequest;
import com.greenorange.promotion.model.entity.UserAccount;
import com.greenorange.promotion.service.common.CommonService;
import com.greenorange.promotion.service.settle.impl.UserAccountServiceImpl;
import jakarta.servlet.http.HttpServletRequest;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.ExtendWith;
import org.mockito.ArgumentCaptor;
import org.mockito.InjectMocks;
import org.mockito.Mock;
import org.mockito.Spy;
import org.mockito.junit.jupiter.MockitoExtension;
import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertSame;
import static org.mockito.Mockito.*;
@ExtendWith(MockitoExtension.class)
class UserAccountServiceImplTest {
@Spy
@InjectMocks
private UserAccountServiceImpl userAccountService;
// Spy + InjectMocks用真实的 serviceImpl但可以对它的方法做部分 stub
@Mock
private CommonService commonService;
// Mock CommonService用于模拟 copyProperties
@Mock
private HttpServletRequest request;
// Mock HttpServletRequest用于模拟获取 userId
@Test
void updateUserAccount_shouldCopyProperties_setUserId_andUpdateById() {
// --- Arrange 准备阶段 ---
Long userId = 456L;
// 模拟从 request 中拿到当前登录用户 ID
when(request.getAttribute("userId")).thenReturn(userId);
// 构造更新请求 DTO并设置要更新的账户 ID
UserAccountUpdateRequest req = new UserAccountUpdateRequest();
req.setId(99L);
req.setCardHolder("李四");
req.setIdCardNumber("110101199002022345");
req.setPhoneNumber("15900001111");
req.setBankCardNumber("6222020202020202");
req.setOpenBank("中国农业银行");
// 准备一个空实体,模拟 commonService.copyProperties 拷贝结果
UserAccount stubEntity = new UserAccount();
// 假设 copyProperties 会拷贝所有字段,包括 id
stubEntity.setId(req.getId());
stubEntity.setCardHolder(req.getCardHolder());
stubEntity.setIdCardNumber(req.getIdCardNumber());
stubEntity.setPhoneNumber(req.getPhoneNumber());
stubEntity.setBankCardNumber(req.getBankCardNumber());
stubEntity.setOpenBank(req.getOpenBank());
// stub copyProperties 返回我们准备的 stubEntity
when(commonService.copyProperties(req, UserAccount.class))
.thenReturn(stubEntity);
// 对 Spy 的 updateById(...) 方法做 stub避免走到 MyBatis-Plus 真逻辑
doReturn(true).when(userAccountService).updateById(any(UserAccount.class));
// --- Act 执行阶段 ---
userAccountService.updateUserAccount(req, request);
// 方法内部执行顺序:
// 1. 取 request.getAttribute("userId") -> 456L
// 2. commonService.copyProperties(req, UserAccount.class) -> stubEntity
// 3. stubEntity.setUserId(456L)
// 4. 调用 updateById(stubEntity)
// --- Assert 验证阶段 ---
// 捕获 updateById 调用时传入的参数
ArgumentCaptor<UserAccount> captor = ArgumentCaptor.forClass(UserAccount.class);
verify(userAccountService).updateById(captor.capture());
UserAccount updated = captor.getValue();
// 验证传给 updateById 的正是 stubEntity 对象
assertSame(stubEntity, updated, "应该传入同一个 stubEntity 实例");
// 验证 userId 已正确赋值
assertEquals(userId, updated.getUserId(), "userId 应该从 request 中取出并赋值");
// 验证其他字段都被 copyProperties 拷贝过来,包括账户 ID
assertEquals(req.getId(), updated.getId(), "账户 ID 应保持一致");
assertEquals("李四", updated.getCardHolder(), "持卡人应一致");
assertEquals("110101199002022345", updated.getIdCardNumber(), "身份证号应一致");
assertEquals("15900001111", updated.getPhoneNumber(), "手机号应一致");
assertEquals("6222020202020202", updated.getBankCardNumber(),"银行卡号应一致");
assertEquals("中国农业银行", updated.getOpenBank(), "开户银行应一致");
}
}
//package com.greenorange.promotion.junit;
//
//import com.greenorange.promotion.model.dto.userAccount.UserAccountUpdateRequest;
//import com.greenorange.promotion.model.entity.UserAccount;
//import com.greenorange.promotion.service.common.CommonService;
//import com.greenorange.promotion.service.settle.impl.UserAccountServiceImpl;
//import jakarta.servlet.http.HttpServletRequest;
//import org.junit.jupiter.api.Test;
//import org.junit.jupiter.api.extension.ExtendWith;
//import org.mockito.ArgumentCaptor;
//import org.mockito.InjectMocks;
//import org.mockito.Mock;
//import org.mockito.Spy;
//import org.mockito.junit.jupiter.MockitoExtension;
//
//import static org.junit.jupiter.api.Assertions.assertEquals;
//import static org.junit.jupiter.api.Assertions.assertSame;
//import static org.mockito.Mockito.*;
//
//@ExtendWith(MockitoExtension.class)
//class UserAccountServiceImplTest {
//
// @Spy
// @InjectMocks
// private UserAccountServiceImpl userAccountService;
// // Spy + InjectMocks用真实的 serviceImpl但可以对它的方法做部分 stub
//
// @Mock
// private CommonService commonService;
// // Mock CommonService用于模拟 copyProperties
//
// @Mock
// private HttpServletRequest request;
// // Mock HttpServletRequest用于模拟获取 userId
//
// @Test
// void updateUserAccount_shouldCopyProperties_setUserId_andUpdateById() {
// // --- Arrange 准备阶段 ---
// Long userId = 456L;
// // 模拟从 request 中拿到当前登录用户 ID
// when(request.getAttribute("userId")).thenReturn(userId);
//
// // 构造更新请求 DTO并设置要更新的账户 ID
// UserAccountUpdateRequest req = new UserAccountUpdateRequest();
// req.setId(99L);
// req.setCardHolder("李四");
// req.setIdCardNumber("110101199002022345");
// req.setPhoneNumber("15900001111");
// req.setBankCardNumber("6222020202020202");
// req.setOpenBank("中国农业银行");
//
// // 准备一个空实体,模拟 commonService.copyProperties 拷贝结果
// UserAccount stubEntity = new UserAccount();
// // 假设 copyProperties 会拷贝所有字段,包括 id
// stubEntity.setId(req.getId());
// stubEntity.setCardHolder(req.getCardHolder());
// stubEntity.setIdCardNumber(req.getIdCardNumber());
// stubEntity.setPhoneNumber(req.getPhoneNumber());
// stubEntity.setBankCardNumber(req.getBankCardNumber());
// stubEntity.setOpenBank(req.getOpenBank());
// // stub copyProperties 返回我们准备的 stubEntity
// when(commonService.copyProperties(req, UserAccount.class))
// .thenReturn(stubEntity);
//
// // 对 Spy 的 updateById(...) 方法做 stub避免走到 MyBatis-Plus 真逻辑
// doReturn(true).when(userAccountService).updateById(any(UserAccount.class));
//
// // --- Act 执行阶段 ---
// userAccountService.updateUserAccount(req, request);
// // 方法内部执行顺序:
// // 1. 取 request.getAttribute("userId") -> 456L
// // 2. commonService.copyProperties(req, UserAccount.class) -> stubEntity
// // 3. stubEntity.setUserId(456L)
// // 4. 调用 updateById(stubEntity)
//
// // --- Assert 验证阶段 ---
// // 捕获 updateById 调用时传入的参数
// ArgumentCaptor<UserAccount> captor = ArgumentCaptor.forClass(UserAccount.class);
// verify(userAccountService).updateById(captor.capture());
// UserAccount updated = captor.getValue();
//
// // 验证传给 updateById 的正是 stubEntity 对象
// assertSame(stubEntity, updated, "应该传入同一个 stubEntity 实例");
//
// // 验证 userId 已正确赋值
// assertEquals(userId, updated.getUserId(), "userId 应该从 request 中取出并赋值");
//
// // 验证其他字段都被 copyProperties 拷贝过来,包括账户 ID
// assertEquals(req.getId(), updated.getId(), "账户 ID 应保持一致");
// assertEquals("李四", updated.getCardHolder(), "持卡人应一致");
// assertEquals("110101199002022345", updated.getIdCardNumber(), "身份证号应一致");
// assertEquals("15900001111", updated.getPhoneNumber(), "手机号应一致");
// assertEquals("6222020202020202", updated.getBankCardNumber(),"银行卡号应一致");
// assertEquals("中国农业银行", updated.getOpenBank(), "开户银行应一致");
// }
//}

View File

@ -1,103 +1,103 @@
package com.greenorange.promotion.junit;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.greenorange.promotion.model.dto.userInfo.UserInfoMiniPasswordLoginRequest;
import com.greenorange.promotion.model.entity.UserInfo;
import com.greenorange.promotion.service.userInfo.impl.UserInfoServiceImpl;
import com.greenorange.promotion.utils.JWTUtils;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.ExtendWith;
import org.mockito.*;
import org.mockito.junit.jupiter.MockitoExtension;
import java.util.Map;
import static org.junit.jupiter.api.Assertions.*;
import static org.mockito.ArgumentMatchers.any;
import static org.mockito.Mockito.*;
@ExtendWith(MockitoExtension.class)
class UserInfoServiceImplTest {
@Spy
@InjectMocks
private UserInfoServiceImpl userService; // 真实的 ServiceImpl部分方法用 spy
@Mock
private JWTUtils jwtUtils; // 用来生成 token 的工具
@BeforeEach
void setUp() {
// MockitoAnnotations.openMocks(this); // @ExtendWith 已自动初始化
}
private UserInfoMiniPasswordLoginRequest buildRequest(String phone, String pwd) {
UserInfoMiniPasswordLoginRequest req = new UserInfoMiniPasswordLoginRequest();
req.setPhoneNumber(phone);
req.setUserPassword(pwd);
return req;
}
/** 1. 手机号格式无效,应直接抛参数错误 */
@Test
void givenInvalidPhone_whenLoginByPwd_thenThrow() {
UserInfoMiniPasswordLoginRequest req = buildRequest("not-a-phone", "whatever");
assertThrows(RuntimeException.class, () -> userService.userInfoMiniLoginByPwd(req));
}
/** 2. 手机号未注册,应抛操作错误 */
@Test
void givenUnregisteredPhone_whenLoginByPwd_thenThrow() {
// stub 第一次 getOne(...) 返回 null
doReturn(null)
.when(userService).getOne(any(LambdaQueryWrapper.class));
UserInfoMiniPasswordLoginRequest req = buildRequest("13812345678", "pwd");
assertThrows(RuntimeException.class, () -> userService.userInfoMiniLoginByPwd(req));
}
/** 3. 密码不正确,应抛操作错误 */
@Test
void givenWrongPassword_whenLoginByPwd_thenThrow() {
UserInfo dummy = new UserInfo();
dummy.setPhoneNumber("13812345678");
dummy.setUserPassword("rightPwd");
// stub 第一次 getOne(...) 返回非 null手机号存在
// stub 第二次 getOne(...) 返回 null密码校验失败
doReturn(dummy, null)
.when(userService).getOne(any(LambdaQueryWrapper.class));
UserInfoMiniPasswordLoginRequest req = buildRequest("13812345678", "wrongPwd");
assertThrows(RuntimeException.class, () -> userService.userInfoMiniLoginByPwd(req));
}
/** 4. 成功场景:正确手机号 + 密码,返回 token并验证 jwtUtils 调用 */
@Test
void givenValidCredentials_whenLoginByPwd_thenReturnToken() {
UserInfo dummy = new UserInfo();
dummy.setPhoneNumber("13812345678");
dummy.setUserPassword("rightPwd");
// stub getOne(...) 两次都返回同一个 dummy表示手机号存在且密码正确
doReturn(dummy, dummy)
.when(userService).getOne(any(LambdaQueryWrapper.class));
// stub jwtUtils.generateToken(...)
String expectedToken = "jwt-token-xyz";
when(jwtUtils.generateToken(anyMap())).thenReturn(expectedToken);
UserInfoMiniPasswordLoginRequest req = buildRequest("13812345678", "rightPwd");
String token = userService.userInfoMiniLoginByPwd(req);
assertEquals(expectedToken, token);
// 验证 payload 内容正确
ArgumentCaptor<Map<String, String>> captor = ArgumentCaptor.forClass(Map.class);
verify(jwtUtils, times(1)).generateToken(captor.capture());
Map<String, String> payload = captor.getValue();
assertEquals("13812345678", payload.get("userAccount"));
assertEquals("rightPwd", payload.get("userPassword"));
}
}
//package com.greenorange.promotion.junit;
//
//import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
//import com.greenorange.promotion.model.dto.userInfo.UserInfoMiniPasswordLoginRequest;
//import com.greenorange.promotion.model.entity.UserInfo;
//import com.greenorange.promotion.service.userInfo.impl.UserInfoServiceImpl;
//import com.greenorange.promotion.utils.JWTUtils;
//import org.junit.jupiter.api.BeforeEach;
//import org.junit.jupiter.api.Test;
//import org.junit.jupiter.api.extension.ExtendWith;
//import org.mockito.*;
//import org.mockito.junit.jupiter.MockitoExtension;
//
//import java.util.Map;
//
//import static org.junit.jupiter.api.Assertions.*;
//import static org.mockito.ArgumentMatchers.any;
//import static org.mockito.Mockito.*;
//
//@ExtendWith(MockitoExtension.class)
//class UserInfoServiceImplTest {
//
// @Spy
// @InjectMocks
// private UserInfoServiceImpl userService; // 真实的 ServiceImpl部分方法用 spy
//
// @Mock
// private JWTUtils jwtUtils; // 用来生成 token 的工具
//
// @BeforeEach
// void setUp() {
// // MockitoAnnotations.openMocks(this); // @ExtendWith 已自动初始化
// }
//
// private UserInfoMiniPasswordLoginRequest buildRequest(String phone, String pwd) {
// UserInfoMiniPasswordLoginRequest req = new UserInfoMiniPasswordLoginRequest();
// req.setPhoneNumber(phone);
// req.setUserPassword(pwd);
// return req;
// }
//
// /** 1. 手机号格式无效,应直接抛参数错误 */
// @Test
// void givenInvalidPhone_whenLoginByPwd_thenThrow() {
// UserInfoMiniPasswordLoginRequest req = buildRequest("not-a-phone", "whatever");
// assertThrows(RuntimeException.class, () -> userService.userInfoMiniLoginByPwd(req));
// }
//
// /** 2. 手机号未注册,应抛操作错误 */
// @Test
// void givenUnregisteredPhone_whenLoginByPwd_thenThrow() {
// // stub 第一次 getOne(...) 返回 null
// doReturn(null)
// .when(userService).getOne(any(LambdaQueryWrapper.class));
//
// UserInfoMiniPasswordLoginRequest req = buildRequest("13812345678", "pwd");
// assertThrows(RuntimeException.class, () -> userService.userInfoMiniLoginByPwd(req));
// }
//
// /** 3. 密码不正确,应抛操作错误 */
// @Test
// void givenWrongPassword_whenLoginByPwd_thenThrow() {
// UserInfo dummy = new UserInfo();
// dummy.setPhoneNumber("13812345678");
// dummy.setUserPassword("rightPwd");
//
// // stub 第一次 getOne(...) 返回非 null手机号存在
// // stub 第二次 getOne(...) 返回 null密码校验失败
// doReturn(dummy, null)
// .when(userService).getOne(any(LambdaQueryWrapper.class));
//
// UserInfoMiniPasswordLoginRequest req = buildRequest("13812345678", "wrongPwd");
// assertThrows(RuntimeException.class, () -> userService.userInfoMiniLoginByPwd(req));
// }
//
// /** 4. 成功场景:正确手机号 + 密码,返回 token并验证 jwtUtils 调用 */
// @Test
// void givenValidCredentials_whenLoginByPwd_thenReturnToken() {
// UserInfo dummy = new UserInfo();
// dummy.setPhoneNumber("13812345678");
// dummy.setUserPassword("rightPwd");
//
// // stub getOne(...) 两次都返回同一个 dummy表示手机号存在且密码正确
// doReturn(dummy, dummy)
// .when(userService).getOne(any(LambdaQueryWrapper.class));
//
// // stub jwtUtils.generateToken(...)
// String expectedToken = "jwt-token-xyz";
// when(jwtUtils.generateToken(anyMap())).thenReturn(expectedToken);
//
// UserInfoMiniPasswordLoginRequest req = buildRequest("13812345678", "rightPwd");
// String token = userService.userInfoMiniLoginByPwd(req);
//
// assertEquals(expectedToken, token);
//
// // 验证 payload 内容正确
// ArgumentCaptor<Map<String, String>> captor = ArgumentCaptor.forClass(Map.class);
// verify(jwtUtils, times(1)).generateToken(captor.capture());
// Map<String, String> payload = captor.getValue();
// assertEquals("13812345678", payload.get("userAccount"));
// assertEquals("rightPwd", payload.get("userPassword"));
// }
//}