package jnpf.bizcommon.onlyoffice.service.impl;
|
|
import cn.hutool.http.HttpRequest;
|
import cn.hutool.http.HttpResponse;
|
import cn.hutool.http.HttpUtil;
|
import cn.hutool.jwt.JWTUtil;
|
import com.baomidou.mybatisplus.core.MybatisConfiguration;
|
import com.baomidou.mybatisplus.core.metadata.TableInfoHelper;
|
import jnpf.bizcommon.onlyoffice.entity.OnlyOfficeCallbackParam;
|
import jnpf.bizcommon.onlyoffice.entity.OnlyOfficeProperties;
|
import jnpf.bizcommon.onlyoffice.entity.OnlyOfficeSessionEntity;
|
import jnpf.bizcommon.onlyoffice.mapper.OnlyOfficeSessionMapper;
|
import jnpf.exception.DataException;
|
import jnpf.file.FileApi;
|
import jnpf.file.FileUploadApi;
|
import org.apache.ibatis.builder.MapperBuilderAssistant;
|
import org.junit.After;
|
import org.junit.Before;
|
import org.junit.Test;
|
import org.mockito.MockedStatic;
|
|
import java.nio.charset.StandardCharsets;
|
import java.util.Map;
|
|
import static org.junit.Assert.*;
|
import static org.mockito.ArgumentMatchers.*;
|
import static org.mockito.Mockito.*;
|
|
/** 使用内存中的文档响应验证场景分发，不监听端口或访问外部文件服务。 */
|
public class UploadEditCallbackTest {
|
private OnlyOfficeSessionMapper sessionMapper;
|
private FileApi fileApi;
|
private FileUploadApi fileUploadApi;
|
private OnlyOfficeServiceImpl service;
|
private MockedStatic<HttpUtil> http;
|
|
@Before
|
public void setup() {
|
TableInfoHelper.initTableInfo(new MapperBuilderAssistant(new MybatisConfiguration(), ""), OnlyOfficeSessionEntity.class);
|
sessionMapper = mock(OnlyOfficeSessionMapper.class);
|
when(sessionMapper.update(isNull(), any())).thenReturn(1);
|
fileApi = mock(FileApi.class);
|
fileUploadApi = mock(FileUploadApi.class);
|
when(fileApi.getPath("annex")).thenReturn("WebAnnexFile");
|
OnlyOfficeProperties properties = new OnlyOfficeProperties();
|
properties.setJwtSecret("test-secret");
|
service = new OnlyOfficeServiceImpl(properties, sessionMapper, null, null, null, null, null,
|
fileApi, fileUploadApi, new DmsQianpiyeDocxPropertiesWriter());
|
http = mockStatic(HttpUtil.class);
|
HttpRequest request = mock(HttpRequest.class);
|
HttpResponse response = mock(HttpResponse.class);
|
http.when(() -> HttpUtil.createGet("http://document-server/saved.docx")).thenReturn(request);
|
when(request.timeout(anyInt())).thenReturn(request);
|
when(request.executeAsync()).thenReturn(response);
|
when(response.isOk()).thenReturn(true);
|
when(response.bodyBytes()).thenReturn("saved-document".getBytes(StandardCharsets.UTF_8));
|
}
|
|
@After
|
public void cleanup() {
|
if (http != null) http.close();
|
}
|
|
@Test
|
public void uploadCallbackSaveFailureUsesUploadSpecificException() {
|
OnlyOfficeSessionEntity session = session();
|
session.setBizModule("visualdev");
|
session.setBizScene("visualdev.uploadFile");
|
when(sessionMapper.selectOne(any())).thenReturn(session);
|
for (int status : new int[] {2, 6}) {
|
OnlyOfficeCallbackParam callback = callback();
|
callback.setKey(session.getDocKey());
|
callback.setStatus(status);
|
callback.setUrl(null);
|
Map<String, Object> claims = new java.util.LinkedHashMap<>();
|
claims.put("key", callback.getKey());
|
claims.put("status", status);
|
String token = JWTUtil.createToken(claims, "test-secret".getBytes(StandardCharsets.UTF_8));
|
try {
|
service.handleCallback(callback, "Bearer " + token);
|
fail("upload save without a source URL must fail");
|
} catch (jnpf.bizcommon.onlyoffice.exception.UploadEditCallbackException e) {
|
assertTrue(e.getCause() instanceof DataException);
|
assertTrue(e.getMessage().contains("缺少下载 url"));
|
}
|
}
|
verifyNoInteractions(fileApi);
|
verify(fileUploadApi, never()).replaceFile(any(), any(), any());
|
}
|
|
@Test
|
public void uploadCallbackSavesThroughUploadBranchOnly() throws Exception {
|
UploadEditSessionSupport support = mock(UploadEditSessionSupport.class);
|
java.lang.reflect.Field field = OnlyOfficeServiceImpl.class.getDeclaredField("uploadEditSessionSupport");
|
field.setAccessible(true);
|
field.set(service, support);
|
for (int status : new int[] {2, 6}) {
|
OnlyOfficeSessionEntity session = session();
|
session.setBizModule("visualdev");
|
session.setBizScene("visualdev.uploadFile");
|
session.setSessionStatus(OnlyOfficeSessionEntity.SessionStatus.EDITING);
|
when(sessionMapper.selectOne(any())).thenReturn(session);
|
OnlyOfficeCallbackParam callback = callback();
|
callback.setKey(session.getDocKey());
|
callback.setStatus(status);
|
Map<String, Object> claims = new java.util.LinkedHashMap<>();
|
claims.put("key", callback.getKey());
|
claims.put("status", status);
|
claims.put("url", callback.getUrl());
|
callback.setFiletype("docx");
|
claims.put("filetype", callback.getFiletype());
|
String token = JWTUtil.createToken(claims, "test-secret".getBytes(StandardCharsets.UTF_8));
|
|
service.handleCallback(callback, "Bearer " + token);
|
|
verify(support).save(eq(session), eq(callback), any(byte[].class));
|
}
|
http.verify(() -> HttpUtil.createGet("http://document-server/saved.docx"), times(2));
|
verify(fileUploadApi, never()).replaceFile(any(), any(), any());
|
}
|
|
@Test
|
public void nonUploadSaveFailureKeepsOriginalException() {
|
OnlyOfficeSessionEntity session = session();
|
session.setBizModule("other-business");
|
when(sessionMapper.selectOne(any())).thenReturn(session);
|
OnlyOfficeCallbackParam callback = callback();
|
callback.setUrl(null);
|
String token = JWTUtil.createToken(java.util.Collections.singletonMap("scope", "test"),
|
"test-secret".getBytes(StandardCharsets.UTF_8));
|
try {
|
service.handleCallback(callback, "Bearer " + token);
|
fail("invalid save must fail");
|
} catch (DataException e) {
|
assertTrue(e.getMessage().contains("缺少下载 url"));
|
}
|
http.verifyNoInteractions();
|
}
|
|
private OnlyOfficeSessionEntity session() {
|
OnlyOfficeSessionEntity session = new OnlyOfficeSessionEntity();
|
session.setId("session-1");
|
session.setTenantId("tenant-a");
|
session.setFileId("lease-1");
|
session.setFileType("annex");
|
session.setDocKey("key-1");
|
session.setVersionNo(1);
|
return session;
|
}
|
|
private OnlyOfficeCallbackParam callback() {
|
OnlyOfficeCallbackParam callback = new OnlyOfficeCallbackParam();
|
callback.setKey("key-1");
|
callback.setStatus(2);
|
callback.setUrl("http://document-server/saved.docx");
|
return callback;
|
}
|
}
|
