Initial commit: bositao cloud (jnpf-java-cloud-v6x)

刘光辉

5 小时以前 06f314545ac3514d7ce28939920866ae8188d31c

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
package jnpf.bizcommon.onlyoffice.service.impl;
 
import jnpf.dmsEntity.download.model.DmsFileDownloadTicket;
import jnpf.dmsEntity.preview.model.DmsWatermarkSnapshot;
import jnpf.dmsEntity.preview.model.DmsFilePreviewTicket;
import jnpf.exception.DataException;
import jnpf.onlyoffice.DmsDownloadPdfRequest;
import jnpf.util.JsonUtil;
import org.junit.Test;
 
import java.util.List;
import java.util.UUID;
 
import static org.junit.Assert.*;
import static org.mockito.Mockito.*;
 
public class DmsOnlyOfficeDownloadServiceTest {
    private static final UUID VERSION = UUID.randomUUID();
 
    @Test
    public void preservesDownloadWatermarkSnapshotAndTicket() {
        DmsFileDownloadTicket ticket = ticket();
        DmsWatermarkSnapshot mark = new DmsWatermarkSnapshot();
        mark.setContent("REC-004");
        mark.setBarcodeText("Page four");
        mark.setRenderPages(List.of(4));
        ticket.setWatermarks(List.of(mark, mark));
        OnlyOfficeDmsPrintRenderer renderer = mock(OnlyOfficeDmsPrintRenderer.class);
        when(renderer.render(any(), eq("internal-source"))).thenReturn(new byte[]{1, 2});
        java.util.List<String> revoked = new java.util.ArrayList<>();
        DmsOnlyOfficeDownloadService service = new DmsOnlyOfficeDownloadService(renderer,
                value -> JsonUtil.getObjectToString(ticket), source -> {
                    assertEquals(ticket.getFileVersionId(), source.getFileVersionId());
                    assertEquals(ticket.getTenantId(), source.getTenantId());
                    assertEquals(ticket.getUserId(), source.getUserId());
                    assertEquals(ticket.getObjectName(), source.getObjectName());
                    assertEquals(ticket.getWatermarks(), source.getWatermarks());
                    assertEquals(DmsFilePreviewTicket.SCOPE, source.getScope());
                    assertTrue(source.getExpiresAt() <= ticket.getExpiresAt());
                    assertTrue(source.getExpiresAt() > System.currentTimeMillis());
                    return "internal-source";
                }, revoked::add);
 
        assertArrayEquals(new byte[]{1, 2}, service.render(new DmsDownloadPdfRequest(VERSION, "download-ticket")));
        verify(renderer).render(any(), eq("internal-source"));
        assertEquals(List.of("internal-source"), revoked);
        assertEquals(DmsFileDownloadTicket.SCOPE, ticket.getScope());
    }
 
    @Test
    public void conversionFailureAlsoRevokesTheDelegatedTicket() {
        OnlyOfficeDmsPrintRenderer renderer = mock(OnlyOfficeDmsPrintRenderer.class);
        when(renderer.render(any(), anyString())).thenThrow(new DataException("OnlyOffice 文件转换失败"));
        java.util.List<String> revoked = new java.util.ArrayList<>();
        DmsOnlyOfficeDownloadService service = new DmsOnlyOfficeDownloadService(renderer,
                value -> JsonUtil.getObjectToString(ticket()), source -> "internal-source", revoked::add);
        try {
            service.render(new DmsDownloadPdfRequest(VERSION, "ticket"));
            fail("expected conversion failure");
        } catch (DataException expected) {
            assertEquals("OnlyOffice 文件转换失败", expected.getMessage());
        }
        assertEquals(List.of("internal-source"), revoked);
    }
 
    @Test
    public void rejectsExpiredWrongScopeWrongVersionAndNonDocxTicketsBeforeRendering() {
        OnlyOfficeDmsPrintRenderer renderer = mock(OnlyOfficeDmsPrintRenderer.class);
        for (int scenario = 0; scenario < 5; scenario++) {
            DmsFileDownloadTicket ticket = ticket();
            switch (scenario) {
                case 0: ticket.setExpiresAt(System.currentTimeMillis() - 1); break;
                case 1: ticket.setScope("dms:onlyoffice:preview"); break;
                case 2: ticket.setScope("dms:onlyoffice:print"); break;
                case 3: ticket.setFileVersionId(UUID.randomUUID()); break;
                case 4: ticket.setFileName("quality.txt"); break;
            }
            DmsOnlyOfficeDownloadService service = new DmsOnlyOfficeDownloadService(renderer,
                    value -> JsonUtil.getObjectToString(ticket), source -> {
                        fail("invalid download must not issue source tickets");
                        return null;
                    }, value -> fail("no source ticket to revoke"));
            try {
                service.render(new DmsDownloadPdfRequest(VERSION, "ticket"));
                fail("invalid ticket accepted");
            } catch (DataException expected) {
                assertNotNull(expected.getMessage());
            }
        }
        verifyNoInteractions(renderer);
    }
 
    private DmsFileDownloadTicket ticket() {
        DmsFileDownloadTicket ticket = new DmsFileDownloadTicket();
        ticket.setScope(DmsFileDownloadTicket.SCOPE);
        ticket.setTenantId("tenant-1");
        ticket.setUserId("user-1");
        ticket.setFileVersionId(VERSION);
        ticket.setFileName("quality.docx");
        ticket.setStoragePlatform("seaweedfs-1");
        ticket.setBucketName("jnpf-dms");
        ticket.setObjectPath("synthetic/");
        ticket.setObjectName("quality.docx");
        ticket.setExpiresAt(System.currentTimeMillis() + 60_000);
        return ticket;
    }
}