add unit tests

Author: d050150

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

@@ -22,8 +22,8 @@ class TextLine {
 
 	private static final char SPACE_CHARACTER = ' ';
 
-	private int lineLength;
-	private char[] line;
+	private final int lineLength;
+	private final char[] line;
 	private int lastIndex;
 
 	TextLine(int lineLength) {

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

@@ -1,17 +1,91 @@
 package org.springframework.ai.reader.pdf.layout;
 
 import org.junit.jupiter.api.Test;
+import org.junit.jupiter.params.ParameterizedTest;
+import org.junit.jupiter.params.provider.Arguments;
+import org.junit.jupiter.params.provider.MethodSource;
+
+import java.util.stream.Stream;
 
 import static org.junit.jupiter.api.Assertions.assertEquals;
 
 
 class TextLineTest {
 
+	public static Stream<Arguments> testWriteCharacterAtIndexValidIndex() {
+		return Stream.of(
+				Arguments.of(new Character('A', 0, false, false, false, false)),
+				Arguments.of(new Character('A', 10, true, false, false, false)),
+				Arguments.of(new Character('A', 0, false, true, false, false))
+		);
+	}
+
+	@ParameterizedTest
+	@MethodSource
+	void testWriteCharacterAtIndexValidIndex(Character character) {
+		TextLine textLine = new TextLine(100);
+		textLine.writeCharacterAtIndex(character);
+		assertEquals(" A" + " ".repeat(23), textLine.getLine());
+	}
+
+	@Test
+	void testWriteCharacterAtIndex_PartOfPreviousWord() {
+		TextLine textLine = new TextLine(100);
+		Character character = new Character('A', 10, true, false, false, false);
+		textLine.writeCharacterAtIndex(character);
+		assertEquals(" A" + " ".repeat(23), textLine.getLine());
+	}
+
 	@Test
 	void testWriteCharacterAtIndex_BeginningOfNewLine() {
 		TextLine textLine = new TextLine(100);
 		Character character = new Character('A', 0, false, true, false, false);
 		textLine.writeCharacterAtIndex(character);
-		assertEquals("A" + " ".repeat(99), textLine.getLine());
+		assertEquals(" A" + " ".repeat(23), textLine.getLine());
+	}
+
+	@Test
+	void testWriteCharacterAtIndex_InvalidIndex() {
+		TextLine textLine = new TextLine(100);
+		Character character = new Character('A', 150, false, false, false, false);
+		textLine.writeCharacterAtIndex(character);
+		assertEquals(" ".repeat(25), textLine.getLine());
+	}
+
+	@Test
+	void testWriteCharacterAtIndex_NegativeIndex() {
+		TextLine textLine = new TextLine(100);
+		Character character = new Character('A', -1, false, false, false, false);
+		textLine.writeCharacterAtIndex(character);
+		assertEquals(" ".repeat(25), textLine.getLine());
+	}
+
+	@Test
+	void testWriteCharacterAtIndex_SpaceCharacter() {
+		TextLine textLine = new TextLine(100);
+		Character character = new Character('A', 10, false, false, false, false);
+		textLine.writeCharacterAtIndex(character);
+		assertEquals(" ".repeat(11) + "A" + " ".repeat(13), textLine.getLine());
+	}
+
+
+	@Test
+	void testWriteCharacterAtIndex_CloseToPreviousWord() {
+		TextLine textLine = new TextLine(100);
+		Character character = new Character('A', 10, false, false, true, false);
+		textLine.writeCharacterAtIndex(character);
+		assertEquals(" ".repeat(11) + "A" + " ".repeat(13), textLine.getLine());
+	}
+
+	@Test
+	void testGetLineLength() {
+		TextLine textLine = new TextLine(100);
+		assertEquals(100 / ForkPDFLayoutTextStripper.OUTPUT_SPACE_CHARACTER_WIDTH_IN_PT, textLine.getLineLength());
+	}
+
+	@Test
+	void testGetLine() {
+		TextLine textLine = new TextLine(100);
+		assertEquals(" ".repeat(100 / ForkPDFLayoutTextStripper.OUTPUT_SPACE_CHARACTER_WIDTH_IN_PT), textLine.getLine());
 	}
 }