This commit is contained in:
37
docs/snippets/java/plugins/plugin_testing.md
Normal file
37
docs/snippets/java/plugins/plugin_testing.md
Normal file
@@ -0,0 +1,37 @@
|
||||
```java title="Java"
|
||||
import dev.kreuzberg.ExtractionResult;
|
||||
import dev.kreuzberg.PostProcessor;
|
||||
import org.junit.jupiter.api.Test;
|
||||
import java.util.HashMap;
|
||||
import java.util.Map;
|
||||
import static org.junit.jupiter.api.Assertions.*;
|
||||
|
||||
class PostProcessorTest {
|
||||
@Test
|
||||
void testWordCountProcessor() {
|
||||
PostProcessor processor = result -> {
|
||||
long count = result.getContent().split("\\s+").length;
|
||||
|
||||
Map<String, Object> metadata = new HashMap<>(result.getMetadata());
|
||||
metadata.put("word_count", count);
|
||||
|
||||
return result;
|
||||
};
|
||||
|
||||
ExtractionResult input = new ExtractionResult(
|
||||
"Hello world test",
|
||||
"text/plain",
|
||||
new HashMap<>(),
|
||||
java.util.List.of(),
|
||||
java.util.List.of(),
|
||||
java.util.List.of(),
|
||||
java.util.List.of(),
|
||||
true
|
||||
);
|
||||
|
||||
ExtractionResult output = processor.process(input);
|
||||
|
||||
assertEquals(3, output.getMetadata().get("word_count"));
|
||||
}
|
||||
}
|
||||
```
|
||||
Reference in New Issue
Block a user