vincent
2 months ago
18 changed files with 26 additions and 965 deletions
@ -1,301 +0,0 @@ |
|||
//package com.zdxt.rest.test;
|
|||
//
|
|||
//import cn.dev33.satoken.context.mock.SaTokenContextMockUtil;
|
|||
//import org.dromara.ZdxtAdminApplication;
|
|||
//import org.junit.jupiter.api.*;
|
|||
//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.http.MediaType;
|
|||
//import org.springframework.test.web.servlet.MockMvc;
|
|||
//import org.springframework.test.web.servlet.MvcResult;
|
|||
//
|
|||
//import java.nio.charset.StandardCharsets;
|
|||
//
|
|||
//import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.post;
|
|||
//import static org.springframework.test.web.servlet.result.MockMvcResultHandlers.print;
|
|||
//import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.*;
|
|||
//
|
|||
///**
|
|||
// * AppEnterpriseLoginController 企业端-用户登录接口测试
|
|||
// * <p>
|
|||
// * 覆盖以下接口：
|
|||
// * 1. POST /app-ent/login/siqLogin - 深i企用户登录
|
|||
// * </p>
|
|||
// * <p>
|
|||
// * 注意：此测试需要配置真实的深i企接口地址，或在测试环境中 Mock 相关服务
|
|||
// * </p>
|
|||
// *
|
|||
// * @author zdxt
|
|||
// */
|
|||
//@SpringBootTest(classes = ZdxtAdminApplication.class)
|
|||
//@AutoConfigureMockMvc
|
|||
//@DisplayName("AppEnterpriseLoginController 企业端-用户登录接口测试")
|
|||
//@TestMethodOrder(MethodOrderer.OrderAnnotation.class)
|
|||
//public class AppEnterpriseLoginControllerTest {
|
|||
//
|
|||
// @Autowired
|
|||
// private MockMvc mockMvc;
|
|||
//
|
|||
// private static final String BASE_URL = "/app-ent/login";
|
|||
// private static final String TEST_CLIENT_ID = "e5cd7e4891bf95d1d19206ce24a7b32e";
|
|||
//
|
|||
// @BeforeEach
|
|||
// public void setUp() {
|
|||
// SaTokenContextMockUtil.setMockContext();
|
|||
// }
|
|||
//
|
|||
// @AfterEach
|
|||
// public void tearDown() {
|
|||
// SaTokenContextMockUtil.clearContext();
|
|||
// }
|
|||
//
|
|||
// // ==================== siqLogin 接口测试 ====================
|
|||
//
|
|||
// @DisplayName("深i企登录 - 正常请求（需要有效的授权码）")
|
|||
// @Test
|
|||
// @Order(1)
|
|||
// public void testSiqLoginSuccess() throws Exception {
|
|||
// String requestBody = "{\"code\":\"test_authorization_code\"}";
|
|||
//
|
|||
// MvcResult result = mockMvc.perform(post(BASE_URL + "/siqLogin")
|
|||
// .contentType(MediaType.APPLICATION_JSON)
|
|||
// .content(requestBody))
|
|||
// .andDo(print())
|
|||
// .andExpect(status().isOk())
|
|||
// .andReturn();
|
|||
//
|
|||
// String responseBody = result.getResponse().getContentAsString(StandardCharsets.UTF_8);
|
|||
// System.out.println("深i企登录接口响应: " + responseBody);
|
|||
//
|
|||
// if (responseBody.contains("\"code\":200")) {
|
|||
// Assertions.assertTrue(responseBody.contains("token"), "返回数据应包含 token");
|
|||
// Assertions.assertTrue(responseBody.contains("userId"), "返回数据应包含 userId");
|
|||
// Assertions.assertTrue(responseBody.contains("userName"), "返回数据应包含 userName");
|
|||
// Assertions.assertTrue(responseBody.contains("enterpriseId"), "返回数据应包含 enterpriseId");
|
|||
// Assertions.assertTrue(responseBody.contains("enterpriseName"), "返回数据应包含 enterpriseName");
|
|||
// System.out.println("深i企登录成功，返回了完整的用户和企业信息");
|
|||
// } else {
|
|||
// System.out.println("深i企登录失败（可能由于授权码无效或深i企接口未配置）: " + responseBody);
|
|||
// }
|
|||
// }
|
|||
//
|
|||
// @DisplayName("深i企登录 - 缺少授权码参数应返回失败")
|
|||
// @Test
|
|||
// @Order(2)
|
|||
// public void testSiqLoginMissingCode() throws Exception {
|
|||
// String requestBody = "{}";
|
|||
//
|
|||
// mockMvc.perform(post(BASE_URL + "/siqLogin")
|
|||
// .contentType(MediaType.APPLICATION_JSON)
|
|||
// .content(requestBody))
|
|||
// .andDo(print())
|
|||
// .andExpect(status().isOk())
|
|||
// .andExpect(jsonPath("$.code").value(500));
|
|||
//
|
|||
// System.out.println("缺少授权码参数返回失败测试成功");
|
|||
// }
|
|||
//
|
|||
// @DisplayName("深i企登录 - 空授权码应返回失败")
|
|||
// @Test
|
|||
// @Order(3)
|
|||
// public void testSiqLoginEmptyCode() throws Exception {
|
|||
// String requestBody = "{\"code\":\"\"}";
|
|||
//
|
|||
// mockMvc.perform(post(BASE_URL + "/siqLogin")
|
|||
// .contentType(MediaType.APPLICATION_JSON)
|
|||
// .content(requestBody))
|
|||
// .andDo(print())
|
|||
// .andExpect(status().isOk())
|
|||
// .andExpect(jsonPath("$.code").value(500));
|
|||
//
|
|||
// System.out.println("空授权码返回失败测试成功");
|
|||
// }
|
|||
//
|
|||
// @DisplayName("深i企登录 - 无效授权码应返回失败")
|
|||
// @Test
|
|||
// @Order(4)
|
|||
// public void testSiqLoginInvalidCode() throws Exception {
|
|||
// String requestBody = "{\"code\":\"invalid_code_12345\"}";
|
|||
//
|
|||
// MvcResult result = mockMvc.perform(post(BASE_URL + "/siqLogin")
|
|||
// .contentType(MediaType.APPLICATION_JSON)
|
|||
// .content(requestBody))
|
|||
// .andDo(print())
|
|||
// .andExpect(status().isOk())
|
|||
// .andReturn();
|
|||
//
|
|||
// String responseBody = result.getResponse().getContentAsString(StandardCharsets.UTF_8);
|
|||
//
|
|||
// if (responseBody.contains("\"code\":500")) {
|
|||
// System.out.println("无效授权码返回失败测试成功");
|
|||
// } else {
|
|||
// System.out.println("注意：无效授权码未返回预期错误，响应: " + responseBody);
|
|||
// }
|
|||
// }
|
|||
//
|
|||
// @DisplayName("深i企登录 - 验证返回数据结构完整性")
|
|||
// @Test
|
|||
// @Order(5)
|
|||
// public void testSiqLoginResponseStructure() throws Exception {
|
|||
// String requestBody = "{\"code\":\"test_code\"}";
|
|||
//
|
|||
// MvcResult result = mockMvc.perform(post(BASE_URL + "/siqLogin")
|
|||
// .contentType(MediaType.APPLICATION_JSON)
|
|||
// .content(requestBody))
|
|||
// .andDo(print())
|
|||
// .andExpect(status().isOk())
|
|||
// .andReturn();
|
|||
//
|
|||
// String responseBody = result.getResponse().getContentAsString(StandardCharsets.UTF_8);
|
|||
// System.out.println("验证返回数据结构响应: " + responseBody);
|
|||
//
|
|||
// if (responseBody.contains("\"code\":200")) {
|
|||
// Assertions.assertTrue(responseBody.contains("\"data\""), "返回数据应包含 data 字段");
|
|||
//
|
|||
// String data = extractDataJson(responseBody);
|
|||
// if (data != null) {
|
|||
// Assertions.assertTrue(data.contains("token"), "data 中应包含 token");
|
|||
// Assertions.assertTrue(data.contains("userId"), "data 中应包含 userId");
|
|||
// Assertions.assertTrue(data.contains("userName"), "data 中应包含 userName");
|
|||
// Assertions.assertTrue(data.contains("loginName"), "data 中应包含 loginName");
|
|||
// Assertions.assertTrue(data.contains("phonenumber"), "data 中应包含 phonenumber");
|
|||
// Assertions.assertTrue(data.contains("enterpriseId"), "data 中应包含 enterpriseId");
|
|||
// Assertions.assertTrue(data.contains("enterpriseName"), "data 中应包含 enterpriseName");
|
|||
// Assertions.assertTrue(data.contains("enterpriseNumber"), "data 中应包含 enterpriseNumber");
|
|||
// System.out.println("返回数据结构完整，包含所有必需字段");
|
|||
// }
|
|||
// } else {
|
|||
// System.out.println("登录失败，无法验证数据结构");
|
|||
// }
|
|||
// }
|
|||
//
|
|||
// @DisplayName("深i企登录 - 重复登录应能正常处理（同一授权码多次调用）")
|
|||
// @Test
|
|||
// @Order(6)
|
|||
// public void testSiqLoginDuplicateCall() throws Exception {
|
|||
// String requestBody = "{\"code\":\"test_code\"}";
|
|||
//
|
|||
// for (int i = 1; i <= 3; i++) {
|
|||
// System.out.println("========== 第 " + i + " 次调用深i企登录 ==========");
|
|||
//
|
|||
// MvcResult result = mockMvc.perform(post(BASE_URL + "/siqLogin")
|
|||
// .contentType(MediaType.APPLICATION_JSON)
|
|||
// .content(requestBody))
|
|||
// .andDo(print())
|
|||
// .andExpect(status().isOk())
|
|||
// .andReturn();
|
|||
//
|
|||
// String responseBody = result.getResponse().getContentAsString(StandardCharsets.UTF_8);
|
|||
// System.out.println("第 " + i + " 次调用响应: " + responseBody);
|
|||
//
|
|||
// Thread.sleep(1000);
|
|||
// }
|
|||
//
|
|||
// System.out.println("========== 重复登录测试完成 ==========");
|
|||
// }
|
|||
//
|
|||
// @DisplayName("深i企登录 - 验证 Token 格式")
|
|||
// @Test
|
|||
// @Order(7)
|
|||
// public void testSiqLoginTokenFormat() throws Exception {
|
|||
// String requestBody = "{\"code\":\"test_code\"}";
|
|||
//
|
|||
// MvcResult result = mockMvc.perform(post(BASE_URL + "/siqLogin")
|
|||
// .contentType(MediaType.APPLICATION_JSON)
|
|||
// .content(requestBody))
|
|||
// .andDo(print())
|
|||
// .andExpect(status().isOk())
|
|||
// .andReturn();
|
|||
//
|
|||
// String responseBody = result.getResponse().getContentAsString(StandardCharsets.UTF_8);
|
|||
//
|
|||
// if (responseBody.contains("\"code\":200")) {
|
|||
// String token = extractTokenFromResponse(responseBody);
|
|||
// if (token != null && !token.isEmpty()) {
|
|||
// Assertions.assertTrue(token.length() > 10, "Token 长度应大于10个字符");
|
|||
// System.out.println("Token 格式验证通过，Token: " + token);
|
|||
// } else {
|
|||
// System.out.println("未能提取到 Token");
|
|||
// }
|
|||
// } else {
|
|||
// System.out.println("登录失败，无法验证 Token 格式");
|
|||
// }
|
|||
// }
|
|||
//
|
|||
// @DisplayName("深i企登录 - 验证企业信息关联")
|
|||
// @Test
|
|||
// @Order(8)
|
|||
// public void testSiqLoginEnterpriseAssociation() throws Exception {
|
|||
// String requestBody = "{\"code\":\"test_code\"}";
|
|||
//
|
|||
// MvcResult result = mockMvc.perform(post(BASE_URL + "/siqLogin")
|
|||
// .contentType(MediaType.APPLICATION_JSON)
|
|||
// .content(requestBody))
|
|||
// .andDo(print())
|
|||
// .andExpect(status().isOk())
|
|||
// .andReturn();
|
|||
//
|
|||
// String responseBody = result.getResponse().getContentAsString(StandardCharsets.UTF_8);
|
|||
//
|
|||
// if (responseBody.contains("\"code\":200")) {
|
|||
// String data = extractDataJson(responseBody);
|
|||
// if (data != null) {
|
|||
// String enterpriseId = extractJsonValue(data, "enterpriseId");
|
|||
// String enterpriseName = extractJsonValue(data, "enterpriseName");
|
|||
// String enterpriseNumber = extractJsonValue(data, "enterpriseNumber");
|
|||
//
|
|||
// if (enterpriseId != null && enterpriseName != null && enterpriseNumber != null) {
|
|||
// Assertions.assertFalse(enterpriseId.isEmpty(), "企业ID不应为空");
|
|||
// Assertions.assertFalse(enterpriseName.isEmpty(), "企业名称不应为空");
|
|||
// Assertions.assertFalse(enterpriseNumber.isEmpty(), "企业统一信用代码不应为空");
|
|||
// System.out.println("企业信息关联验证通过");
|
|||
// System.out.println("企业ID: " + enterpriseId);
|
|||
// System.out.println("企业名称: " + enterpriseName);
|
|||
// System.out.println("企业统一信用代码: " + enterpriseNumber);
|
|||
// }
|
|||
// }
|
|||
// } else {
|
|||
// System.out.println("登录失败，无法验证企业信息关联");
|
|||
// }
|
|||
// }
|
|||
//
|
|||
// // ==================== 工具方法 ====================
|
|||
//
|
|||
// /**
|
|||
// * 从 JSON 响应中提取 data 字段
|
|||
// */
|
|||
// private String extractDataJson(String responseBody) {
|
|||
// if (responseBody != null && responseBody.contains("\"data\":{")) {
|
|||
// int start = responseBody.indexOf("\"data\":{") + 7;
|
|||
// int end = responseBody.lastIndexOf("}");
|
|||
// if (start > 0 && end > start) {
|
|||
// return responseBody.substring(start, end + 1);
|
|||
// }
|
|||
// }
|
|||
// return null;
|
|||
// }
|
|||
//
|
|||
// /**
|
|||
// * 从响应中提取 Token
|
|||
// */
|
|||
// private String extractTokenFromResponse(String responseBody) {
|
|||
// return extractJsonValue(responseBody, "token");
|
|||
// }
|
|||
//
|
|||
// /**
|
|||
// * 从 JSON 字符串中提取指定字段的值
|
|||
// */
|
|||
// private String extractJsonValue(String json, String key) {
|
|||
// if (json != null && json.contains("\"" + key + "\"")) {
|
|||
// int start = json.indexOf("\"" + key + "\":\"") + key.length() + 4;
|
|||
// int end = json.indexOf("\"", start);
|
|||
// if (start > 0 && end > start) {
|
|||
// return json.substring(start, end);
|
|||
// }
|
|||
// }
|
|||
// return null;
|
|||
// }
|
|||
//}
|
|||
//{
|
|||
//}
|
|||
@ -1,329 +0,0 @@ |
|||
package com.zdxt.rest.test.cp; |
|||
|
|||
import cn.dev33.satoken.context.mock.SaTokenContextMockUtil; |
|||
import cn.dev33.satoken.stp.StpUtil; |
|||
import cn.dev33.satoken.stp.parameter.SaLoginParameter; |
|||
import com.fasterxml.jackson.databind.JsonNode; |
|||
import com.fasterxml.jackson.databind.ObjectMapper; |
|||
import com.zdxt.enforcementcode.domain.bo.EnterpriseInformationBo; |
|||
import com.zdxt.enforcementcode.domain.vo.EnterpriseInformationVo; |
|||
import com.zdxt.enforcementcode.service.IEnterpriseInformationService; |
|||
import org.dromara.ZdxtAdminApplication; |
|||
import org.dromara.common.core.domain.model.LoginUser; |
|||
import org.dromara.common.satoken.utils.LoginHelper; |
|||
import org.junit.jupiter.api.*; |
|||
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.http.MediaType; |
|||
import org.springframework.mock.web.MockHttpServletRequest; |
|||
import org.springframework.test.context.ActiveProfiles; |
|||
import org.springframework.test.web.servlet.MockMvc; |
|||
import org.springframework.test.web.servlet.MvcResult; |
|||
import org.springframework.web.context.request.RequestContextHolder; |
|||
import org.springframework.web.context.request.ServletRequestAttributes; |
|||
|
|||
import javax.crypto.Cipher; |
|||
import javax.crypto.spec.IvParameterSpec; |
|||
import javax.crypto.spec.SecretKeySpec; |
|||
import java.nio.charset.StandardCharsets; |
|||
import java.text.SimpleDateFormat; |
|||
import java.util.*; |
|||
|
|||
import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.post; |
|||
import static org.springframework.test.web.servlet.result.MockMvcResultHandlers.print; |
|||
import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.*; |
|||
|
|||
/** |
|||
* AppEnfCpPlanController 新增计划批量测试 |
|||
* <p> |
|||
* 为多个用户批量创建执法计划（每人100条） |
|||
* </p> |
|||
*/ |
|||
@SpringBootTest(classes = ZdxtAdminApplication.class) |
|||
@AutoConfigureMockMvc |
|||
@ActiveProfiles({"dev", "dev-cust-zrb"}) |
|||
@DisplayName("AppEnfCpPlanController 批量新增计划测试") |
|||
@TestMethodOrder(MethodOrderer.OrderAnnotation.class) |
|||
public class AppEnfCpPlanControllerTest { |
|||
|
|||
@Autowired |
|||
private MockMvc mockMvc; |
|||
|
|||
@Autowired |
|||
private ObjectMapper objectMapper; |
|||
|
|||
@Autowired |
|||
private IEnterpriseInformationService enterpriseInformationService; |
|||
|
|||
private static final String BASE_URL = "/app-enf/cpPlan"; |
|||
private static final String LOGIN_URL = "/app-enf/login/do"; |
|||
private static final String AES_KEY = "1234567898745621"; |
|||
private static final String CLIENT_ID = "f0535c2a32ab4d0496ab9a699b2a7d36"; |
|||
private static final String USER_AGENT = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36"; |
|||
|
|||
/** 通用密码 */ |
|||
private static final String PASSWORD = "szsf@zfjd"; |
|||
|
|||
@BeforeEach |
|||
public void setUp() { |
|||
// 初始化 Sa-Token 上下文
|
|||
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(); |
|||
} |
|||
|
|||
/** |
|||
* 登录指定账号，返回登录信息 |
|||
*/ |
|||
private Map<String, String> doLogin(String account) throws Exception { |
|||
String encryptedPassword = encryptByAes(PASSWORD); |
|||
Map<String, String> loginParams = new HashMap<>(); |
|||
loginParams.put("secretName", account); |
|||
loginParams.put("secretKey", encryptedPassword); |
|||
|
|||
MvcResult loginResult = mockMvc.perform(post(LOGIN_URL) |
|||
.header("User-Agent", USER_AGENT) |
|||
.contentType(MediaType.APPLICATION_JSON) |
|||
.content(objectMapper.writeValueAsString(loginParams))) |
|||
.andExpect(status().isOk()) |
|||
.andExpect(jsonPath("$.code").value(200)) |
|||
.andReturn(); |
|||
|
|||
String loginBody = loginResult.getResponse().getContentAsString(StandardCharsets.UTF_8); |
|||
JsonNode loginJson = objectMapper.readTree(loginBody); |
|||
JsonNode data = loginJson.get("data"); |
|||
|
|||
Map<String, String> info = new HashMap<>(); |
|||
info.put("token", data.get("token").asText()); |
|||
info.put("userId", data.get("userId").asText()); |
|||
// nickName 可能为空，回退到 userName（中文姓名）
|
|||
String nickName = data.has("nickName") ? data.get("nickName").asText() : ""; |
|||
if (nickName == null || nickName.isBlank()) { |
|||
nickName = data.has("userName") ? data.get("userName").asText() : account; |
|||
} |
|||
info.put("nickName", nickName); |
|||
info.put("deptId", data.get("deptId").asText()); |
|||
info.put("deptName", data.get("deptName").asText()); |
|||
info.put("organId", data.get("organId").asText()); |
|||
info.put("organName", data.get("organName").asText()); |
|||
info.put("bureauUnitId", data.get("bureauUnitId").asText()); |
|||
info.put("bureauUnitName", data.get("bureauUnitName").asText()); |
|||
info.put("phonenumber", data.has("phonenumber") ? data.get("phonenumber").asText() : ""); |
|||
|
|||
System.out.println("[" + account + "] 登录成功: userId=" + info.get("userId")); |
|||
return info; |
|||
} |
|||
|
|||
// ==================== 各账号批量新增测试 ====================
|
|||
|
|||
@DisplayName("liuqianlin（刘倩琳）批量新增100条执法计划") |
|||
@Test |
|||
@Order(1) |
|||
public void testBatchCreate_liuqianlin() throws Exception { |
|||
batchCreatePlans("liuqianlin", 100); |
|||
} |
|||
|
|||
@DisplayName("baining（白宁）批量新增100条执法计划") |
|||
@Test |
|||
@Order(2) |
|||
public void testBatchCreate_baining() throws Exception { |
|||
batchCreatePlans("baining", 100); |
|||
} |
|||
|
|||
@DisplayName("caizifan（蔡梓帆）批量新增100条执法计划") |
|||
@Test |
|||
@Order(3) |
|||
public void testBatchCreate_caizifan() throws Exception { |
|||
batchCreatePlans("caizifan", 100); |
|||
} |
|||
|
|||
@DisplayName("chenchong4（陈冲）批量新增100条执法计划") |
|||
@Test |
|||
@Order(4) |
|||
public void testBatchCreate_chenchong4() throws Exception { |
|||
batchCreatePlans("chenchong4", 100); |
|||
} |
|||
|
|||
/** |
|||
* 为指定账号批量新增计划 |
|||
* <p> |
|||
* 唯一性约束：同一人 + 日期 + 时间段 + 企业 只能有一条计划。 |
|||
* 日期循环31天(7月1~31日)，时间段2个(上午/下午)，企业循环7家， |
|||
* 因 31、2、7 两两互素，由中国剩余定理(CRT)保证前 31×2×7=434 条组合均唯一， |
|||
* 故 count≤434 时所有计划不会重复。 |
|||
* </p> |
|||
*/ |
|||
private void batchCreatePlans(String account, int count) throws Exception { |
|||
Map<String, String> loginInfo = doLogin(account); |
|||
String token = loginInfo.get("token"); |
|||
String userId = loginInfo.get("userId"); |
|||
String nickName = loginInfo.get("nickName"); |
|||
|
|||
// 从数据库加载指定企业列表
|
|||
List<EnterpriseInformationVo> enterprises = loadEnterprises(); |
|||
Assertions.assertFalse(enterprises.isEmpty(), "数据库中无可用企业数据"); |
|||
|
|||
// --- 唯一性预测验证 ---
|
|||
int days = 31; // 7月有31天
|
|||
int timeSlots = 2; // 上午/下午
|
|||
int entCount = enterprises.size(); |
|||
// 由CRT保证的最大不重复组合数 = lcm(days, timeSlots, entCount) 当两两互素时 = days * timeSlots * entCount
|
|||
int maxUnique = days * timeSlots * entCount; |
|||
int expectedSuccess = Math.min(count, maxUnique); |
|||
System.out.println("[" + account + "] 企业数=" + entCount |
|||
+ ", 可用唯一组合(日期×时段×企业)=" + maxUnique |
|||
+ ", 计划生成数=" + count |
|||
+ ", 预计成功数=" + expectedSuccess); |
|||
Assertions.assertTrue(count <= maxUnique, |
|||
"计划数(" + count + ")超过最大唯一组合数(" + maxUnique + ")，必然产生重复"); |
|||
|
|||
Random random = new Random(); |
|||
SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd"); |
|||
Calendar cal = Calendar.getInstance(); |
|||
|
|||
int successCount = 0; |
|||
int failCount = 0; |
|||
|
|||
for (int i = 0; i < count; i++) { |
|||
// 所有计划日期限定在2026年7月内，循环使用7月1日~31日
|
|||
int dayOfMonth = (i % days) + 1; |
|||
cal.set(2026, Calendar.JULY, dayOfMonth); |
|||
String planDate = sdf.format(cal.getTime()); |
|||
long planTime = (i % 2 == 0) ? 1L : 2L; // 交替上午/下午
|
|||
|
|||
// 确定性循环选取企业，保证 (日期, 时段, 企业) 三元组唯一
|
|||
EnterpriseInformationVo enterprise = enterprises.get(i % entCount); |
|||
Map<String, Object> planBo = buildPlanBo(planDate, planTime, i, nickName, userId, loginInfo, enterprise, random); |
|||
String requestBody = objectMapper.writeValueAsString(planBo); |
|||
|
|||
MvcResult result = mockMvc.perform(post(BASE_URL) |
|||
.header("Authorization", "Bearer " + token) |
|||
.header("clientid", CLIENT_ID) |
|||
.header("User-Agent", USER_AGENT) |
|||
.contentType(MediaType.APPLICATION_JSON) |
|||
.content(requestBody)) |
|||
.andReturn(); |
|||
|
|||
String responseBody = result.getResponse().getContentAsString(StandardCharsets.UTF_8); |
|||
JsonNode respJson = objectMapper.readTree(responseBody); |
|||
int code = respJson.get("code").asInt(); |
|||
|
|||
if (code == 200) { |
|||
successCount++; |
|||
} else { |
|||
failCount++; |
|||
System.out.println("[" + account + "] 第" + (i + 1) + "条失败: " + responseBody); |
|||
} |
|||
|
|||
if ((i + 1) % 10 == 0) { |
|||
System.out.println("[" + account + "] 已处理 " + (i + 1) + "/" + count |
|||
+ ", 成功=" + successCount + ", 失败=" + failCount); |
|||
} |
|||
} |
|||
|
|||
System.out.println("========== [" + account + "] 批量新增完成 =========="); |
|||
System.out.println("总计: " + count + ", 成功: " + successCount + ", 失败: " + failCount); |
|||
Assertions.assertEquals(expectedSuccess, successCount, |
|||
"[" + account + "] 预期成功" + expectedSuccess + "条，实际成功" + successCount + "条"); |
|||
} |
|||
|
|||
/** 指定的企业统一社会信用代码列表 */ |
|||
private static final String[][] ENTERPRISE_DATA = { |
|||
{"吉田拉链（深圳）有限公司公明工厂", "91440300788302281W"}, |
|||
{"欧姆龙电子部件（深圳）有限公司", "914403007261811621"}, |
|||
{"深圳农牧美益肉业有限公司", "91440300662678582M"}, |
|||
{"深圳市深水龙岗污水处理有限公司", "914403006820066603"}, |
|||
{"深圳市强达电路股份有限公司", "91440300763461366N"}, |
|||
{"深圳康泰生物制品股份有限公司", "91440300618837873J"}, |
|||
{"深圳市里程食品有限公司", "91440300738848091F"}, |
|||
}; |
|||
|
|||
/** |
|||
* 从数据库加载指定的7家企业 |
|||
*/ |
|||
private List<EnterpriseInformationVo> loadEnterprises() { |
|||
List<EnterpriseInformationVo> result = new ArrayList<>(); |
|||
for (String[] entry : ENTERPRISE_DATA) { |
|||
EnterpriseInformationBo queryBo = new EnterpriseInformationBo(); |
|||
queryBo.setEnterpriseNumber(entry[1]); |
|||
queryBo.setIsDelete(0L); |
|||
List<EnterpriseInformationVo> list = enterpriseInformationService.queryList(queryBo); |
|||
if (!list.isEmpty()) { |
|||
result.add(list.get(0)); |
|||
} else { |
|||
System.out.println("[警告] 未找到企业: " + entry[0] + " (" + entry[1] + ")"); |
|||
} |
|||
} |
|||
return result; |
|||
} |
|||
|
|||
/** |
|||
* 构建计划BO的Map |
|||
*/ |
|||
private Map<String, Object> buildPlanBo(String planDate, long planTime, int index, |
|||
String userName, String userId, Map<String, String> loginInfo, |
|||
EnterpriseInformationVo enterprise, Random random) { |
|||
Map<String, Object> bo = new LinkedHashMap<>(); |
|||
bo.put("planType", 1L); // 日常计划
|
|||
bo.put("planDate", planDate); |
|||
bo.put("planTime", planTime); |
|||
bo.put("enforceContent", "批量测试执法事项-第" + (index + 1) + "条"); |
|||
bo.put("enterpriseName", enterprise.getEnterpriseName()); |
|||
bo.put("enterpriseNumber", enterprise.getEnterpriseNumber()); |
|||
bo.put("enterpriseId", enterprise.getId()); |
|||
bo.put("enforceUserNames", userName); |
|||
bo.put("enforceUserIds", userId); |
|||
bo.put("enforceUserTels", loginInfo.get("phonenumber")); |
|||
bo.put("enforceUserCodes", "-"); |
|||
bo.put("isPublic", random.nextInt(2) == 0 ? 0L : 1L); // 随机公开/不公开
|
|||
bo.put("isUnion", 1L); // 必须为联合执法，否则afterInsert会直接生成任务导致status变为3
|
|||
bo.put("reserveStatus", 1L); // 待预约
|
|||
bo.put("systemReserved", 0L); |
|||
bo.put("manualReserved", 0L); |
|||
bo.put("bureauDeptId", loginInfo.get("bureauUnitId")); |
|||
bo.put("bureauDeptName", loginInfo.get("bureauUnitName")); |
|||
bo.put("organId", loginInfo.get("organId")); |
|||
bo.put("organName", loginInfo.get("organName")); |
|||
bo.put("deptId", loginInfo.get("deptId")); |
|||
bo.put("deptName", loginInfo.get("deptName")); |
|||
bo.put("isDelete", 0L); |
|||
return bo; |
|||
} |
|||
|
|||
/** |
|||
* AES/CBC/ZeroPadding 加密（与前端 CryptoJS 对应） |
|||
*/ |
|||
private String encryptByAes(String plainText) throws Exception { |
|||
byte[] keyBytes = AES_KEY.getBytes(StandardCharsets.UTF_8); |
|||
byte[] ivBytes = AES_KEY.getBytes(StandardCharsets.UTF_8); |
|||
|
|||
SecretKeySpec secretKeySpec = new SecretKeySpec(keyBytes, "AES"); |
|||
IvParameterSpec ivParameterSpec = new IvParameterSpec(ivBytes); |
|||
|
|||
// NoPadding 需要手动补零到16字节倍数
|
|||
byte[] plainBytes = plainText.getBytes(StandardCharsets.UTF_8); |
|||
int blockSize = 16; |
|||
int paddedLen = ((plainBytes.length / blockSize) + 1) * blockSize; |
|||
byte[] padded = new byte[paddedLen]; |
|||
System.arraycopy(plainBytes, 0, padded, 0, plainBytes.length); |
|||
// 剩余字节默认为0（ZeroPadding）
|
|||
|
|||
Cipher cipher = Cipher.getInstance("AES/CBC/NoPadding"); |
|||
cipher.init(Cipher.ENCRYPT_MODE, secretKeySpec, ivParameterSpec); |
|||
byte[] encrypted = cipher.doFinal(padded); |
|||
|
|||
return Base64.getEncoder().encodeToString(encrypted); |
|||
} |
|||
} |
|||
@ -1,225 +0,0 @@ |
|||
package com.zdxt.service.test; |
|||
|
|||
import cn.hutool.crypto.digest.DigestUtil; |
|||
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; |
|||
import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper; |
|||
import com.zdxt.enforcementcode.domain.EnterpriseUser; |
|||
import com.zdxt.enforcementcode.mapper.EnterpriseUserMapper; |
|||
import org.dromara.ZdxtAdminApplication; |
|||
import org.dromara.system.domain.SysUser; |
|||
import org.dromara.system.mapper.SysUserMapper; |
|||
import org.junit.jupiter.api.DisplayName; |
|||
import org.junit.jupiter.api.Test; |
|||
import org.springframework.beans.factory.annotation.Autowired; |
|||
import org.springframework.boot.test.autoconfigure.web.servlet.AutoConfigureMockMvc; |
|||
import org.springframework.boot.test.context.SpringBootTest; |
|||
|
|||
import java.util.Arrays; |
|||
import java.util.HashSet; |
|||
import java.util.List; |
|||
|
|||
import cn.dev33.satoken.context.mock.SaTokenContextMockUtil; |
|||
import cn.dev33.satoken.stp.parameter.SaLoginParameter; |
|||
import cn.dev33.satoken.stp.StpUtil; |
|||
import org.dromara.common.core.domain.model.LoginUser; |
|||
import org.dromara.common.satoken.utils.LoginHelper; |
|||
import org.junit.jupiter.api.AfterEach; |
|||
import org.junit.jupiter.api.BeforeEach; |
|||
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; |
|||
|
|||
/** |
|||
* 批量重置 enterprise_user 表用户密码 |
|||
* <p> |
|||
* 密码加密规则: MD5(userName + password + salt) |
|||
* </p> |
|||
*/ |
|||
@SpringBootTest(classes = ZdxtAdminApplication.class) |
|||
@AutoConfigureMockMvc |
|||
@ActiveProfiles({"dev", "dev-cust-zrb"}) |
|||
@DisplayName("批量重置企业用户密码") |
|||
public class ResetUserPasswordTest { |
|||
|
|||
@Autowired |
|||
private EnterpriseUserMapper enterpriseUserMapper; |
|||
|
|||
@Autowired |
|||
private SysUserMapper sysUserMapper; |
|||
|
|||
/** 重置后的明文密码 */ |
|||
private static final String NEW_PASSWORD = "szsf@zfjd"; |
|||
//private static final String NEW_PASSWORD = "Szsf@123";
|
|||
|
|||
private static final String TEST_CLIENT_ID = "e5cd7e4891bf95d1d19206ce24a7b32e"; |
|||
|
|||
@BeforeEach |
|||
public void setUp() { |
|||
// 初始化 Sa-Token 上下文（非 Web 环境下 StpUtil.login 需要）
|
|||
SaTokenContextMockUtil.setMockContext(); |
|||
|
|||
// 设置模拟请求上下文（数据权限拦截器需要从 SaToken 上下文获取用户信息）
|
|||
MockHttpServletRequest mockRequest = new MockHttpServletRequest(); |
|||
mockRequest.addHeader("User-Agent", "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36"); |
|||
RequestContextHolder.setRequestAttributes(new ServletRequestAttributes(mockRequest)); |
|||
|
|||
// 构造测试用登录用户
|
|||
LoginUser loginUser = new LoginUser(); |
|||
loginUser.setUserId(1L); |
|||
loginUser.setUserType("sys_user"); |
|||
loginUser.setTenantId("000000"); |
|||
loginUser.setUsername("test_admin"); |
|||
loginUser.setNickname("测试管理员"); |
|||
loginUser.setDeptId("103"); |
|||
loginUser.setDeptName("测试部门"); |
|||
loginUser.setRolePermission(new HashSet<>()); |
|||
loginUser.setMenuPermission(new HashSet<>()); |
|||
|
|||
// 模拟登录
|
|||
SaLoginParameter param = new SaLoginParameter(); |
|||
param.setExtra(LoginHelper.CLIENT_KEY, TEST_CLIENT_ID); |
|||
LoginHelper.login(loginUser, param); |
|||
} |
|||
|
|||
@AfterEach |
|||
public void tearDown() { |
|||
try { |
|||
StpUtil.logout(); |
|||
} catch (Exception ignored) { |
|||
} |
|||
RequestContextHolder.resetRequestAttributes(); |
|||
SaTokenContextMockUtil.clearContext(); |
|||
} |
|||
|
|||
@DisplayName("批量重置 enterprise_user 所有用户密码为 szsf@zfjd") |
|||
@Test |
|||
public void resetAllEnterpriseUserPassword() { |
|||
|
|||
/*-- 17608540150,weil45,"",15817307171,深圳农牧美益肉业有限公司,91440300662678582M,21 |
|||
-- 17608557492,DGD440881199506180622,"",15817381630,深圳市深水龙岗污水处理有限公司,914403006820066603,12 |
|||
-- 17608540121,smiledzb1982,"",13723454151,深圳康泰生物制品股份有限公司,91440300618837873J,10 |
|||
-- 17608594890,A7271888,"",13751661770,深圳市里程食品有限公司,91440300738848091F,9 |
|||
-- 17608565612,qintongzhen138,"",13823610579,欧姆龙电子部件（深圳）有限公司,914403007261811621,9 |
|||
-- 17608601734,XP073215,"",15012671807,深圳市强达电路股份有限公司,91440300763461366N,9 |
|||
-- 17608556724,33ef1970fd449676fa4bc197,"",13537602265,吉田拉链（深圳）有限公司公明工厂,91440300788302281W,8*/ |
|||
|
|||
// 1. 指定需要重置密码的用户ID列表
|
|||
List<String> userIds = Arrays.asList( |
|||
// TODO: 在此填入需要重置的用户ID
|
|||
"17608540150", "17608557492", "17608540121", "17608594890", "17608565612", "17608601734", "17608556724","2064275054386966530","2064274295779979265","17608629742" |
|||
); |
|||
|
|||
// 2. 根据ID列表查询企业用户
|
|||
List<EnterpriseUser> userList = enterpriseUserMapper.selectList( |
|||
new LambdaQueryWrapper<EnterpriseUser>() |
|||
.select(EnterpriseUser::getUserId, EnterpriseUser::getLoginName, EnterpriseUser::getUserName, EnterpriseUser::getSalt) |
|||
.in(EnterpriseUser::getUserId, userIds) |
|||
); |
|||
|
|||
System.out.println("查询到企业用户总数: " + userList.size()); |
|||
|
|||
int successCount = 0; |
|||
int skipCount = 0; |
|||
|
|||
for (EnterpriseUser user : userList) { |
|||
String loginName = user.getLoginName(); |
|||
String salt = user.getSalt(); |
|||
|
|||
if (loginName == null || loginName.isBlank()) { |
|||
System.out.println("跳过: userId=" + user.getUserId() + ", loginName为空"); |
|||
skipCount++; |
|||
continue; |
|||
} |
|||
if (salt == null || salt.isBlank()) { |
|||
System.out.println("跳过: userId=" + user.getUserId() + ", salt为空"); |
|||
skipCount++; |
|||
continue; |
|||
} |
|||
|
|||
// 3. 按规则生成新密码: MD5(loginName + newPassword + salt)
|
|||
String newMd5Password = DigestUtil.md5Hex(loginName + NEW_PASSWORD + salt); |
|||
|
|||
// 4. 更新密码
|
|||
int rows = enterpriseUserMapper.update(null, |
|||
new LambdaUpdateWrapper<EnterpriseUser>() |
|||
.eq(EnterpriseUser::getUserId, user.getUserId()) |
|||
.set(EnterpriseUser::getPassword, newMd5Password) |
|||
); |
|||
|
|||
if (rows > 0) { |
|||
successCount++; |
|||
} |
|||
} |
|||
|
|||
System.out.println("========== 重置完成 =========="); |
|||
System.out.println("成功: " + successCount + " 条"); |
|||
System.out.println("跳过: " + skipCount + " 条"); |
|||
System.out.println("新密码(明文): " + NEW_PASSWORD); |
|||
} |
|||
|
|||
|
|||
@DisplayName("批量重置 sys_user 指定用户密码为 szsf@zfjd") |
|||
@Test |
|||
public void resetSysUserPassword() { |
|||
// 1. 执法端的测试用户
|
|||
List<Long> userIds = Arrays.asList( |
|||
// TODO: 在此填入需要重置的用户ID
|
|||
7897L, 17608416858L,17608416876L, 17608416904L,17608416912L, 17608416926L,17608416927L, 17608416945L,17608417058L,17608642193L,17608502852L,17608515249L,17608497168L |
|||
); |
|||
|
|||
// 监督端的报表测试用户
|
|||
// userIds = Arrays.asList(
|
|||
// // TODO: 在此填入需要重置的用户ID
|
|||
// 17608417340L, 17608417451L,17608417716L, 17608614169L,17608614499L,7897L,17608634290L
|
|||
// );
|
|||
|
|||
// 2. 根据ID列表查询系统用户
|
|||
List<SysUser> userList = sysUserMapper.selectList( |
|||
new LambdaQueryWrapper<SysUser>() |
|||
.select(SysUser::getUserId, SysUser::getLoginName, SysUser::getSalt) |
|||
.in(SysUser::getUserId, userIds) |
|||
); |
|||
|
|||
System.out.println("查询到系统用户总数: " + userList.size()); |
|||
|
|||
int successCount = 0; |
|||
int skipCount = 0; |
|||
|
|||
for (SysUser user : userList) { |
|||
String loginName = user.getLoginName(); |
|||
String salt = user.getSalt(); |
|||
|
|||
if (loginName == null || loginName.isBlank()) { |
|||
System.out.println("跳过: userId=" + user.getUserId() + ", loginName为空"); |
|||
skipCount++; |
|||
continue; |
|||
} |
|||
if (salt == null || salt.isBlank()) { |
|||
System.out.println("跳过: userId=" + user.getUserId() + ", salt为空"); |
|||
skipCount++; |
|||
continue; |
|||
} |
|||
|
|||
// 3. 按规则生成新密码: MD5( + newPassword + salt)
|
|||
String newMd5Password = DigestUtil.md5Hex(loginName + NEW_PASSWORD + salt); |
|||
|
|||
// 4. 更新密码
|
|||
int rows = sysUserMapper.update(null, |
|||
new LambdaUpdateWrapper<SysUser>() |
|||
.eq(SysUser::getUserId, user.getUserId()) |
|||
.set(SysUser::getPassword, newMd5Password) |
|||
); |
|||
|
|||
if (rows > 0) { |
|||
successCount++; |
|||
} |
|||
} |
|||
|
|||
System.out.println("========== sys_user 重置完成 =========="); |
|||
System.out.println("成功: " + successCount + " 条"); |
|||
System.out.println("跳过: " + skipCount + " 条"); |
|||
System.out.println("新密码(明文): " + NEW_PASSWORD); |
|||
} |
|||
|
|||
} |
|||
Loading…
Reference in new issue