ZF/weiyu
1
0
Fork 0
mirror of https://gitee.com/270580156/weiyu.git synced 2026-05-17 12:48:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

Sync from bytedesk-private: update

Browse Source
This commit is contained in:
jack ning
2024-12-24 10:44:25 +08:00
parent b3adda2c45
commit 4d78d41813
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

194
modules/service/src/test/java/com/bytedesk/service/visitor/VisitorAnonymousControllerTest.java
View File

@@ -1,194 +0,0 @@
package com.bytedesk.service.visitor;
import static org.mockito.ArgumentMatchers.any;
import static org.mockito.Mockito.times;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;
import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.post;
import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.status;
import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.CompletableFuture;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import org.junit.jupiter.api.BeforeEach;
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 org.springframework.boot.test.mock.mockito.MockBean;
import org.springframework.http.MediaType;
import org.springframework.test.web.servlet.MockMvc;
import com.bytedesk.core.utils.JsonResult;
import com.fasterxml.jackson.databind.ObjectMapper;
import lombok.extern.slf4j.Slf4j;
@Slf4j
@SpringBootTest
@AutoConfigureMockMvc
public class VisitorAnonymousControllerTest {
@Autowired
private MockMvc mockMvc;
@Autowired
private ObjectMapper objectMapper;
@MockBean
private VisitorRestService visitorRestService;
private static final String ORG_UID = "df_org_uid";
private static final String AGENT_UID = "df_ag_uid"; // df_ag_uid or df_wg_uid
private static final String TYPE = "0";// 0: agent, 1: workgroup
private static final int VISITOR_COUNT = 100;
private static final int REQUEST_PER_VISITOR = 100;
private List<VisitorResponse> visitors;
@BeforeEach
void setUp() {
// 初始化访客列表
visitors = new ArrayList<>();
for (int i = 0; i < VISITOR_COUNT; i++) {
VisitorResponse visitor = VisitorResponse.builder()
.nickname("Visitor " + i)
.avatar("https://example.com/avatar/" + i + ".jpg")
.build();
visitor.setUid("visitor_" + i);
visitors.add(visitor);
}
// Mock visitorService的响应
when(visitorRestService.create(any())).thenAnswer(invocation -> {
int index = (int)(Math.random() * VISITOR_COUNT);
return visitors.get(index);
});
}
@Test
void testMassVisitorRequests() throws Exception {
ExecutorService executor = Executors.newFixedThreadPool(10);
List<CompletableFuture<Void>> futures = new ArrayList<>();
// 创建访客
for (int i = 0; i < VISITOR_COUNT; i++) {
final int visitorIndex = i;
CompletableFuture<Void> future = CompletableFuture.runAsync(() -> {
try {
// 1. 初始化访客
VisitorRequest initRequest = VisitorRequest.builder()
.sid(AGENT_UID)
.build();
initRequest.setType(TYPE);
initRequest.setOrgUid(ORG_UID);
String initResult = mockMvc.perform(post("/visitor/api/v1/init")
.contentType(MediaType.APPLICATION_JSON)
.content(objectMapper.writeValueAsString(initRequest)))
.andExpect(status().isOk())
.andReturn()
.getResponse()
.getContentAsString();
// 解析返回的访客信息
JsonResult<?> initResponse = objectMapper.readValue(initResult, JsonResult.class);
VisitorResponse visitor = objectMapper.convertValue(initResponse.getData(), VisitorResponse.class);
// 2. 请求会话
for (int j = 0; j < REQUEST_PER_VISITOR; j++) {
VisitorRequest threadRequest = VisitorRequest.builder()
.sid(AGENT_UID)
.nickname(visitor.getNickname())
.avatar(visitor.getAvatar())
.build();
threadRequest.setOrgUid(ORG_UID);
threadRequest.setType(TYPE);
threadRequest.setUid(visitor.getUid());
mockMvc.perform(post("/visitor/api/v1/request")
.contentType(MediaType.APPLICATION_JSON)
.content(objectMapper.writeValueAsString(threadRequest)))
.andExpect(status().isOk());
}
} catch (Exception e) {
log.error("访客请求失败: {}", visitorIndex, e);
}
}, executor);
futures.add(future);
}
// 等待所有请求完成
CompletableFuture.allOf(futures.toArray(new CompletableFuture[0])).join();
executor.shutdown();
// 验证调用次数
verify(visitorRestService, times(VISITOR_COUNT)).create(any());
verify(visitorRestService, times(VISITOR_COUNT * REQUEST_PER_VISITOR)).requestThread(any());
}
@Test
void testConcurrentVisitorRequests() throws Exception {
int concurrentVisitors = 10;
int concurrentRequests = 10;
ExecutorService executor = Executors.newFixedThreadPool(concurrentVisitors);
List<CompletableFuture<Void>> futures = new ArrayList<>();
for (int i = 0; i < concurrentVisitors; i++) {
CompletableFuture<Void> future = CompletableFuture.runAsync(() -> {
try {
// 1. 初始化访客
VisitorRequest initRequest = VisitorRequest.builder()
.sid(AGENT_UID)
.build();
initRequest.setType(TYPE);
initRequest.setOrgUid(ORG_UID);
String initResult = mockMvc.perform(post("/visitor/api/v1/init")
.contentType(MediaType.APPLICATION_JSON)
.content(objectMapper.writeValueAsString(initRequest)))
.andExpect(status().isOk())
.andReturn()
.getResponse()
.getContentAsString();
// 解析返回的访客信息
JsonResult<?> initResponse = objectMapper.readValue(initResult, JsonResult.class);
VisitorResponse visitor = objectMapper.convertValue(initResponse.getData(), VisitorResponse.class);
// 2. 并发请求会话
for (int j = 0; j < concurrentRequests; j++) {
VisitorRequest threadRequest = VisitorRequest.builder()
.sid(AGENT_UID)
.nickname(visitor.getNickname())
.avatar(visitor.getAvatar())
.build();
threadRequest.setType(TYPE);
threadRequest.setOrgUid(ORG_UID);
threadRequest.setUid(visitor.getUid());
mockMvc.perform(post("/visitor/api/v1/request")
.contentType(MediaType.APPLICATION_JSON)
.content(objectMapper.writeValueAsString(threadRequest)))
.andExpect(status().isOk());
}
} catch (Exception e) {
log.error("并发请求失败", e);
}
}, executor);
futures.add(future);
}
// 等待所有并发请求完成
CompletableFuture.allOf(futures.toArray(new CompletableFuture[0])).join();
executor.shutdown();
// 验证调用次数
verify(visitorRestService, times(concurrentVisitors)).create(any());
verify(visitorRestService, times(concurrentVisitors * concurrentRequests)).requestThread(any());
}
}