mindee-api-java/src/test/java/com/mindee/pdf/PdfOperationTest.java at c5805fdfe47728c3de0200c9054cf7d4ff274081 · mindee/mindee-api-java · GitHub

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
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
package com.mindee.pdf;

import com.mindee.input.PageOptions;
import com.mindee.input.PageOptionsOperation;
import com.mindee.MindeeException;
import org.junit.jupiter.api.Assertions;
import org.junit.jupiter.api.Test;

import java.io.File;
import java.io.IOException;
import java.nio.file.Files;
import java.util.ArrayList;
import java.util.List;


public class PdfOperationTest {

  private final PdfOperation pdfOperation = new PdfBoxApi();

  @Test
  public void givenADocumentAndPageToKeep_whenSplit_thenReturnsOnlyKeptPage()
    throws IOException {

    PageOptions pageOptions = new PageOptions.Builder()
        .pageIndexes(new Integer[]{ 2 })
        .operation(PageOptionsOperation.KEEP_ONLY)
        .build();

    byte[] fileBytes = Files.readAllBytes(
        new File("src/test/resources/file_types/pdf/multipage.pdf").toPath());

    SplitQuery splitQuery = new SplitQuery(fileBytes, pageOptions);
    SplitPdf splitPdf = pdfOperation.split(splitQuery);

    Assertions.assertNotNull(splitPdf);
    Assertions.assertNotNull(splitPdf.getFile());
    Assertions.assertEquals(1, splitPdf.getTotalPageNumber());
  }

  @Test
  public void givenADocumentAndListOfPagesToKeep_whenSplit_thenReturnsOnlyKeptPages()
    throws IOException {

    List<Integer> pageNumbersToKeep = new ArrayList<>();
    pageNumbersToKeep.add(0);
    pageNumbersToKeep.add(1);

    PageOptions pageOptions = new PageOptions.Builder()
        .pageIndexes(pageNumbersToKeep)
        .operation(PageOptionsOperation.KEEP_ONLY)
        .build();

    SplitQuery splitQuery = new SplitQuery(
        Files.readAllBytes(new File("src/test/resources/file_types/pdf/multipage.pdf").toPath()),
        pageOptions
    );
    SplitPdf splitPdf = pdfOperation.split(splitQuery);

    Assertions.assertNotNull(splitPdf);
    Assertions.assertNotNull(splitPdf.getFile());
    Assertions.assertEquals(2, splitPdf.getTotalPageNumber());
  }

  @Test
  public void givenADocumentAndListOfPagesToRemove_whenSplit_thenReturnsOnlyNotRemovedPages()
    throws IOException {

    PageOptions pageOptions = new PageOptions.Builder()
        .pageIndexes(new Integer[]{ 0,1,2 })
        .operation(PageOptionsOperation.REMOVE)
        .build();

    SplitQuery splitQuery = new SplitQuery(
        Files.readAllBytes(new File("src/test/resources/file_types/pdf/multipage.pdf").toPath()),
        pageOptions
    );
    SplitPdf splitPdf = pdfOperation.split(splitQuery);

    Assertions.assertNotNull(splitPdf);
    Assertions.assertNotNull(splitPdf.getFile());
    Assertions.assertEquals(9, splitPdf.getTotalPageNumber());
  }

  @Test
  public void givenADocumentOtherThantAPdf_whenSplit_mustFail()
    throws IOException {

    PageOptions pageOptions = new PageOptions.Builder()
        .pageIndexes(new Integer[]{ 1,2,3 })
        .operation(PageOptionsOperation.REMOVE)
        .build();

    SplitQuery splitQuery = new SplitQuery(
        Files.readAllBytes(new File("src/test/resources/file_types/receipt.jpg").toPath()),
        pageOptions
    );

    Assertions.assertThrows(
      MindeeException.class,
      () -> pdfOperation.split(splitQuery));
  }

  @Test
  public void givenADocumentAndListPagesToRemoveAndMinPagesCondition_whenSplit_mustNotRemovePages()
    throws IOException {

    PageOptions pageOptions = new PageOptions.Builder()
        .pageIndexes(new Integer[]{ 0 })
        .operation(PageOptionsOperation.REMOVE)
        .onMinPages(5)
        .build();

    SplitQuery splitQuery = new SplitQuery(
        Files.readAllBytes(new File("src/test/resources/file_types/pdf/multipage_cut-2.pdf").toPath()),
        pageOptions
    );
    SplitPdf splitPdf = pdfOperation.split(splitQuery);

    Assertions.assertNotNull(splitPdf);
    Assertions.assertNotNull(splitPdf.getFile());
    Assertions.assertEquals(2, splitPdf.getTotalPageNumber());
  }

  @Test
  public void givenADocumentAndNegativeListPagesToKeep_whenSplit_thenReturnsOnlyKeptPages()
    throws IOException {

    PageOptions pageOptions = new PageOptions.Builder()
        .pageIndexes(new Integer[]{ 0,-2,-1 })
        .operation(PageOptionsOperation.KEEP_ONLY)
        .build();

    SplitQuery splitQuery = new SplitQuery(
        Files.readAllBytes(new File("src/test/resources/file_types/pdf/multipage.pdf").toPath()),
        pageOptions
    );
    SplitPdf splitPdf = pdfOperation.split(splitQuery);

    Assertions.assertNotNull(splitPdf);
    Assertions.assertNotNull(splitPdf.getFile());
    Assertions.assertEquals(3, splitPdf.getTotalPageNumber());
  }
}