Convert the PdfTestUtils into a generic FileDocumentWriter

document-readers/pdf-reader/src/main/java/org/springframework/ai/reader/pdf/PdfTestUtils.java

@@ -1,62 +0,0 @@
-/*
- * Copyright 2023-2023 the original author or authors.
- *
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- *      https://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
-
-package org.springframework.ai.reader.pdf;
-
-import java.io.FileWriter;
-import java.io.IOException;
-import java.util.List;
-
-import org.springframework.ai.document.Document;
-
-/**
- * Utility class for working with PDF documents in tests.
- *
- * This utility class provides methods for writing extracted PDF documents to a file for
- * testing purposes. It can write documents with or without document markers.
- *
- * @author Christian Tzolov
- */
-public class PdfTestUtils {
-
-	private PdfTestUtils() {
-	}
-
-	/**
-	 * Writes extracted PDF documents to a file.
-	 * @param fileName The name of the file to write the documents to.
-	 * @param docs The list of {@link Document} objects to write.
-	 * @param withDocumentMarkers Whether to include document markers in the output.
-	 * @throws IOException If an I/O error occurs while writing to the file.
-	 */
-	public static void writeToFile(String fileName, List<Document> docs, boolean withDocumentMarkers)
-			throws IOException {
-		try (var writer = new FileWriter(fileName, false)) {
-
-			int i = 0;
-			for (Document doc : docs) {
-				if (withDocumentMarkers) {
-					writer.write(String.format("%n### Doc: %s, pages:[%s,%s]\n", i,
-							doc.getMetadata().get(PagePdfDocumentReader.METADATA_START_PAGE_NUMBER),
-							doc.getMetadata().get(PagePdfDocumentReader.METADATA_END_PAGE_NUMBER)));
-				}
-				writer.write(doc.getContent());
-				i++;
-			}
-		}
-	}
-
-}