Skip to content

Commit 0298fdb

Browse files
committed
Complete test for document reader
1 parent 9b8a44d commit 0298fdb

File tree

1 file changed

+6
-5
lines changed

1 file changed

+6
-5
lines changed

document-readers/pdf-reader/src/test/java/org/springframework/ai/reader/pdf/PagePdfDocumentReaderTests.java

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -29,10 +29,10 @@
2929
/**
3030
* @author Christian Tzolov
3131
*/
32-
public class PagePdfDocumentReaderTests {
32+
class PagePdfDocumentReaderTests {
3333

3434
@Test
35-
public void classpathRead() {
35+
void classpathRead() {
3636

3737
PagePdfDocumentReader pdfReader = new PagePdfDocumentReader("classpath:/sample1.pdf",
3838
PdfDocumentReaderConfig.builder()
@@ -50,20 +50,21 @@ public void classpathRead() {
5050

5151
assertThat(docs).hasSize(4);
5252

53-
String allText = docs.stream().map(d -> d.getContent()).collect(Collectors.joining(System.lineSeparator()));
53+
String allText = docs.stream().map(Document::getContent).collect(Collectors.joining(System.lineSeparator()));
5454

5555
assertThat(allText).doesNotContain(
5656
List.of("Page 1 of 4", "Page 2 of 4", "Page 3 of 4", "Page 4 of 4", "PDF Bookmark Sample"));
5757
}
5858

5959
@Test
6060
void testIndexOutOfBound() {
61-
new PagePdfDocumentReader("classpath:/sample2.pdf",
61+
var documents = new PagePdfDocumentReader("classpath:/sample2.pdf",
6262
PdfDocumentReaderConfig.builder()
6363
.withPageExtractedTextFormatter(ExtractedTextFormatter.builder()
6464
.build())
6565
.withPagesPerDocument(1)
6666
.build()).get();
67-
}
6867

68+
assertThat(documents).hasSize(64);
69+
}
6970
}

0 commit comments

Comments
 (0)