庄锐波
5 months ago
12 changed files with 599 additions and 614 deletions
@ -0,0 +1,95 @@ |
|||
package com.zdxt.rest.test.cp; |
|||
|
|||
import com.zdxt.enforcementcode.domain.EfcodeCpReserve; |
|||
import com.zdxt.enforcementcode.service.IEfcodeCpPlanService; |
|||
import org.dromara.ZdxtAdminApplication; |
|||
import org.junit.jupiter.api.*; |
|||
import org.springframework.beans.factory.annotation.Autowired; |
|||
import org.springframework.boot.test.context.SpringBootTest; |
|||
import org.springframework.test.context.ActiveProfiles; |
|||
|
|||
import java.util.List; |
|||
import java.util.Map; |
|||
|
|||
/** |
|||
* 计划自动匹配预约测试 |
|||
* <p> |
|||
* 强制调用系统匹配方法，验证待预约计划的自动联合匹配逻辑。 |
|||
* 匹配条件：同一企业 + 一周内 + 不同执法单位 ≥ 2 个单位的待预约计划 |
|||
* </p> |
|||
*/ |
|||
@SpringBootTest(classes = ZdxtAdminApplication.class) |
|||
@ActiveProfiles({"dev", "dev-cust-zrb"}) |
|||
@DisplayName("计划自动匹配预约测试") |
|||
public class AppEnfCpAutoReserveTest { |
|||
|
|||
@Autowired |
|||
private IEfcodeCpPlanService cpPlanService; |
|||
|
|||
/** |
|||
* 强制执行系统自动匹配（干跑模式：不更新计划、不插入预约记录） |
|||
* <p>只计算匹配结果，不修改数据库，用于验证匹配逻辑是否正确</p> |
|||
*/ |
|||
@DisplayName("干跑模式 - 强制系统匹配（不落库）") |
|||
@Test |
|||
@Order(1) |
|||
public void testAutoMatch_dryRun() { |
|||
System.out.println("========== 开始执行自动匹配预约（干跑模式） =========="); |
|||
|
|||
// forceMatch=true: 跳过日期校验
|
|||
// updatePlan=false: 不更新计划状态
|
|||
// insertReserve=false: 不插入预约记录
|
|||
Map<String, List<EfcodeCpReserve>> result = cpPlanService.autoMatchReserveV2(true, false, false); |
|||
|
|||
System.out.println("匹配结果组数: " + result.size()); |
|||
for (Map.Entry<String, List<EfcodeCpReserve>> entry : result.entrySet()) { |
|||
String planGroupId = entry.getKey(); |
|||
List<EfcodeCpReserve> reserves = entry.getValue(); |
|||
System.out.println(" planGroupId=" + planGroupId + ", 联合计划数=" + reserves.size()); |
|||
for (EfcodeCpReserve r : reserves) { |
|||
System.out.println(" - planId=" + r.getPlanId() |
|||
+ ", receiverId=" + r.getReceiverId() |
|||
+ ", opStatus=" + r.getOpStatus()); |
|||
} |
|||
} |
|||
|
|||
System.out.println("========== 干跑模式执行完毕 =========="); |
|||
} |
|||
|
|||
/** |
|||
* 强制执行系统自动匹配（实际执行：更新计划 + 插入预约记录） |
|||
* <p>将匹配结果落库，计划状态变为“预约中(2)”，生成预约记录</p> |
|||
*/ |
|||
@DisplayName("实际执行 - 强制系统匹配（落库）") |
|||
@Test |
|||
@Order(2) |
|||
public void testAutoMatch_execute() { |
|||
System.out.println("========== 开始执行自动匹配预约（实际落库） =========="); |
|||
|
|||
// forceMatch=true: 跳过日期校验
|
|||
// updatePlan=true: 更新计划状态(system_reserved=1, reserve_status=2)
|
|||
// insertReserve=true: 插入预约记录到数据库
|
|||
Map<String, List<EfcodeCpReserve>> result = cpPlanService.autoMatchReserveV2(true, true, true); |
|||
|
|||
System.out.println("匹配结果组数: " + result.size()); |
|||
int totalReserves = 0; |
|||
for (Map.Entry<String, List<EfcodeCpReserve>> entry : result.entrySet()) { |
|||
String planGroupId = entry.getKey(); |
|||
List<EfcodeCpReserve> reserves = entry.getValue(); |
|||
totalReserves += reserves.size(); |
|||
System.out.println(" planGroupId=" + planGroupId + ", 联合计划数=" + reserves.size()); |
|||
for (EfcodeCpReserve r : reserves) { |
|||
System.out.println(" - planId=" + r.getPlanId() |
|||
+ ", receiverId=" + r.getReceiverId() |
|||
+ ", opStatus=" + r.getOpStatus() |
|||
+ ", opSetDate=" + r.getOpSetDate()); |
|||
} |
|||
} |
|||
|
|||
System.out.println("总计生成预约记录数: " + totalReserves); |
|||
System.out.println("========== 实际落库执行完毕 =========="); |
|||
|
|||
// 验证：如果之前有多个账号创建了同一企业的计划，应能匹配到
|
|||
Assertions.assertFalse(result.isEmpty(), "应至少匹配到一组联合预约"); |
|||
} |
|||
} |
|||
@ -1,220 +0,0 @@ |
|||
package com.zdxt.rest.test.cp; |
|||
|
|||
import cn.dev33.satoken.context.mock.SaTokenContextMockUtil; |
|||
import cn.dev33.satoken.stp.StpUtil; |
|||
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; |
|||
import com.baomidou.mybatisplus.core.toolkit.Wrappers; |
|||
import com.zdxt.enforcementcode.domain.EfcodeCpPlan; |
|||
import com.zdxt.enforcementcode.domain.EfcodeCpReserve; |
|||
import com.zdxt.enforcementcode.mapper.EfcodeCpPlanMapper; |
|||
import com.zdxt.enforcementcode.mapper.EfcodeCpReserveMapper; |
|||
import org.dromara.ZdxtAdminApplication; |
|||
import org.junit.jupiter.api.*; |
|||
import org.junit.jupiter.api.Tag; |
|||
import org.springframework.beans.factory.annotation.Autowired; |
|||
import org.springframework.boot.test.autoconfigure.web.servlet.AutoConfigureMockMvc; |
|||
import org.springframework.boot.test.context.SpringBootTest; |
|||
import org.springframework.mock.web.MockHttpServletRequest; |
|||
import org.springframework.test.context.ActiveProfiles; |
|||
import org.springframework.web.context.request.RequestContextHolder; |
|||
import org.springframework.web.context.request.ServletRequestAttributes; |
|||
|
|||
import java.util.*; |
|||
import java.util.stream.Collectors; |
|||
|
|||
/** |
|||
* 预约记录批量生成测试 |
|||
* <p> |
|||
* 基于已创建的计划数据，为多个用户交替创建 efcode_cp_reserve 预约记录。 |
|||
* 模拟场景：用户A的计划被用户B预约，用户B的计划被用户C预约...循环交替。 |
|||
* 预约状态交替：待确认(3)、已同意(1)、已拒绝(2) |
|||
* 预约来源交替：手动预约(0)、系统匹配(1) |
|||
* </p> |
|||
*/ |
|||
@SpringBootTest(classes = ZdxtAdminApplication.class) |
|||
@AutoConfigureMockMvc |
|||
@ActiveProfiles({"dev", "dev-cust-zrb"}) |
|||
@DisplayName("AppEnfCpReserve 批量生成预约记录测试") |
|||
@TestMethodOrder(MethodOrderer.OrderAnnotation.class) |
|||
@Tag("dev") |
|||
public class AppEnfCpReserveTest { |
|||
|
|||
@Autowired |
|||
private EfcodeCpPlanMapper cpPlanMapper; |
|||
|
|||
@Autowired |
|||
private EfcodeCpReserveMapper cpReserveMapper; |
|||
|
|||
/** |
|||
* 测试账号对应的 userId（需从数据库匹配） |
|||
* 这里通过查询 plan 表的 create_by 字段获取对应用户ID |
|||
*/ |
|||
private static final String[] TEST_ACCOUNTS = {"liuqianlin", "baining", "caizifan", "chenchong4"}; |
|||
|
|||
/** 每个用户最多取多少条计划来生成预约 */ |
|||
private static final int MAX_PLANS_PER_USER = 50; |
|||
|
|||
/** 预约留言模板 */ |
|||
private static final String[] MESSAGES = { |
|||
"希望联合执法，请确认时间", |
|||
"我单位有相关检查需求，申请联合", |
|||
"建议合并检查，减轻企业负担", |
|||
"时间安排一致，申请联合执法", |
|||
"", // 部分无留言
|
|||
"请尽快确认预约", |
|||
"", |
|||
"我方已做好准备，等待确认" |
|||
}; |
|||
|
|||
@BeforeEach |
|||
public void setUp() { |
|||
SaTokenContextMockUtil.setMockContext(); |
|||
MockHttpServletRequest mockRequest = new MockHttpServletRequest(); |
|||
mockRequest.addHeader("User-Agent", "Mozilla/5.0 (Windows NT 10.0; Win64; x64)"); |
|||
RequestContextHolder.setRequestAttributes(new ServletRequestAttributes(mockRequest)); |
|||
} |
|||
|
|||
@AfterEach |
|||
public void tearDown() { |
|||
try { |
|||
StpUtil.logout(); |
|||
} catch (Exception ignored) { |
|||
} |
|||
RequestContextHolder.resetRequestAttributes(); |
|||
SaTokenContextMockUtil.clearContext(); |
|||
} |
|||
|
|||
@DisplayName("基于已有计划批量生成交替预约记录") |
|||
@Test |
|||
@Order(1) |
|||
public void testBatchCreateReserves() { |
|||
// 1. 收集每个用户的计划（通过 enforce_user_names LIKE 账号中文名识别）
|
|||
// 实际通过 create_by 字段分组获取不同用户的计划
|
|||
Map<Long, List<EfcodeCpPlan>> userPlansMap = loadUserPlansMap(); |
|||
|
|||
List<Long> userIds = new ArrayList<>(userPlansMap.keySet()); |
|||
Assertions.assertTrue(userIds.size() >= 2, "至少需要2个用户的计划数据才能交叉预约"); |
|||
|
|||
System.out.println("========== 检测到 " + userIds.size() + " 个用户的计划数据 =========="); |
|||
for (Long uid : userIds) { |
|||
System.out.println(" userId=" + uid + ", 计划数=" + userPlansMap.get(uid).size()); |
|||
} |
|||
|
|||
Random random = new Random(42); // 固定种子，可重复
|
|||
int totalCreated = 0; |
|||
int reserveIndex = 0; |
|||
|
|||
// 2. 交替为每个用户的计划创建来自其他用户的预约
|
|||
for (int ownerIdx = 0; ownerIdx < userIds.size(); ownerIdx++) { |
|||
Long ownerUserId = userIds.get(ownerIdx); |
|||
List<EfcodeCpPlan> ownerPlans = userPlansMap.get(ownerUserId); |
|||
|
|||
// 限制数量
|
|||
int planCount = Math.min(ownerPlans.size(), MAX_PLANS_PER_USER); |
|||
|
|||
for (int planIdx = 0; planIdx < planCount; planIdx++) { |
|||
EfcodeCpPlan targetPlan = ownerPlans.get(planIdx); |
|||
|
|||
// 选择预约发起者：循环选取其他用户
|
|||
int reserverIdx = (ownerIdx + 1 + planIdx) % userIds.size(); |
|||
if (reserverIdx == ownerIdx) { |
|||
reserverIdx = (reserverIdx + 1) % userIds.size(); |
|||
} |
|||
Long reserverUserId = userIds.get(reserverIdx); |
|||
List<EfcodeCpPlan> reserverPlans = userPlansMap.get(reserverUserId); |
|||
|
|||
// 找预约者匹配的计划（同日期+同时段+同企业），如果没有就用预约者的任意一条计划
|
|||
EfcodeCpPlan reserverPlan = findMatchingPlan(reserverPlans, targetPlan); |
|||
if (reserverPlan == null && !reserverPlans.isEmpty()) { |
|||
reserverPlan = reserverPlans.get(planIdx % reserverPlans.size()); |
|||
} |
|||
if (reserverPlan == null) continue; |
|||
|
|||
// 3. 确定 planGroupId
|
|||
String planGroupId = targetPlan.getId(); |
|||
|
|||
// 4. 交替 opStatus: 3(待确认), 1(已同意), 2(已拒绝), 3, 1, 2...
|
|||
long opStatus = getAlternatingStatus(reserveIndex); |
|||
// 5. 交替 resFrom: 0(手动), 1(系统)
|
|||
long resFrom = reserveIndex % 2; |
|||
// 6. 留言
|
|||
String message = MESSAGES[reserveIndex % MESSAGES.length]; |
|||
|
|||
// 插入目标计划方的记录（计划拥有者）
|
|||
insertReserve(targetPlan.getId(), planGroupId, "", targetPlan.getPlanDate(), |
|||
targetPlan.getPlanTime(), resFrom, opStatus); |
|||
|
|||
// 插入预约发起方的记录
|
|||
insertReserve(reserverPlan.getId(), planGroupId, message, targetPlan.getPlanDate(), |
|||
targetPlan.getPlanTime(), resFrom, 3L); // 发起方始终待确认
|
|||
|
|||
totalCreated += 2; |
|||
reserveIndex++; |
|||
} |
|||
|
|||
System.out.println("[userId=" + ownerUserId + "] 已为其 " + planCount + " 条计划生成预约记录"); |
|||
} |
|||
|
|||
System.out.println("========== 预约记录生成完成 =========="); |
|||
System.out.println("总计插入: " + totalCreated + " 条预约记录"); |
|||
Assertions.assertTrue(totalCreated > 0, "应至少生成1条预约记录"); |
|||
} |
|||
|
|||
/** |
|||
* 从数据库加载各用户的计划，按 create_by 分组 |
|||
* 只取未删除、reserve_status=1（待预约）的计划 |
|||
*/ |
|||
private Map<Long, List<EfcodeCpPlan>> loadUserPlansMap() { |
|||
LambdaQueryWrapper<EfcodeCpPlan> lqw = Wrappers.lambdaQuery(); |
|||
lqw.eq(EfcodeCpPlan::getIsDelete, 0L); |
|||
lqw.eq(EfcodeCpPlan::getReserveStatus, 1L); // 待预约状态
|
|||
lqw.like(EfcodeCpPlan::getEnforceContent, "批量测试执法事项"); // 只取测试数据
|
|||
lqw.orderByAsc(EfcodeCpPlan::getPlanDate); |
|||
List<EfcodeCpPlan> allPlans = cpPlanMapper.selectList(lqw); |
|||
|
|||
return allPlans.stream() |
|||
.filter(p -> p.getCreateBy() != null) |
|||
.collect(Collectors.groupingBy(EfcodeCpPlan::getCreateBy)); |
|||
} |
|||
|
|||
/** |
|||
* 在预约者的计划列表中查找与目标计划匹配的计划（同日期+同时段+同企业） |
|||
*/ |
|||
private EfcodeCpPlan findMatchingPlan(List<EfcodeCpPlan> plans, EfcodeCpPlan target) { |
|||
if (plans == null || plans.isEmpty()) return null; |
|||
for (EfcodeCpPlan p : plans) { |
|||
if (p.getPlanDate() != null && p.getPlanDate().equals(target.getPlanDate()) |
|||
&& Objects.equals(p.getPlanTime(), target.getPlanTime()) |
|||
&& Objects.equals(p.getEnterpriseId(), target.getEnterpriseId())) { |
|||
return p; |
|||
} |
|||
} |
|||
return null; |
|||
} |
|||
|
|||
/** |
|||
* 交替状态: 3, 1, 2, 3, 1, 2... |
|||
*/ |
|||
private long getAlternatingStatus(int index) { |
|||
int mod = index % 3; |
|||
return mod == 0 ? 3L : (mod == 1 ? 1L : 2L); |
|||
} |
|||
|
|||
/** |
|||
* 插入一条预约记录 |
|||
*/ |
|||
private void insertReserve(String planId, String planGroupId, String opMessage, |
|||
Date opSetDate, Long opSetTime, Long resFrom, Long opStatus) { |
|||
EfcodeCpReserve record = new EfcodeCpReserve(); |
|||
record.setPlanId(planId); |
|||
record.setPlanGroupId(planGroupId); |
|||
record.setReserveTime(new Date()); |
|||
record.setOpStatus(opStatus); |
|||
record.setOpOrder(0L); |
|||
record.setOpMessage(opMessage); |
|||
record.setOpSetDate(opSetDate); |
|||
record.setOpSetTime(opSetTime); |
|||
record.setResFrom(resFrom); |
|||
cpReserveMapper.insert(record); |
|||
} |
|||
} |
|||
@ -1,354 +0,0 @@ |
|||
package com.zdxt.rest.test.cp; |
|||
|
|||
import cn.dev33.satoken.context.mock.SaTokenContextMockUtil; |
|||
import cn.dev33.satoken.stp.StpUtil; |
|||
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; |
|||
import com.baomidou.mybatisplus.core.toolkit.Wrappers; |
|||
import com.zdxt.enforcementcode.domain.EfcodeCpPlan; |
|||
import com.zdxt.enforcementcode.domain.EfcodeCpReserve; |
|||
import com.zdxt.enforcementcode.mapper.EfcodeCpPlanMapper; |
|||
import com.zdxt.enforcementcode.mapper.EfcodeCpReserveMapper; |
|||
import com.zdxt.enforcementcode.service.IEfcodeCpPlanService; |
|||
import org.dromara.ZdxtAdminApplication; |
|||
import org.junit.jupiter.api.*; |
|||
import org.junit.jupiter.api.Tag; |
|||
import org.springframework.beans.factory.annotation.Autowired; |
|||
import org.springframework.boot.test.autoconfigure.web.servlet.AutoConfigureMockMvc; |
|||
import org.springframework.boot.test.context.SpringBootTest; |
|||
import org.springframework.mock.web.MockHttpServletRequest; |
|||
import org.springframework.test.context.ActiveProfiles; |
|||
import org.springframework.web.context.request.RequestContextHolder; |
|||
import org.springframework.web.context.request.ServletRequestAttributes; |
|||
|
|||
import java.time.LocalDate; |
|||
import java.time.ZoneId; |
|||
import java.util.*; |
|||
import java.util.stream.Collectors; |
|||
|
|||
/** |
|||
* 计划自动匹配预约 autoMatchReserve 方法测试 |
|||
* <p> |
|||
* 测试场景： |
|||
* 1. 强制模式下执行自动匹配（绕过日期校验） |
|||
* 2. 验证匹配结果：同企业不同单位的计划被联合 |
|||
* 3. 验证计划状态更新：system_reserved=1, reserve_status=2 |
|||
* 4. 验证预约记录生成：相同 plan_group_id 关联多条记录 |
|||
* </p> |
|||
*/ |
|||
@SpringBootTest(classes = ZdxtAdminApplication.class) |
|||
@AutoConfigureMockMvc |
|||
@ActiveProfiles({"dev", "dev-cust-zrb"}) |
|||
@DisplayName("autoMatchReserve 自动匹配预约测试") |
|||
@TestMethodOrder(MethodOrderer.OrderAnnotation.class) |
|||
@Tag("dev") |
|||
public class AutoMatchReserveTest { |
|||
|
|||
@Autowired |
|||
private IEfcodeCpPlanService cpPlanService; |
|||
|
|||
@Autowired |
|||
private EfcodeCpPlanMapper cpPlanMapper; |
|||
|
|||
@Autowired |
|||
private EfcodeCpReserveMapper cpReserveMapper; |
|||
|
|||
@BeforeEach |
|||
public void setUp() { |
|||
SaTokenContextMockUtil.setMockContext(); |
|||
MockHttpServletRequest mockRequest = new MockHttpServletRequest(); |
|||
mockRequest.addHeader("User-Agent", "Mozilla/5.0 (Windows NT 10.0; Win64; x64)"); |
|||
RequestContextHolder.setRequestAttributes(new ServletRequestAttributes(mockRequest)); |
|||
} |
|||
|
|||
@AfterEach |
|||
public void tearDown() { |
|||
try { |
|||
StpUtil.logout(); |
|||
} catch (Exception ignored) { |
|||
} |
|||
RequestContextHolder.resetRequestAttributes(); |
|||
SaTokenContextMockUtil.clearContext(); |
|||
} |
|||
|
|||
@DisplayName("0. 插入1000条适合匹配的计划数据") |
|||
@Test |
|||
@Order(0) |
|||
public void testInsert1000MatchablePlans() { |
|||
// 模拟登录（避免自动填充 createBy 报空）
|
|||
StpUtil.login(1); |
|||
|
|||
// 准备数据：200个企业 × 每企业5个不同执法单位 = 1000条计划
|
|||
int enterpriseCount = 200; |
|||
int unitsPerEnterprise = 5; |
|||
Random random = new Random(); |
|||
|
|||
// 模拟执法单位ID池（确保不同单位）
|
|||
String[] bureauDeptIds = { |
|||
"BUREAU_TEST_A", "BUREAU_TEST_B", "BUREAU_TEST_C", |
|||
"BUREAU_TEST_D", "BUREAU_TEST_E", "BUREAU_TEST_F", |
|||
"BUREAU_TEST_G", "BUREAU_TEST_H" |
|||
}; |
|||
String[] bureauDeptNames = { |
|||
"测试执法局A", "测试执法局B", "测试执法局C", |
|||
"测试执法局D", "测试执法局E", "测试执法局F", |
|||
"测试执法局G", "测试执法局H" |
|||
}; |
|||
|
|||
// 基准日期：从明天开始
|
|||
LocalDate baseDate = LocalDate.now().plusDays(1); |
|||
int insertedCount = 0; |
|||
|
|||
for (int e = 0; e < enterpriseCount; e++) { |
|||
String enterpriseNumber = String.format("91440300TEST%06d", e); |
|||
String enterpriseName = "测试企业" + e + "号"; |
|||
String enterpriseId = "ENT_TEST_" + e; |
|||
|
|||
// 同一企业的计划日期在基准日期附3天内（确保一周内可匹配）
|
|||
Date planDate = Date.from( |
|||
baseDate.plusDays(random.nextInt(3)) |
|||
.atStartOfDay(ZoneId.systemDefault()).toInstant() |
|||
); |
|||
|
|||
for (int u = 0; u < unitsPerEnterprise; u++) { |
|||
int bureauIdx = (e + u) % bureauDeptIds.length; |
|||
|
|||
EfcodeCpPlan plan = new EfcodeCpPlan(); |
|||
plan.setId(UUID.randomUUID().toString().replace("-", "")); |
|||
plan.setPlanType(random.nextBoolean() ? 1L : 2L); |
|||
plan.setPlanDate(planDate); |
|||
plan.setPlanTime((long) (random.nextInt(3) + 1)); // 1=上午,2=下午,3=其他
|
|||
plan.setPlanActualDate(plan.getPlanDate()); |
|||
plan.setPlanActualTime(plan.getPlanTime()); // 1=上午,2=下午,3=其他
|
|||
plan.setEnforceContent("自动匹配测试-执法事项" + e + "-" + u); |
|||
plan.setEnterpriseName(enterpriseName); |
|||
plan.setEnterpriseNumber(enterpriseNumber); |
|||
plan.setEnterpriseId(enterpriseId); |
|||
plan.setEnforceUserNames("测试人员" + u); |
|||
plan.setEnforceUserIds(String.valueOf(1000 + u)); |
|||
plan.setEnforceUserCodes("CODE" + u); |
|||
plan.setEnforceUserTels("1380000" + String.format("%04d", u)); |
|||
plan.setIsPublic(1L); |
|||
plan.setIsUnion(1L); |
|||
plan.setReserveStatus(1L); // 待预约
|
|||
plan.setSystemReserved(0L); // 未经系统匹配
|
|||
plan.setManualReserved(0L); |
|||
plan.setBureauDeptId(bureauDeptIds[bureauIdx]); |
|||
plan.setBureauDeptName(bureauDeptNames[bureauIdx]); |
|||
plan.setOrganId(plan.getBureauDeptId()); |
|||
plan.setOrganName(plan.getBureauDeptName()); |
|||
plan.setDeptId(plan.getBureauDeptId()); |
|||
plan.setDeptName(plan.getBureauDeptName()); |
|||
plan.setIsDelete(0L); |
|||
plan.setCreateBy(1L); |
|||
plan.setCreateTime(new Date()); |
|||
plan.setUpdateBy(1L); |
|||
plan.setUpdateTime(new Date()); |
|||
|
|||
cpPlanMapper.insert(plan); |
|||
insertedCount++; |
|||
} |
|||
} |
|||
|
|||
System.out.println("========== 插入匹配测试数据完成 =========="); |
|||
System.out.println("共插入计划数: " + insertedCount); |
|||
System.out.println("涉及企业数: " + enterpriseCount); |
|||
System.out.println("每企业不同单位数: " + unitsPerEnterprise); |
|||
System.out.println("预期可匹配企业组数: " + enterpriseCount + " (每企业都有>=2个单位)"); |
|||
System.out.println("==========================================="); |
|||
|
|||
Assertions.assertEquals(1000, insertedCount, "应插入恰好1000条计划"); |
|||
} |
|||
|
|||
@DisplayName("1. 查看当前待匹配计划数据概况") |
|||
@Test |
|||
@Order(1) |
|||
public void testCheckPendingPlans() { |
|||
// 查询符合自动匹配条件的计划
|
|||
LambdaQueryWrapper<EfcodeCpPlan> lqw = Wrappers.lambdaQuery(); |
|||
lqw.gt(EfcodeCpPlan::getPlanDate, new Date()); |
|||
lqw.eq(EfcodeCpPlan::getSystemReserved, 0L); |
|||
lqw.eq(EfcodeCpPlan::getReserveStatus, 1L); |
|||
lqw.eq(EfcodeCpPlan::getIsPublic, 1L); |
|||
lqw.eq(EfcodeCpPlan::getIsUnion, 1L); |
|||
lqw.eq(EfcodeCpPlan::getIsDelete, 0L); |
|||
List<EfcodeCpPlan> pendingPlans = cpPlanMapper.selectList(lqw); |
|||
|
|||
System.out.println("========== 待匹配计划数据概况 =========="); |
|||
System.out.println("符合条件的计划总数: " + pendingPlans.size()); |
|||
|
|||
// 按企业信用代码分组统计
|
|||
Map<String, List<EfcodeCpPlan>> groupByEnterprise = pendingPlans.stream() |
|||
.filter(p -> p.getEnterpriseNumber() != null) |
|||
.collect(Collectors.groupingBy(EfcodeCpPlan::getEnterpriseNumber)); |
|||
|
|||
int potentialMatchCount = 0; |
|||
for (Map.Entry<String, List<EfcodeCpPlan>> entry : groupByEnterprise.entrySet()) { |
|||
// 统计不同执法单位数量
|
|||
long distinctUnits = entry.getValue().stream() |
|||
.map(EfcodeCpPlan::getBureauDeptId) |
|||
.distinct() |
|||
.count(); |
|||
if (distinctUnits >= 2) { |
|||
potentialMatchCount++; |
|||
System.out.println(" 企业[" + entry.getKey() + "] - 有" + distinctUnits + "个不同单位的计划，可匹配"); |
|||
} |
|||
} |
|||
|
|||
System.out.println("潜在可匹配的企业数: " + potentialMatchCount); |
|||
System.out.println("======================================="); |
|||
} |
|||
|
|||
@DisplayName("2. 强制执行 autoMatchReserve 并验证结果") |
|||
@Test |
|||
@Order(2) |
|||
public void testAutoMatchReserveForceMode() { |
|||
// 执行前：统计当前系统匹配预约记录数
|
|||
long beforeReserveCount = cpReserveMapper.selectCount( |
|||
Wrappers.<EfcodeCpReserve>lambdaQuery().eq(EfcodeCpReserve::getResFrom, 1L) |
|||
); |
|||
System.out.println("执行前系统匹配预约记录数: " + beforeReserveCount); |
|||
|
|||
// 执行前：统计已被系统预约的计划数
|
|||
long beforeMatchedPlanCount = cpPlanMapper.selectCount( |
|||
Wrappers.<EfcodeCpPlan>lambdaQuery() |
|||
.eq(EfcodeCpPlan::getSystemReserved, 1L) |
|||
.eq(EfcodeCpPlan::getIsDelete, 0L) |
|||
); |
|||
System.out.println("执行前已系统匹配计划数: " + beforeMatchedPlanCount); |
|||
|
|||
// 执行自动匹配（强制模式，忽略日期校验）
|
|||
System.out.println("\n========== 开始执行 autoMatchReserve(true) =========="); |
|||
cpPlanService.autoMatchReserve(true, true, true); |
|||
System.out.println("========== 执行完毕 ==========\n"); |
|||
|
|||
// 执行后：统计系统匹配预约记录数
|
|||
long afterReserveCount = cpReserveMapper.selectCount( |
|||
Wrappers.<EfcodeCpReserve>lambdaQuery().eq(EfcodeCpReserve::getResFrom, 1L) |
|||
); |
|||
System.out.println("执行后系统匹配预约记录数: " + afterReserveCount); |
|||
System.out.println("本次新增预约记录数: " + (afterReserveCount - beforeReserveCount)); |
|||
|
|||
// 执行后：统计已被系统预约的计划数
|
|||
long afterMatchedPlanCount = cpPlanMapper.selectCount( |
|||
Wrappers.<EfcodeCpPlan>lambdaQuery() |
|||
.eq(EfcodeCpPlan::getSystemReserved, 1L) |
|||
.eq(EfcodeCpPlan::getIsDelete, 0L) |
|||
); |
|||
System.out.println("执行后已系统匹配计划数: " + afterMatchedPlanCount); |
|||
System.out.println("本次新增匹配计划数: " + (afterMatchedPlanCount - beforeMatchedPlanCount)); |
|||
|
|||
// 验证：如果有符合条件的数据，则应产生匹配结果
|
|||
if (afterReserveCount > beforeReserveCount) { |
|||
System.out.println("\n[验证通过] 自动匹配产生了新的预约记录"); |
|||
} else { |
|||
System.out.println("\n[提示] 未产生新预约记录（可能没有符合条件的跨单位同企业计划）"); |
|||
} |
|||
} |
|||
|
|||
@DisplayName("3. 验证匹配后的 plan_group_id 分组一致性") |
|||
@Test |
|||
@Order(3) |
|||
public void testVerifyPlanGroupConsistency() { |
|||
// 查询所有系统匹配生成的预约记录
|
|||
LambdaQueryWrapper<EfcodeCpReserve> lqw = Wrappers.lambdaQuery(); |
|||
lqw.eq(EfcodeCpReserve::getResFrom, 1L); |
|||
lqw.eq(EfcodeCpReserve::getOpStatus, 3L); // 待确认
|
|||
lqw.orderByDesc(EfcodeCpReserve::getCreateTime); |
|||
List<EfcodeCpReserve> reserves = cpReserveMapper.selectList(lqw); |
|||
|
|||
if (reserves.isEmpty()) { |
|||
System.out.println("[跳过] 无系统匹配的预约记录，无法验证分组一致性"); |
|||
return; |
|||
} |
|||
|
|||
// 按 plan_group_id 分组
|
|||
Map<String, List<EfcodeCpReserve>> groupMap = reserves.stream() |
|||
.collect(Collectors.groupingBy(EfcodeCpReserve::getPlanGroupId)); |
|||
|
|||
System.out.println("========== plan_group_id 分组验证 =========="); |
|||
System.out.println("共有 " + groupMap.size() + " 组联合预约"); |
|||
|
|||
int validGroups = 0; |
|||
for (Map.Entry<String, List<EfcodeCpReserve>> entry : groupMap.entrySet()) { |
|||
List<EfcodeCpReserve> group = entry.getValue(); |
|||
// 每组至少2条记录（2个不同单位）
|
|||
if (group.size() >= 2) { |
|||
validGroups++; |
|||
|
|||
// 验证同组内关联的计划属于不同执法单位
|
|||
Set<String> planIds = group.stream() |
|||
.map(EfcodeCpReserve::getPlanId) |
|||
.collect(Collectors.toSet()); |
|||
List<EfcodeCpPlan> plans = cpPlanMapper.selectByIds(planIds); |
|||
Set<String> bureauDeptIds = plans.stream() |
|||
.map(EfcodeCpPlan::getBureauDeptId) |
|||
.collect(Collectors.toSet()); |
|||
|
|||
System.out.println(" groupId=" + entry.getKey() |
|||
+ " | 记录数=" + group.size() |
|||
+ " | 涉及单位数=" + bureauDeptIds.size()); |
|||
|
|||
// 断言：同组关联的计划应来自不同单位
|
|||
Assertions.assertTrue(bureauDeptIds.size() >= 2, |
|||
"同一planGroupId下应关联至少2个不同执法单位的计划"); |
|||
} |
|||
} |
|||
System.out.println("有效分组数(>=2条记录): " + validGroups); |
|||
System.out.println("============================================="); |
|||
} |
|||
|
|||
@DisplayName("4. 验证匹配后计划状态已更新") |
|||
@Test |
|||
@Order(4) |
|||
public void testVerifyPlanStatusUpdated() { |
|||
// 查询已被系统匹配的计划
|
|||
LambdaQueryWrapper<EfcodeCpPlan> lqw = Wrappers.lambdaQuery(); |
|||
lqw.eq(EfcodeCpPlan::getSystemReserved, 1L); |
|||
lqw.eq(EfcodeCpPlan::getReserveStatus, 2L); // 预约中
|
|||
lqw.eq(EfcodeCpPlan::getIsDelete, 0L); |
|||
List<EfcodeCpPlan> matchedPlans = cpPlanMapper.selectList(lqw); |
|||
|
|||
System.out.println("========== 计划状态验证 =========="); |
|||
System.out.println("system_reserved=1 且 reserve_status=2 的计划数: " + matchedPlans.size()); |
|||
|
|||
if (!matchedPlans.isEmpty()) { |
|||
// 验证每条计划都满足状态要求
|
|||
for (EfcodeCpPlan plan : matchedPlans) { |
|||
Assertions.assertEquals(1L, plan.getSystemReserved(), |
|||
"计划[" + plan.getId() + "]的system_reserved应为1"); |
|||
Assertions.assertEquals(2L, plan.getReserveStatus(), |
|||
"计划[" + plan.getId() + "]的reserve_status应为2(预约中)"); |
|||
} |
|||
System.out.println("[验证通过] 所有已匹配计划状态正确"); |
|||
} else { |
|||
System.out.println("[提示] 无已匹配计划数据"); |
|||
} |
|||
System.out.println("=================================="); |
|||
} |
|||
|
|||
@DisplayName("5. 再次执行 autoMatchReserve 验证幂等性（不重复匹配）") |
|||
@Test |
|||
@Order(5) |
|||
public void testAutoMatchReserveIdempotent() { |
|||
// 先统计当前预约记录数
|
|||
long beforeCount = cpReserveMapper.selectCount( |
|||
Wrappers.<EfcodeCpReserve>lambdaQuery().eq(EfcodeCpReserve::getResFrom, 1L) |
|||
); |
|||
|
|||
// 再次执行
|
|||
System.out.println("========== 二次执行 autoMatchReserve(true) 验证幂等性 =========="); |
|||
cpPlanService.autoMatchReserve(true, true, true); |
|||
|
|||
long afterCount = cpReserveMapper.selectCount( |
|||
Wrappers.<EfcodeCpReserve>lambdaQuery().eq(EfcodeCpReserve::getResFrom, 1L) |
|||
); |
|||
|
|||
long newRecords = afterCount - beforeCount; |
|||
System.out.println("二次执行新增预约记录数: " + newRecords); |
|||
|
|||
// 已匹配过的计划不应再次匹配（system_reserved已设为1）
|
|||
// 但如果有新的符合条件的数据也可能产生新记录
|
|||
System.out.println("[说明] 已匹配的计划(system_reserved=1)不会被重复匹配"); |
|||
System.out.println("================================================================"); |
|||
} |
|||
} |
|||
Loading…
Reference in new issue