kreuzberg 4.0.0.rc2 → 4.0.1

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (446) hide show
  1. checksums.yaml +4 -4
  2. data/.gitignore +14 -14
  3. data/.rspec +3 -3
  4. data/.rubocop.yaml +1 -1
  5. data/.rubocop.yml +543 -538
  6. data/Gemfile +8 -8
  7. data/Gemfile.lock +194 -6
  8. data/README.md +391 -426
  9. data/Rakefile +34 -25
  10. data/Steepfile +51 -47
  11. data/examples/async_patterns.rb +283 -341
  12. data/ext/kreuzberg_rb/extconf.rb +65 -45
  13. data/ext/kreuzberg_rb/native/.cargo/config.toml +23 -0
  14. data/ext/kreuzberg_rb/native/Cargo.lock +7619 -6535
  15. data/ext/kreuzberg_rb/native/Cargo.toml +75 -44
  16. data/ext/kreuzberg_rb/native/README.md +425 -425
  17. data/ext/kreuzberg_rb/native/build.rs +15 -15
  18. data/ext/kreuzberg_rb/native/include/ieeefp.h +11 -11
  19. data/ext/kreuzberg_rb/native/include/msvc_compat/strings.h +14 -14
  20. data/ext/kreuzberg_rb/native/include/strings.h +20 -20
  21. data/ext/kreuzberg_rb/native/include/unistd.h +47 -47
  22. data/ext/kreuzberg_rb/native/src/lib.rs +3802 -2998
  23. data/extconf.rb +60 -28
  24. data/kreuzberg.gemspec +199 -148
  25. data/lib/kreuzberg/api_proxy.rb +126 -142
  26. data/lib/kreuzberg/cache_api.rb +67 -46
  27. data/lib/kreuzberg/cli.rb +47 -55
  28. data/lib/kreuzberg/cli_proxy.rb +117 -127
  29. data/lib/kreuzberg/config.rb +936 -691
  30. data/lib/kreuzberg/error_context.rb +136 -32
  31. data/lib/kreuzberg/errors.rb +116 -118
  32. data/lib/kreuzberg/extraction_api.rb +313 -85
  33. data/lib/kreuzberg/mcp_proxy.rb +177 -186
  34. data/lib/kreuzberg/ocr_backend_protocol.rb +40 -113
  35. data/lib/kreuzberg/post_processor_protocol.rb +15 -86
  36. data/lib/kreuzberg/result.rb +334 -216
  37. data/lib/kreuzberg/setup_lib_path.rb +99 -80
  38. data/lib/kreuzberg/types.rb +170 -0
  39. data/lib/kreuzberg/validator_protocol.rb +16 -89
  40. data/lib/kreuzberg/version.rb +5 -5
  41. data/lib/kreuzberg.rb +96 -103
  42. data/lib/libpdfium.so +0 -0
  43. data/sig/kreuzberg/internal.rbs +184 -184
  44. data/sig/kreuzberg.rbs +561 -520
  45. data/spec/binding/async_operations_spec.rb +473 -0
  46. data/spec/binding/batch_operations_spec.rb +595 -0
  47. data/spec/binding/batch_spec.rb +359 -0
  48. data/spec/binding/cache_spec.rb +227 -227
  49. data/spec/binding/cli_proxy_spec.rb +85 -85
  50. data/spec/binding/cli_spec.rb +55 -55
  51. data/spec/binding/config_result_spec.rb +377 -0
  52. data/spec/binding/config_spec.rb +419 -345
  53. data/spec/binding/config_validation_spec.rb +377 -283
  54. data/spec/binding/embeddings_spec.rb +816 -0
  55. data/spec/binding/error_handling_spec.rb +399 -213
  56. data/spec/binding/error_recovery_spec.rb +488 -0
  57. data/spec/binding/errors_spec.rb +66 -66
  58. data/spec/binding/font_config_spec.rb +220 -0
  59. data/spec/binding/images_spec.rb +738 -0
  60. data/spec/binding/keywords_extraction_spec.rb +600 -0
  61. data/spec/binding/metadata_types_spec.rb +1228 -0
  62. data/spec/binding/pages_extraction_spec.rb +471 -0
  63. data/spec/binding/plugins/ocr_backend_spec.rb +307 -307
  64. data/spec/binding/plugins/postprocessor_spec.rb +269 -269
  65. data/spec/binding/plugins/validator_spec.rb +273 -274
  66. data/spec/binding/tables_spec.rb +641 -0
  67. data/spec/fixtures/config.toml +38 -39
  68. data/spec/fixtures/config.yaml +41 -41
  69. data/spec/fixtures/invalid_config.toml +3 -4
  70. data/spec/smoke/package_spec.rb +177 -178
  71. data/spec/spec_helper.rb +40 -42
  72. data/spec/unit/config/chunking_config_spec.rb +213 -0
  73. data/spec/unit/config/embedding_config_spec.rb +343 -0
  74. data/spec/unit/config/extraction_config_spec.rb +438 -0
  75. data/spec/unit/config/font_config_spec.rb +285 -0
  76. data/spec/unit/config/hierarchy_config_spec.rb +314 -0
  77. data/spec/unit/config/image_extraction_config_spec.rb +209 -0
  78. data/spec/unit/config/image_preprocessing_config_spec.rb +249 -0
  79. data/spec/unit/config/keyword_config_spec.rb +229 -0
  80. data/spec/unit/config/language_detection_config_spec.rb +258 -0
  81. data/spec/unit/config/ocr_config_spec.rb +171 -0
  82. data/spec/unit/config/page_config_spec.rb +221 -0
  83. data/spec/unit/config/pdf_config_spec.rb +267 -0
  84. data/spec/unit/config/postprocessor_config_spec.rb +290 -0
  85. data/spec/unit/config/tesseract_config_spec.rb +181 -0
  86. data/spec/unit/config/token_reduction_config_spec.rb +251 -0
  87. data/test/metadata_types_test.rb +959 -0
  88. data/vendor/Cargo.toml +61 -0
  89. data/vendor/kreuzberg/Cargo.toml +259 -204
  90. data/vendor/kreuzberg/README.md +263 -175
  91. data/vendor/kreuzberg/build.rs +782 -474
  92. data/vendor/kreuzberg/examples/bench_fixes.rs +71 -0
  93. data/vendor/kreuzberg/examples/test_pdfium_fork.rs +62 -0
  94. data/vendor/kreuzberg/src/api/error.rs +81 -81
  95. data/vendor/kreuzberg/src/api/handlers.rs +320 -199
  96. data/vendor/kreuzberg/src/api/mod.rs +94 -79
  97. data/vendor/kreuzberg/src/api/server.rs +518 -353
  98. data/vendor/kreuzberg/src/api/types.rs +206 -170
  99. data/vendor/kreuzberg/src/cache/mod.rs +1167 -1167
  100. data/vendor/kreuzberg/src/chunking/mod.rs +2303 -677
  101. data/vendor/kreuzberg/src/chunking/processor.rs +219 -0
  102. data/vendor/kreuzberg/src/core/batch_mode.rs +95 -95
  103. data/vendor/kreuzberg/src/core/batch_optimizations.rs +385 -0
  104. data/vendor/kreuzberg/src/core/config.rs +1914 -1032
  105. data/vendor/kreuzberg/src/core/config_validation.rs +949 -0
  106. data/vendor/kreuzberg/src/core/extractor.rs +1200 -1024
  107. data/vendor/kreuzberg/src/core/formats.rs +235 -0
  108. data/vendor/kreuzberg/src/core/io.rs +329 -329
  109. data/vendor/kreuzberg/src/core/mime.rs +605 -605
  110. data/vendor/kreuzberg/src/core/mod.rs +61 -45
  111. data/vendor/kreuzberg/src/core/pipeline.rs +1223 -984
  112. data/vendor/kreuzberg/src/core/server_config.rs +1220 -0
  113. data/vendor/kreuzberg/src/embeddings.rs +471 -432
  114. data/vendor/kreuzberg/src/error.rs +431 -431
  115. data/vendor/kreuzberg/src/extraction/archive.rs +959 -954
  116. data/vendor/kreuzberg/src/extraction/capacity.rs +263 -0
  117. data/vendor/kreuzberg/src/extraction/docx.rs +404 -40
  118. data/vendor/kreuzberg/src/extraction/email.rs +855 -854
  119. data/vendor/kreuzberg/src/extraction/excel.rs +697 -688
  120. data/vendor/kreuzberg/src/extraction/html.rs +1830 -553
  121. data/vendor/kreuzberg/src/extraction/image.rs +492 -368
  122. data/vendor/kreuzberg/src/extraction/libreoffice.rs +574 -563
  123. data/vendor/kreuzberg/src/extraction/markdown.rs +216 -213
  124. data/vendor/kreuzberg/src/extraction/mod.rs +93 -81
  125. data/vendor/kreuzberg/src/extraction/office_metadata/app_properties.rs +398 -398
  126. data/vendor/kreuzberg/src/extraction/office_metadata/core_properties.rs +247 -247
  127. data/vendor/kreuzberg/src/extraction/office_metadata/custom_properties.rs +240 -240
  128. data/vendor/kreuzberg/src/extraction/office_metadata/mod.rs +130 -130
  129. data/vendor/kreuzberg/src/extraction/office_metadata/odt_properties.rs +284 -287
  130. data/vendor/kreuzberg/src/extraction/pptx.rs +3102 -3000
  131. data/vendor/kreuzberg/src/extraction/structured.rs +491 -490
  132. data/vendor/kreuzberg/src/extraction/table.rs +329 -328
  133. data/vendor/kreuzberg/src/extraction/text.rs +277 -269
  134. data/vendor/kreuzberg/src/extraction/xml.rs +333 -333
  135. data/vendor/kreuzberg/src/extractors/archive.rs +447 -446
  136. data/vendor/kreuzberg/src/extractors/bibtex.rs +470 -469
  137. data/vendor/kreuzberg/src/extractors/docbook.rs +504 -502
  138. data/vendor/kreuzberg/src/extractors/docx.rs +400 -367
  139. data/vendor/kreuzberg/src/extractors/email.rs +157 -143
  140. data/vendor/kreuzberg/src/extractors/epub.rs +696 -707
  141. data/vendor/kreuzberg/src/extractors/excel.rs +385 -343
  142. data/vendor/kreuzberg/src/extractors/fictionbook.rs +492 -491
  143. data/vendor/kreuzberg/src/extractors/html.rs +419 -393
  144. data/vendor/kreuzberg/src/extractors/image.rs +219 -198
  145. data/vendor/kreuzberg/src/extractors/jats.rs +1054 -1051
  146. data/vendor/kreuzberg/src/extractors/jupyter.rs +368 -367
  147. data/vendor/kreuzberg/src/extractors/latex.rs +653 -652
  148. data/vendor/kreuzberg/src/extractors/markdown.rs +701 -700
  149. data/vendor/kreuzberg/src/extractors/mod.rs +429 -365
  150. data/vendor/kreuzberg/src/extractors/odt.rs +628 -628
  151. data/vendor/kreuzberg/src/extractors/opml.rs +635 -634
  152. data/vendor/kreuzberg/src/extractors/orgmode.rs +529 -528
  153. data/vendor/kreuzberg/src/extractors/pdf.rs +761 -493
  154. data/vendor/kreuzberg/src/extractors/pptx.rs +279 -248
  155. data/vendor/kreuzberg/src/extractors/rst.rs +577 -576
  156. data/vendor/kreuzberg/src/extractors/rtf.rs +809 -810
  157. data/vendor/kreuzberg/src/extractors/security.rs +484 -484
  158. data/vendor/kreuzberg/src/extractors/security_tests.rs +367 -367
  159. data/vendor/kreuzberg/src/extractors/structured.rs +142 -140
  160. data/vendor/kreuzberg/src/extractors/text.rs +265 -260
  161. data/vendor/kreuzberg/src/extractors/typst.rs +651 -650
  162. data/vendor/kreuzberg/src/extractors/xml.rs +147 -135
  163. data/vendor/kreuzberg/src/image/dpi.rs +164 -164
  164. data/vendor/kreuzberg/src/image/mod.rs +6 -6
  165. data/vendor/kreuzberg/src/image/preprocessing.rs +417 -417
  166. data/vendor/kreuzberg/src/image/resize.rs +89 -89
  167. data/vendor/kreuzberg/src/keywords/config.rs +154 -154
  168. data/vendor/kreuzberg/src/keywords/mod.rs +237 -237
  169. data/vendor/kreuzberg/src/keywords/processor.rs +275 -267
  170. data/vendor/kreuzberg/src/keywords/rake.rs +293 -293
  171. data/vendor/kreuzberg/src/keywords/types.rs +68 -68
  172. data/vendor/kreuzberg/src/keywords/yake.rs +163 -163
  173. data/vendor/kreuzberg/src/language_detection/mod.rs +985 -942
  174. data/vendor/kreuzberg/src/language_detection/processor.rs +218 -0
  175. data/vendor/kreuzberg/src/lib.rs +114 -105
  176. data/vendor/kreuzberg/src/mcp/mod.rs +35 -32
  177. data/vendor/kreuzberg/src/mcp/server.rs +2090 -1968
  178. data/vendor/kreuzberg/src/ocr/cache.rs +469 -469
  179. data/vendor/kreuzberg/src/ocr/error.rs +37 -37
  180. data/vendor/kreuzberg/src/ocr/hocr.rs +216 -216
  181. data/vendor/kreuzberg/src/ocr/language_registry.rs +520 -0
  182. data/vendor/kreuzberg/src/ocr/mod.rs +60 -58
  183. data/vendor/kreuzberg/src/ocr/processor.rs +858 -863
  184. data/vendor/kreuzberg/src/ocr/table/mod.rs +4 -4
  185. data/vendor/kreuzberg/src/ocr/table/tsv_parser.rs +144 -144
  186. data/vendor/kreuzberg/src/ocr/tesseract_backend.rs +456 -450
  187. data/vendor/kreuzberg/src/ocr/types.rs +393 -393
  188. data/vendor/kreuzberg/src/ocr/utils.rs +47 -47
  189. data/vendor/kreuzberg/src/ocr/validation.rs +206 -206
  190. data/vendor/kreuzberg/src/panic_context.rs +154 -154
  191. data/vendor/kreuzberg/src/pdf/bindings.rs +306 -0
  192. data/vendor/kreuzberg/src/pdf/bundled.rs +408 -0
  193. data/vendor/kreuzberg/src/pdf/error.rs +214 -122
  194. data/vendor/kreuzberg/src/pdf/fonts.rs +358 -0
  195. data/vendor/kreuzberg/src/pdf/hierarchy.rs +903 -0
  196. data/vendor/kreuzberg/src/pdf/images.rs +139 -139
  197. data/vendor/kreuzberg/src/pdf/metadata.rs +509 -346
  198. data/vendor/kreuzberg/src/pdf/mod.rs +81 -50
  199. data/vendor/kreuzberg/src/pdf/rendering.rs +369 -369
  200. data/vendor/kreuzberg/src/pdf/table.rs +417 -393
  201. data/vendor/kreuzberg/src/pdf/text.rs +553 -158
  202. data/vendor/kreuzberg/src/plugins/extractor.rs +1042 -1013
  203. data/vendor/kreuzberg/src/plugins/mod.rs +212 -209
  204. data/vendor/kreuzberg/src/plugins/ocr.rs +637 -620
  205. data/vendor/kreuzberg/src/plugins/processor.rs +650 -642
  206. data/vendor/kreuzberg/src/plugins/registry.rs +1339 -1337
  207. data/vendor/kreuzberg/src/plugins/traits.rs +258 -258
  208. data/vendor/kreuzberg/src/plugins/validator.rs +967 -956
  209. data/vendor/kreuzberg/src/stopwords/mod.rs +1470 -1470
  210. data/vendor/kreuzberg/src/text/mod.rs +27 -19
  211. data/vendor/kreuzberg/src/text/quality.rs +710 -697
  212. data/vendor/kreuzberg/src/text/quality_processor.rs +231 -0
  213. data/vendor/kreuzberg/src/text/string_utils.rs +229 -217
  214. data/vendor/kreuzberg/src/text/token_reduction/cjk_utils.rs +164 -164
  215. data/vendor/kreuzberg/src/text/token_reduction/config.rs +100 -100
  216. data/vendor/kreuzberg/src/text/token_reduction/core.rs +832 -796
  217. data/vendor/kreuzberg/src/text/token_reduction/filters.rs +923 -902
  218. data/vendor/kreuzberg/src/text/token_reduction/mod.rs +160 -160
  219. data/vendor/kreuzberg/src/text/token_reduction/semantic.rs +619 -619
  220. data/vendor/kreuzberg/src/text/token_reduction/simd_text.rs +148 -147
  221. data/vendor/kreuzberg/src/text/utf8_validation.rs +193 -0
  222. data/vendor/kreuzberg/src/types.rs +1713 -903
  223. data/vendor/kreuzberg/src/utils/mod.rs +31 -17
  224. data/vendor/kreuzberg/src/utils/pool.rs +503 -0
  225. data/vendor/kreuzberg/src/utils/pool_sizing.rs +364 -0
  226. data/vendor/kreuzberg/src/utils/quality.rs +968 -959
  227. data/vendor/kreuzberg/src/utils/string_pool.rs +761 -0
  228. data/vendor/kreuzberg/src/utils/string_utils.rs +381 -381
  229. data/vendor/kreuzberg/stopwords/af_stopwords.json +53 -53
  230. data/vendor/kreuzberg/stopwords/ar_stopwords.json +482 -482
  231. data/vendor/kreuzberg/stopwords/bg_stopwords.json +261 -261
  232. data/vendor/kreuzberg/stopwords/bn_stopwords.json +400 -400
  233. data/vendor/kreuzberg/stopwords/br_stopwords.json +1205 -1205
  234. data/vendor/kreuzberg/stopwords/ca_stopwords.json +280 -280
  235. data/vendor/kreuzberg/stopwords/cs_stopwords.json +425 -425
  236. data/vendor/kreuzberg/stopwords/da_stopwords.json +172 -172
  237. data/vendor/kreuzberg/stopwords/de_stopwords.json +622 -622
  238. data/vendor/kreuzberg/stopwords/el_stopwords.json +849 -849
  239. data/vendor/kreuzberg/stopwords/en_stopwords.json +1300 -1300
  240. data/vendor/kreuzberg/stopwords/eo_stopwords.json +175 -175
  241. data/vendor/kreuzberg/stopwords/es_stopwords.json +734 -734
  242. data/vendor/kreuzberg/stopwords/et_stopwords.json +37 -37
  243. data/vendor/kreuzberg/stopwords/eu_stopwords.json +100 -100
  244. data/vendor/kreuzberg/stopwords/fa_stopwords.json +801 -801
  245. data/vendor/kreuzberg/stopwords/fi_stopwords.json +849 -849
  246. data/vendor/kreuzberg/stopwords/fr_stopwords.json +693 -693
  247. data/vendor/kreuzberg/stopwords/ga_stopwords.json +111 -111
  248. data/vendor/kreuzberg/stopwords/gl_stopwords.json +162 -162
  249. data/vendor/kreuzberg/stopwords/gu_stopwords.json +226 -226
  250. data/vendor/kreuzberg/stopwords/ha_stopwords.json +41 -41
  251. data/vendor/kreuzberg/stopwords/he_stopwords.json +196 -196
  252. data/vendor/kreuzberg/stopwords/hi_stopwords.json +227 -227
  253. data/vendor/kreuzberg/stopwords/hr_stopwords.json +181 -181
  254. data/vendor/kreuzberg/stopwords/hu_stopwords.json +791 -791
  255. data/vendor/kreuzberg/stopwords/hy_stopwords.json +47 -47
  256. data/vendor/kreuzberg/stopwords/id_stopwords.json +760 -760
  257. data/vendor/kreuzberg/stopwords/it_stopwords.json +634 -634
  258. data/vendor/kreuzberg/stopwords/ja_stopwords.json +136 -136
  259. data/vendor/kreuzberg/stopwords/kn_stopwords.json +84 -84
  260. data/vendor/kreuzberg/stopwords/ko_stopwords.json +681 -681
  261. data/vendor/kreuzberg/stopwords/ku_stopwords.json +64 -64
  262. data/vendor/kreuzberg/stopwords/la_stopwords.json +51 -51
  263. data/vendor/kreuzberg/stopwords/lt_stopwords.json +476 -476
  264. data/vendor/kreuzberg/stopwords/lv_stopwords.json +163 -163
  265. data/vendor/kreuzberg/stopwords/ml_stopwords.json +1 -1
  266. data/vendor/kreuzberg/stopwords/mr_stopwords.json +101 -101
  267. data/vendor/kreuzberg/stopwords/ms_stopwords.json +477 -477
  268. data/vendor/kreuzberg/stopwords/ne_stopwords.json +490 -490
  269. data/vendor/kreuzberg/stopwords/nl_stopwords.json +415 -415
  270. data/vendor/kreuzberg/stopwords/no_stopwords.json +223 -223
  271. data/vendor/kreuzberg/stopwords/pl_stopwords.json +331 -331
  272. data/vendor/kreuzberg/stopwords/pt_stopwords.json +562 -562
  273. data/vendor/kreuzberg/stopwords/ro_stopwords.json +436 -436
  274. data/vendor/kreuzberg/stopwords/ru_stopwords.json +561 -561
  275. data/vendor/kreuzberg/stopwords/si_stopwords.json +193 -193
  276. data/vendor/kreuzberg/stopwords/sk_stopwords.json +420 -420
  277. data/vendor/kreuzberg/stopwords/sl_stopwords.json +448 -448
  278. data/vendor/kreuzberg/stopwords/so_stopwords.json +32 -32
  279. data/vendor/kreuzberg/stopwords/st_stopwords.json +33 -33
  280. data/vendor/kreuzberg/stopwords/sv_stopwords.json +420 -420
  281. data/vendor/kreuzberg/stopwords/sw_stopwords.json +76 -76
  282. data/vendor/kreuzberg/stopwords/ta_stopwords.json +129 -129
  283. data/vendor/kreuzberg/stopwords/te_stopwords.json +54 -54
  284. data/vendor/kreuzberg/stopwords/th_stopwords.json +118 -118
  285. data/vendor/kreuzberg/stopwords/tl_stopwords.json +149 -149
  286. data/vendor/kreuzberg/stopwords/tr_stopwords.json +506 -506
  287. data/vendor/kreuzberg/stopwords/uk_stopwords.json +75 -75
  288. data/vendor/kreuzberg/stopwords/ur_stopwords.json +519 -519
  289. data/vendor/kreuzberg/stopwords/vi_stopwords.json +647 -647
  290. data/vendor/kreuzberg/stopwords/yo_stopwords.json +62 -62
  291. data/vendor/kreuzberg/stopwords/zh_stopwords.json +796 -796
  292. data/vendor/kreuzberg/stopwords/zu_stopwords.json +31 -31
  293. data/vendor/kreuzberg/tests/api_embed.rs +360 -0
  294. data/vendor/kreuzberg/tests/api_extract_multipart.rs +52 -52
  295. data/vendor/kreuzberg/tests/api_large_pdf_extraction.rs +471 -0
  296. data/vendor/kreuzberg/tests/api_large_pdf_extraction_diagnostics.rs +289 -0
  297. data/vendor/kreuzberg/tests/api_tests.rs +1472 -966
  298. data/vendor/kreuzberg/tests/archive_integration.rs +545 -543
  299. data/vendor/kreuzberg/tests/batch_orchestration.rs +587 -556
  300. data/vendor/kreuzberg/tests/batch_pooling_benchmark.rs +154 -0
  301. data/vendor/kreuzberg/tests/batch_processing.rs +328 -316
  302. data/vendor/kreuzberg/tests/bibtex_parity_test.rs +421 -421
  303. data/vendor/kreuzberg/tests/concurrency_stress.rs +541 -525
  304. data/vendor/kreuzberg/tests/config_features.rs +612 -598
  305. data/vendor/kreuzberg/tests/config_integration_test.rs +753 -0
  306. data/vendor/kreuzberg/tests/config_loading_tests.rs +416 -415
  307. data/vendor/kreuzberg/tests/core_integration.rs +519 -510
  308. data/vendor/kreuzberg/tests/csv_integration.rs +414 -414
  309. data/vendor/kreuzberg/tests/data/hierarchy_ground_truth.json +294 -0
  310. data/vendor/kreuzberg/tests/docbook_extractor_tests.rs +500 -498
  311. data/vendor/kreuzberg/tests/docx_metadata_extraction_test.rs +122 -122
  312. data/vendor/kreuzberg/tests/docx_vs_pandoc_comparison.rs +370 -370
  313. data/vendor/kreuzberg/tests/email_integration.rs +327 -325
  314. data/vendor/kreuzberg/tests/epub_native_extractor_tests.rs +275 -275
  315. data/vendor/kreuzberg/tests/error_handling.rs +402 -393
  316. data/vendor/kreuzberg/tests/fictionbook_extractor_tests.rs +228 -228
  317. data/vendor/kreuzberg/tests/format_integration.rs +165 -159
  318. data/vendor/kreuzberg/tests/helpers/mod.rs +202 -142
  319. data/vendor/kreuzberg/tests/html_table_test.rs +551 -551
  320. data/vendor/kreuzberg/tests/image_integration.rs +255 -253
  321. data/vendor/kreuzberg/tests/instrumentation_test.rs +139 -139
  322. data/vendor/kreuzberg/tests/jats_extractor_tests.rs +639 -639
  323. data/vendor/kreuzberg/tests/jupyter_extractor_tests.rs +704 -704
  324. data/vendor/kreuzberg/tests/keywords_integration.rs +479 -479
  325. data/vendor/kreuzberg/tests/keywords_quality.rs +509 -509
  326. data/vendor/kreuzberg/tests/latex_extractor_tests.rs +496 -496
  327. data/vendor/kreuzberg/tests/markdown_extractor_tests.rs +490 -490
  328. data/vendor/kreuzberg/tests/mime_detection.rs +429 -428
  329. data/vendor/kreuzberg/tests/ocr_configuration.rs +514 -510
  330. data/vendor/kreuzberg/tests/ocr_errors.rs +698 -676
  331. data/vendor/kreuzberg/tests/ocr_language_registry.rs +191 -0
  332. data/vendor/kreuzberg/tests/ocr_quality.rs +629 -627
  333. data/vendor/kreuzberg/tests/ocr_stress.rs +469 -469
  334. data/vendor/kreuzberg/tests/odt_extractor_tests.rs +674 -695
  335. data/vendor/kreuzberg/tests/opml_extractor_tests.rs +616 -616
  336. data/vendor/kreuzberg/tests/orgmode_extractor_tests.rs +822 -822
  337. data/vendor/kreuzberg/tests/page_markers.rs +297 -0
  338. data/vendor/kreuzberg/tests/pdf_hierarchy_detection.rs +301 -0
  339. data/vendor/kreuzberg/tests/pdf_hierarchy_quality.rs +589 -0
  340. data/vendor/kreuzberg/tests/pdf_integration.rs +45 -43
  341. data/vendor/kreuzberg/tests/pdf_ocr_triggering.rs +301 -0
  342. data/vendor/kreuzberg/tests/pdf_text_merging.rs +475 -0
  343. data/vendor/kreuzberg/tests/pdfium_linking.rs +340 -0
  344. data/vendor/kreuzberg/tests/pipeline_integration.rs +1446 -1411
  345. data/vendor/kreuzberg/tests/plugin_ocr_backend_test.rs +776 -771
  346. data/vendor/kreuzberg/tests/plugin_postprocessor_test.rs +577 -560
  347. data/vendor/kreuzberg/tests/plugin_system.rs +927 -921
  348. data/vendor/kreuzberg/tests/plugin_validator_test.rs +783 -783
  349. data/vendor/kreuzberg/tests/registry_integration_tests.rs +587 -586
  350. data/vendor/kreuzberg/tests/rst_extractor_tests.rs +694 -692
  351. data/vendor/kreuzberg/tests/rtf_extractor_tests.rs +775 -776
  352. data/vendor/kreuzberg/tests/security_validation.rs +416 -415
  353. data/vendor/kreuzberg/tests/stopwords_integration_test.rs +888 -888
  354. data/vendor/kreuzberg/tests/test_fastembed.rs +631 -609
  355. data/vendor/kreuzberg/tests/typst_behavioral_tests.rs +1260 -1259
  356. data/vendor/kreuzberg/tests/typst_extractor_tests.rs +648 -647
  357. data/vendor/kreuzberg/tests/xlsx_metadata_extraction_test.rs +87 -87
  358. data/vendor/kreuzberg-ffi/Cargo.toml +67 -0
  359. data/vendor/kreuzberg-ffi/README.md +851 -0
  360. data/vendor/kreuzberg-ffi/benches/result_view_benchmark.rs +227 -0
  361. data/vendor/kreuzberg-ffi/build.rs +168 -0
  362. data/vendor/kreuzberg-ffi/cbindgen.toml +37 -0
  363. data/vendor/kreuzberg-ffi/kreuzberg-ffi.pc.in +12 -0
  364. data/vendor/kreuzberg-ffi/kreuzberg.h +3012 -0
  365. data/vendor/kreuzberg-ffi/src/batch_streaming.rs +588 -0
  366. data/vendor/kreuzberg-ffi/src/config.rs +1341 -0
  367. data/vendor/kreuzberg-ffi/src/error.rs +901 -0
  368. data/vendor/kreuzberg-ffi/src/extraction.rs +555 -0
  369. data/vendor/kreuzberg-ffi/src/helpers.rs +879 -0
  370. data/vendor/kreuzberg-ffi/src/lib.rs +977 -0
  371. data/vendor/kreuzberg-ffi/src/memory.rs +493 -0
  372. data/vendor/kreuzberg-ffi/src/mime.rs +329 -0
  373. data/vendor/kreuzberg-ffi/src/panic_shield.rs +265 -0
  374. data/vendor/kreuzberg-ffi/src/plugins/document_extractor.rs +442 -0
  375. data/vendor/kreuzberg-ffi/src/plugins/mod.rs +14 -0
  376. data/vendor/kreuzberg-ffi/src/plugins/ocr_backend.rs +628 -0
  377. data/vendor/kreuzberg-ffi/src/plugins/post_processor.rs +438 -0
  378. data/vendor/kreuzberg-ffi/src/plugins/validator.rs +329 -0
  379. data/vendor/kreuzberg-ffi/src/result.rs +510 -0
  380. data/vendor/kreuzberg-ffi/src/result_pool.rs +639 -0
  381. data/vendor/kreuzberg-ffi/src/result_view.rs +773 -0
  382. data/vendor/kreuzberg-ffi/src/string_intern.rs +568 -0
  383. data/vendor/kreuzberg-ffi/src/types.rs +363 -0
  384. data/vendor/kreuzberg-ffi/src/util.rs +210 -0
  385. data/vendor/kreuzberg-ffi/src/validation.rs +848 -0
  386. data/vendor/kreuzberg-ffi/tests.disabled/README.md +48 -0
  387. data/vendor/kreuzberg-ffi/tests.disabled/config_loading_tests.rs +299 -0
  388. data/vendor/kreuzberg-ffi/tests.disabled/config_tests.rs +346 -0
  389. data/vendor/kreuzberg-ffi/tests.disabled/extractor_tests.rs +232 -0
  390. data/vendor/kreuzberg-ffi/tests.disabled/plugin_registration_tests.rs +470 -0
  391. data/vendor/kreuzberg-tesseract/.commitlintrc.json +13 -0
  392. data/vendor/kreuzberg-tesseract/.crate-ignore +2 -0
  393. data/vendor/kreuzberg-tesseract/Cargo.lock +2933 -0
  394. data/vendor/kreuzberg-tesseract/Cargo.toml +57 -0
  395. data/vendor/{rb-sys/LICENSE-MIT → kreuzberg-tesseract/LICENSE} +22 -21
  396. data/vendor/kreuzberg-tesseract/README.md +399 -0
  397. data/vendor/kreuzberg-tesseract/build.rs +1127 -0
  398. data/vendor/kreuzberg-tesseract/patches/README.md +71 -0
  399. data/vendor/kreuzberg-tesseract/patches/tesseract.diff +199 -0
  400. data/vendor/kreuzberg-tesseract/src/api.rs +1371 -0
  401. data/vendor/kreuzberg-tesseract/src/choice_iterator.rs +77 -0
  402. data/vendor/kreuzberg-tesseract/src/enums.rs +297 -0
  403. data/vendor/kreuzberg-tesseract/src/error.rs +81 -0
  404. data/vendor/kreuzberg-tesseract/src/lib.rs +145 -0
  405. data/vendor/kreuzberg-tesseract/src/monitor.rs +57 -0
  406. data/vendor/kreuzberg-tesseract/src/mutable_iterator.rs +197 -0
  407. data/vendor/kreuzberg-tesseract/src/page_iterator.rs +253 -0
  408. data/vendor/kreuzberg-tesseract/src/result_iterator.rs +286 -0
  409. data/vendor/kreuzberg-tesseract/src/result_renderer.rs +183 -0
  410. data/vendor/kreuzberg-tesseract/tests/integration_test.rs +211 -0
  411. metadata +196 -45
  412. data/vendor/kreuzberg/benches/otel_overhead.rs +0 -48
  413. data/vendor/kreuzberg/src/extractors/fictionbook.rs.backup2 +0 -738
  414. data/vendor/rb-sys/.cargo-ok +0 -1
  415. data/vendor/rb-sys/.cargo_vcs_info.json +0 -6
  416. data/vendor/rb-sys/Cargo.lock +0 -393
  417. data/vendor/rb-sys/Cargo.toml +0 -70
  418. data/vendor/rb-sys/Cargo.toml.orig +0 -57
  419. data/vendor/rb-sys/LICENSE-APACHE +0 -190
  420. data/vendor/rb-sys/bin/release.sh +0 -21
  421. data/vendor/rb-sys/build/features.rs +0 -108
  422. data/vendor/rb-sys/build/main.rs +0 -246
  423. data/vendor/rb-sys/build/stable_api_config.rs +0 -153
  424. data/vendor/rb-sys/build/version.rs +0 -48
  425. data/vendor/rb-sys/readme.md +0 -36
  426. data/vendor/rb-sys/src/bindings.rs +0 -21
  427. data/vendor/rb-sys/src/hidden.rs +0 -11
  428. data/vendor/rb-sys/src/lib.rs +0 -34
  429. data/vendor/rb-sys/src/macros.rs +0 -371
  430. data/vendor/rb-sys/src/memory.rs +0 -53
  431. data/vendor/rb-sys/src/ruby_abi_version.rs +0 -38
  432. data/vendor/rb-sys/src/special_consts.rs +0 -31
  433. data/vendor/rb-sys/src/stable_api/compiled.c +0 -179
  434. data/vendor/rb-sys/src/stable_api/compiled.rs +0 -257
  435. data/vendor/rb-sys/src/stable_api/ruby_2_6.rs +0 -316
  436. data/vendor/rb-sys/src/stable_api/ruby_2_7.rs +0 -316
  437. data/vendor/rb-sys/src/stable_api/ruby_3_0.rs +0 -324
  438. data/vendor/rb-sys/src/stable_api/ruby_3_1.rs +0 -317
  439. data/vendor/rb-sys/src/stable_api/ruby_3_2.rs +0 -315
  440. data/vendor/rb-sys/src/stable_api/ruby_3_3.rs +0 -326
  441. data/vendor/rb-sys/src/stable_api/ruby_3_4.rs +0 -327
  442. data/vendor/rb-sys/src/stable_api.rs +0 -261
  443. data/vendor/rb-sys/src/symbol.rs +0 -31
  444. data/vendor/rb-sys/src/tracking_allocator.rs +0 -332
  445. data/vendor/rb-sys/src/utils.rs +0 -89
  446. data/vendor/rb-sys/src/value_type.rs +0 -7
@@ -1,776 +1,775 @@
1
- //! Comprehensive TDD test suite for RTF extraction.
2
- //!
3
- //! This test suite validates RTF extraction capabilities.
4
- //! Tests cover:
5
- //! - Accent and Unicode handling
6
- //! - Bookmarks and internal links
7
- //! - Footnotes and references
8
- //! - Text formatting (bold, italic, underline, strikeout, superscript, subscript, small caps)
9
- //! - Headings and structure
10
- //! - Image extraction
11
- //! - External hyperlinks
12
- //! - List extraction (simple and complex nested lists)
13
- //! - Table extraction (simple and complex with special formatting)
14
- //! - Unicode characters and special symbols
15
- //!
16
- //! Test Organization:
17
- //! - Basic Content Extraction (unicode, accent)
18
- //! - Structure Preservation (heading, list_simple, list_complex)
19
- //! - Table Extraction (table_simple, table_error_codes)
20
- //! - Formatting Detection (formatting)
21
- //! - Special Features (footnote, bookmark, link)
22
- //! - Integration Tests (deterministic extraction, no content loss)
23
- //!
24
- //! Success Criteria:
25
- //! - All tests passing (100%)
26
- //! - No content loss (should extract meaningful text from all files)
27
- //! - Deterministic extraction (same input = same output)
28
- //!
29
- //! Note: These tests require the `office` feature to be enabled.
30
-
31
- #![cfg(feature = "office")]
32
- #![allow(clippy::doc_suspicious_footnotes)]
33
-
34
- use kreuzberg::core::config::ExtractionConfig;
35
- use kreuzberg::core::extractor::extract_file;
36
- use std::path::PathBuf;
37
-
38
- mod helpers;
39
-
40
- /// Helper function to get path to RTF test document
41
- fn get_rtf_path(filename: &str) -> PathBuf {
42
- let manifest_dir = env!("CARGO_MANIFEST_DIR");
43
- PathBuf::from(manifest_dir)
44
- .parent()
45
- .expect("kreuzberg crate should have a parent")
46
- .parent()
47
- .expect("parent should have a parent")
48
- .join("test_documents")
49
- .join("rtf")
50
- .join(filename)
51
- }
52
-
53
- /// Helper for reaching the workspace root from the kreuzberg crate
54
- fn workspace_root() -> PathBuf {
55
- PathBuf::from(env!("CARGO_MANIFEST_DIR"))
56
- .parent()
57
- .expect("kreuzberg crate should have a parent")
58
- .parent()
59
- .expect("workspace root exists")
60
- .to_path_buf()
61
- }
62
-
63
- /// Test extraction of RTF file with accent characters (accented vowels).
64
- ///
65
- /// File: accent.rtf
66
- /// Content: "le café où on ne fume pas"
67
- /// Expected: Correctly extracts French text with accented characters (é, ù)
68
- /// Pandoc baseline: le café où on ne fume pas
69
- #[tokio::test]
70
- async fn test_rtf_accent_extraction() {
71
- let config = ExtractionConfig::default();
72
- let path = get_rtf_path("accent.rtf");
73
-
74
- let result = extract_file(&path, Some("application/rtf"), &config).await;
75
-
76
- assert!(result.is_ok(), "RTF extraction should succeed for accent.rtf");
77
- let extraction = result.unwrap();
78
-
79
- assert_eq!(extraction.mime_type, "application/rtf");
80
-
81
- assert!(!extraction.content.is_empty(), "Content should not be empty");
82
-
83
- let content = extraction.content.to_lowercase();
84
-
85
- assert!(
86
- extraction.content.contains("café") || content.contains("cafe"),
87
- "Should extract French word 'café' or 'cafe'"
88
- );
89
-
90
- assert!(
91
- extraction.content.contains("où") || content.contains("ou"),
92
- "Should extract French word 'où' or 'ou'"
93
- );
94
-
95
- assert!(
96
- content.contains("fume") || content.contains("smoking"),
97
- "Should extract content about smoking"
98
- );
99
- }
100
-
101
- /// Test extraction of RTF file with bookmarks (internal anchors/references).
102
- ///
103
- /// File: bookmark.rtf
104
- /// Content: Bookmark anchor labeled "Bookmark_1" and link text "click me"
105
- /// Expected: Extracts bookmark definition and link text
106
- /// Pandoc baseline: [Bookmark_1]{#bookmark_1} and [click me](#bookmark_1)
107
- #[tokio::test]
108
- async fn test_rtf_bookmark_extraction() {
109
- let config = ExtractionConfig::default();
110
- let path = get_rtf_path("bookmark.rtf");
111
-
112
- let result = extract_file(&path, Some("application/rtf"), &config).await;
113
-
114
- assert!(result.is_ok(), "RTF extraction should succeed for bookmark.rtf");
115
- let extraction = result.unwrap();
116
-
117
- let content = extraction.content.to_lowercase();
118
-
119
- assert!(
120
- content.contains("bookmark") || content.contains("click") || content.contains("me"),
121
- "Should extract bookmark or link text (found: {})",
122
- extraction.content
123
- );
124
- }
125
-
126
- /// Test extraction of RTF file with footnotes.
127
- ///
128
- /// File: footnote.rtf
129
- /// Content: Academic text with footnote references and their content
130
- /// Expected: Extracts both main text and footnote content
131
- /// Pandoc baseline: Uses [^1] and [^2] syntax for footnotes
132
- #[tokio::test]
133
- async fn test_rtf_footnote_extraction() {
134
- let config = ExtractionConfig::default();
135
- let path = get_rtf_path("footnote.rtf");
136
-
137
- let result = extract_file(&path, Some("application/rtf"), &config).await;
138
-
139
- assert!(result.is_ok(), "RTF extraction should succeed for footnote.rtf");
140
- let extraction = result.unwrap();
141
-
142
- assert!(!extraction.content.is_empty(), "Content should not be empty");
143
-
144
- let content = extraction.content.to_lowercase();
145
-
146
- assert!(
147
- content.contains("mead") || content.contains("landmark"),
148
- "Should extract main text about Mead's study"
149
- );
150
-
151
- assert!(
152
- content.contains("note")
153
- || content.contains("annotated")
154
- || content.contains("bibliography")
155
- || content.contains("sahlins"),
156
- "Should extract footnote content or references"
157
- );
158
-
159
- assert!(
160
- content.contains("footnote") || extraction.content.contains("[^") || content.contains("annotated"),
161
- "Should contain footnote indicators"
162
- );
163
- }
164
-
165
- /// Test extraction of RTF file with various text formatting.
166
- ///
167
- /// File: formatting.rtf
168
- /// Content: Text with bold, italic, underline, strikeout, superscript, subscript, small caps
169
- /// Expected: Preserves or indicates all formatting types
170
- /// Pandoc baseline: Detailed formatting in markdown syntax
171
- #[tokio::test]
172
- async fn test_rtf_formatting_extraction() {
173
- let config = ExtractionConfig::default();
174
- let path = get_rtf_path("formatting.rtf");
175
-
176
- let result = extract_file(&path, Some("application/rtf"), &config).await;
177
-
178
- assert!(result.is_ok(), "RTF extraction should succeed for formatting.rtf");
179
- let extraction = result.unwrap();
180
-
181
- assert!(!extraction.content.is_empty(), "Content should not be empty");
182
-
183
- let content = extraction.content.to_lowercase();
184
-
185
- assert!(
186
- content.contains("formatting") || content.contains("test") || content.contains("bold"),
187
- "Should extract formatting-related content"
188
- );
189
-
190
- assert!(
191
- extraction.content.contains("**bold**") || content.contains("bold"),
192
- "Should preserve or indicate bold text"
193
- );
194
-
195
- assert!(
196
- extraction.content.contains("*italic") || content.contains("italic"),
197
- "Should preserve or indicate italic text"
198
- );
199
-
200
- let has_formatting = extraction.content.contains("**")
201
- || extraction.content.contains("*")
202
- || extraction.content.contains("__")
203
- || extraction.content.contains("_")
204
- || extraction.content.contains("~~")
205
- || extraction.content.contains("^")
206
- || extraction.content.contains("~")
207
- || content.contains("bold");
208
-
209
- assert!(has_formatting, "Should preserve or indicate text formatting");
210
- }
211
-
212
- /// Test extraction of RTF file with heading hierarchy.
213
- ///
214
- /// File: heading.rtf
215
- /// Content: Three levels of headings (H1, H2, H3) followed by paragraph
216
- /// Expected: Extracts all headings and paragraph text
217
- /// Pandoc baseline: Markdown heading syntax (# ## ###)
218
- #[tokio::test]
219
- async fn test_rtf_heading_extraction() {
220
- let config = ExtractionConfig::default();
221
- let path = get_rtf_path("heading.rtf");
222
-
223
- let result = extract_file(&path, Some("application/rtf"), &config).await;
224
-
225
- assert!(result.is_ok(), "RTF extraction should succeed for heading.rtf");
226
- let extraction = result.unwrap();
227
-
228
- assert!(!extraction.content.is_empty(), "Content should not be empty");
229
-
230
- assert!(
231
- extraction.content.contains("Heading 1"),
232
- "Should extract Heading 1 text"
233
- );
234
-
235
- assert!(
236
- extraction.content.contains("Heading 2"),
237
- "Should extract Heading 2 text"
238
- );
239
-
240
- assert!(
241
- extraction.content.contains("Heading 3"),
242
- "Should extract Heading 3 text"
243
- );
244
-
245
- assert!(
246
- extraction.content.contains("Paragraph"),
247
- "Should extract paragraph text"
248
- );
249
-
250
- let content_lower = extraction.content.to_lowercase();
251
- assert!(
252
- extraction.content.contains("#")
253
- || (content_lower.contains("heading 1") && content_lower.contains("heading 2")),
254
- "Should preserve heading hierarchy"
255
- );
256
- }
257
-
258
- /// Test extraction of RTF file with embedded or referenced image.
259
- ///
260
- /// File: image.rtf
261
- /// Content: Image reference with dimensions (2.0in x 2.0in)
262
- /// Expected: Extracts image reference and/or dimensions
263
- /// Pandoc baseline: Markdown image syntax with dimensions
264
- #[tokio::test]
265
- async fn test_rtf_image_extraction() {
266
- let config = ExtractionConfig::default();
267
- let path = get_rtf_path("image.rtf");
268
-
269
- let result = extract_file(&path, Some("application/rtf"), &config).await;
270
-
271
- assert!(result.is_ok(), "RTF extraction should succeed for image.rtf");
272
- let extraction = result.unwrap();
273
-
274
- assert!(!extraction.content.is_empty(), "Content should not be empty");
275
-
276
- let content = extraction.content.to_lowercase();
277
-
278
- assert!(
279
- extraction.content.contains("!")
280
- || content.contains("image")
281
- || extraction.content.contains(".jpg")
282
- || content.contains("2.0")
283
- || content.contains("width")
284
- || content.contains("height"),
285
- "Should contain image reference or dimension information (found: {})",
286
- extraction.content
287
- );
288
- }
289
-
290
- /// Test extraction of RTF file with external hyperlink.
291
- ///
292
- /// File: link.rtf
293
- /// Content: Link to pandoc.org website
294
- /// Expected: Extracts link text and/or URL
295
- /// Pandoc baseline: Markdown link syntax [pandoc](http://pandoc.org)
296
- #[tokio::test]
297
- async fn test_rtf_link_extraction() {
298
- let config = ExtractionConfig::default();
299
- let path = get_rtf_path("link.rtf");
300
-
301
- let result = extract_file(&path, Some("application/rtf"), &config).await;
302
-
303
- assert!(result.is_ok(), "RTF extraction should succeed for link.rtf");
304
- let extraction = result.unwrap();
305
-
306
- assert!(!extraction.content.is_empty(), "Content should not be empty");
307
-
308
- let content = extraction.content.to_lowercase();
309
-
310
- assert!(
311
- content.contains("pandoc") || content.contains("http"),
312
- "Should extract link-related content (found: {})",
313
- extraction.content
314
- );
315
- }
316
-
317
- /// Test extraction of RTF file with complex nested list structure.
318
- ///
319
- /// File: list_complex.rtf
320
- /// Content: Multi-level nested list with various numbering (numeric, alphabetic, roman)
321
- /// Expected: Extracts all list items preserving or indicating hierarchy
322
- /// Pandoc baseline: Markdown nested list with mixed numbering schemes
323
- #[tokio::test]
324
- async fn test_rtf_list_complex_extraction() {
325
- let config = ExtractionConfig::default();
326
- let path = get_rtf_path("list_complex.rtf");
327
-
328
- let result = extract_file(&path, Some("application/rtf"), &config).await;
329
-
330
- assert!(result.is_ok(), "RTF extraction should succeed for list_complex.rtf");
331
- let extraction = result.unwrap();
332
-
333
- assert!(!extraction.content.is_empty(), "Content should not be empty");
334
-
335
- let content = extraction.content.to_lowercase();
336
-
337
- assert!(content.contains("one"), "Should extract list item 'One'");
338
-
339
- assert!(content.contains("two"), "Should extract list item 'Two'");
340
-
341
- assert!(
342
- content.contains("three") || content.contains("three"),
343
- "Should extract nested list item 'Three'"
344
- );
345
-
346
- assert!(
347
- content.contains("five") || content.contains("six"),
348
- "Should extract deeply nested list items"
349
- );
350
-
351
- assert!(
352
- extraction.content.contains("1")
353
- || extraction.content.contains("-")
354
- || extraction.content.contains("•")
355
- || content.contains("one"),
356
- "Should preserve list structure indicators"
357
- );
358
-
359
- assert!(
360
- content.contains("out of list") || content.contains("out"),
361
- "Should extract separator text 'Out of list'"
362
- );
363
-
364
- assert!(
365
- content.contains("seven") || content.contains("eight") || content.contains("7") || content.contains("8"),
366
- "Should extract restarted list numbering (7, 8)"
367
- );
368
- }
369
-
370
- /// Test extraction of RTF file with simple bulleted list.
371
- ///
372
- /// File: list_simple.rtf
373
- /// Content: Simple bullet list with one nested item and list break
374
- /// Expected: Extracts all list items and indicates nesting
375
- /// Pandoc baseline: Simple markdown bullet list with nesting
376
- #[tokio::test]
377
- async fn test_rtf_list_simple_extraction() {
378
- let config = ExtractionConfig::default();
379
- let path = get_rtf_path("list_simple.rtf");
380
-
381
- let result = extract_file(&path, Some("application/rtf"), &config).await;
382
-
383
- assert!(result.is_ok(), "RTF extraction should succeed for list_simple.rtf");
384
- let extraction = result.unwrap();
385
-
386
- assert!(!extraction.content.is_empty(), "Content should not be empty");
387
-
388
- let content = extraction.content.to_lowercase();
389
-
390
- assert!(content.contains("one"), "Should extract list item 'one'");
391
-
392
- assert!(content.contains("two"), "Should extract list item 'two'");
393
-
394
- assert!(content.contains("sub"), "Should extract nested list item 'sub'");
395
-
396
- assert!(content.contains("new"), "Should extract 'new list' text");
397
-
398
- assert!(
399
- extraction.content.contains("-") || extraction.content.contains("•") || extraction.content.contains("*"),
400
- "Should contain list markers"
401
- );
402
- }
403
-
404
- /// Test extraction of RTF file with table containing error codes.
405
- ///
406
- /// File: table_error_codes.rtf
407
- /// Content: Table with Code and Error columns, 23 rows of Pandoc error codes
408
- /// Expected: Extracts table structure and all data cells
409
- /// Pandoc baseline: Markdown table format with 2 columns and 23 rows
410
- ///
411
- /// Note: RTF table extraction via Pandoc markdown output may result in empty content
412
- /// due to limitations in Pandoc's markdown table rendering. Tables are present
413
- /// in Pandoc's internal JSON representation but may not render in text format.
414
- #[tokio::test]
415
- async fn test_rtf_table_error_codes_extraction() {
416
- let config = ExtractionConfig::default();
417
- let path = get_rtf_path("table_error_codes.rtf");
418
-
419
- let result = extract_file(&path, Some("application/rtf"), &config).await;
420
-
421
- assert!(
422
- result.is_ok(),
423
- "RTF extraction should succeed for table_error_codes.rtf"
424
- );
425
- let extraction = result.unwrap();
426
-
427
- assert!(
428
- extraction.mime_type == "application/rtf",
429
- "MIME type should be preserved"
430
- );
431
- }
432
-
433
- /// Test extraction of RTF file with simple 4-column, 2-row table.
434
- ///
435
- /// File: table_simple.rtf
436
- /// Content: Table with headers A, B, C, D and data row E, F, G, H
437
- /// Expected: Extracts all cells in correct table structure
438
- /// Pandoc baseline: Markdown table format
439
- ///
440
- /// Note: RTF table extraction via Pandoc markdown output may result in empty content
441
- /// due to limitations in Pandoc's markdown table rendering. Tables are present
442
- /// in Pandoc's internal JSON representation but may not render in text format.
443
- #[tokio::test]
444
- async fn test_rtf_table_simple_extraction() {
445
- let config = ExtractionConfig::default();
446
- let path = get_rtf_path("table_simple.rtf");
447
-
448
- let result = extract_file(&path, Some("application/rtf"), &config).await;
449
-
450
- assert!(result.is_ok(), "RTF extraction should succeed for table_simple.rtf");
451
- let extraction = result.unwrap();
452
-
453
- assert!(
454
- extraction.mime_type == "application/rtf",
455
- "MIME type should be preserved"
456
- );
457
- }
458
-
459
- /// Test extraction of RTF file with various Unicode characters.
460
- ///
461
- /// File: unicode.rtf
462
- /// Content: Smart quotes, Greek letters (α, ä)
463
- /// Expected: Correctly extracts and preserves Unicode characters
464
- /// Pandoc baseline: "hi"'hi'αä
465
- #[tokio::test]
466
- async fn test_rtf_unicode_extraction() {
467
- let config = ExtractionConfig::default();
468
- let path = get_rtf_path("unicode.rtf");
469
-
470
- let result = extract_file(&path, Some("application/rtf"), &config).await;
471
-
472
- assert!(result.is_ok(), "RTF extraction should succeed for unicode.rtf");
473
- let extraction = result.unwrap();
474
-
475
- assert!(!extraction.content.is_empty(), "Content should not be empty");
476
-
477
- assert!(
478
- extraction.content.contains("hi") || extraction.content.contains("α") || extraction.content.contains("ä"),
479
- "Should extract unicode content (found: {})",
480
- extraction.content
481
- );
482
- }
483
-
484
- /// Test that RTF extraction is deterministic
485
- /// Same input should produce identical output
486
- #[tokio::test]
487
- async fn test_rtf_extraction_deterministic_unicode() {
488
- let config = ExtractionConfig::default();
489
- let path = get_rtf_path("unicode.rtf");
490
-
491
- let result1 = extract_file(&path, Some("application/rtf"), &config).await;
492
- let result2 = extract_file(&path, Some("application/rtf"), &config).await;
493
-
494
- assert!(result1.is_ok() && result2.is_ok(), "Both extractions should succeed");
495
-
496
- let extraction1 = result1.unwrap();
497
- let extraction2 = result2.unwrap();
498
-
499
- assert_eq!(
500
- extraction1.content, extraction2.content,
501
- "FAIL: Extraction is not deterministic. Same input produced different outputs."
502
- );
503
- }
504
-
505
- /// Test that RTF extraction is deterministic for complex files
506
- /// Same input should produce identical output
507
- #[tokio::test]
508
- async fn test_rtf_extraction_deterministic_list_complex() {
509
- let config = ExtractionConfig::default();
510
- let path = get_rtf_path("list_complex.rtf");
511
-
512
- let result1 = extract_file(&path, Some("application/rtf"), &config).await;
513
- let result2 = extract_file(&path, Some("application/rtf"), &config).await;
514
-
515
- assert!(result1.is_ok() && result2.is_ok(), "Both extractions should succeed");
516
-
517
- let extraction1 = result1.unwrap();
518
- let extraction2 = result2.unwrap();
519
-
520
- assert_eq!(
521
- extraction1.content, extraction2.content,
522
- "FAIL: Extraction is not deterministic. Same input produced different outputs."
523
- );
524
- }
525
-
526
- /// Test no critical content loss
527
- /// All RTF files should extract non-empty content (except possibly image-only files)
528
- #[tokio::test]
529
- async fn test_rtf_no_critical_content_loss() {
530
- let config = ExtractionConfig::default();
531
-
532
- let must_extract = vec![
533
- "unicode.rtf",
534
- "accent.rtf",
535
- "heading.rtf",
536
- "list_simple.rtf",
537
- "list_complex.rtf",
538
- "formatting.rtf",
539
- "footnote.rtf",
540
- "bookmark.rtf",
541
- "link.rtf",
542
- ];
543
-
544
- for filename in must_extract {
545
- let path = get_rtf_path(filename);
546
- let result = extract_file(&path, Some("application/rtf"), &config).await;
547
-
548
- assert!(
549
- result.is_ok(),
550
- "FAIL: Extraction failed for {} (critical file)",
551
- filename
552
- );
553
-
554
- let extraction = result.unwrap();
555
- assert!(
556
- !extraction.content.is_empty(),
557
- "FAIL: CRITICAL - Extracted 0 bytes from {}. RTF extractor lost all content.",
558
- filename
559
- );
560
-
561
- assert!(
562
- extraction.content.len() >= 5,
563
- "FAIL: Extracted only {} bytes from {} (expected at least 5 characters). Content: '{}'",
564
- extraction.content.len(),
565
- filename,
566
- extraction.content
567
- );
568
- }
569
- }
570
-
571
- /// Test MIME type preservation
572
- /// All RTF extractions should preserve the application/rtf MIME type
573
- #[tokio::test]
574
- async fn test_rtf_mime_type_preservation() {
575
- let config = ExtractionConfig::default();
576
-
577
- let test_files = vec!["unicode.rtf", "accent.rtf", "heading.rtf", "list_simple.rtf"];
578
-
579
- for filename in test_files {
580
- let path = get_rtf_path(filename);
581
- let result = extract_file(&path, Some("application/rtf"), &config).await;
582
-
583
- assert!(result.is_ok(), "Extraction should succeed for {}", filename);
584
-
585
- let extraction = result.unwrap();
586
- assert_eq!(
587
- extraction.mime_type, "application/rtf",
588
- "FAIL: MIME type not preserved for {}",
589
- filename
590
- );
591
- }
592
- }
593
-
594
- /// Parity check: RTF extracted from the DOCX `word_sample.docx` should
595
- /// carry the same content signals and metadata as the DOCX extractor.
596
- #[tokio::test]
597
- async fn test_rtf_word_sample_matches_docx_metadata_and_content() {
598
- let root = workspace_root();
599
- let rtf_path = root.join("test_documents/rtf/word_sample.rtf");
600
- let docx_path = root.join("test_documents/documents/word_sample.docx");
601
-
602
- if !rtf_path.exists() || !docx_path.exists() {
603
- println!("Skipping word_sample parity test: fixtures missing");
604
- return;
605
- }
606
-
607
- let config = ExtractionConfig::default();
608
- let rtf_result = extract_file(&rtf_path, Some("application/rtf"), &config)
609
- .await
610
- .expect("RTF extraction should succeed for word_sample");
611
- let docx_result = extract_file(&docx_path, None, &config)
612
- .await
613
- .expect("DOCX extraction should succeed for word_sample");
614
-
615
- let rtf_content_lower = rtf_result.content.to_lowercase();
616
- assert!(
617
- rtf_content_lower.contains("swim"),
618
- "RTF content should include the same body text as DOCX"
619
- );
620
-
621
- for key in ["created_by", "modified_by", "created_at", "revision"] {
622
- assert_eq!(
623
- rtf_result.metadata.additional.get(key).and_then(|v| v.as_str()),
624
- docx_result.metadata.additional.get(key).and_then(|v| v.as_str()),
625
- "Metadata field {} should align with DOCX",
626
- key
627
- );
628
- }
629
-
630
- for (key, expected) in [
631
- ("page_count", 2),
632
- ("word_count", 108),
633
- ("character_count", 620),
634
- ("line_count", 5),
635
- ("paragraph_count", 1),
636
- ] {
637
- assert_eq!(
638
- rtf_result.metadata.additional.get(key).and_then(|v| v.as_i64()),
639
- Some(expected),
640
- "Metadata field {} should match DOCX values",
641
- key
642
- );
643
- }
644
- }
645
-
646
- /// RTF generated from lorem_ipsum.docx should expose the same document statistics
647
- /// we validate for the DOCX extractor.
648
- #[tokio::test]
649
- async fn test_rtf_lorem_ipsum_metadata_alignment() {
650
- let root = workspace_root();
651
- let rtf_path = root.join("test_documents/rtf/lorem_ipsum.rtf");
652
-
653
- if !rtf_path.exists() {
654
- println!("Skipping lorem_ipsum metadata test: fixture missing");
655
- return;
656
- }
657
-
658
- let config = ExtractionConfig::default();
659
- let result = extract_file(&rtf_path, Some("application/rtf"), &config)
660
- .await
661
- .expect("RTF extraction should succeed for lorem_ipsum");
662
-
663
- assert!(
664
- result.content.to_lowercase().contains("lorem ipsum"),
665
- "Content should contain lorem ipsum text"
666
- );
667
-
668
- for (key, expected) in [
669
- ("page_count", 1),
670
- ("word_count", 520),
671
- ("character_count", 2967),
672
- ("line_count", 24),
673
- ("paragraph_count", 6),
674
- ] {
675
- assert_eq!(
676
- result.metadata.additional.get(key).and_then(|v| v.as_i64()),
677
- Some(expected),
678
- "Metadata field {} should match DOCX values",
679
- key
680
- );
681
- }
682
- }
683
-
684
- /// The comprehensive extraction fixture should mirror the coverage of the ODT/DOCX variants:
685
- /// headings, section text, table content, and metadata fields should all be present.
686
- #[tokio::test]
687
- async fn test_rtf_comprehensive_extraction_alignment() {
688
- let root = workspace_root();
689
- let rtf_path = root.join("test_documents/rtf/extraction_test.rtf");
690
- let docx_path = root.join("test_documents/extraction_test.docx");
691
- let odt_path = root.join("test_documents/extraction_test.odt");
692
-
693
- if !rtf_path.exists() {
694
- println!("⚠️ Test document not found at {:?}, skipping", rtf_path);
695
- return;
696
- }
697
- if !docx_path.exists() || !odt_path.exists() {
698
- println!(
699
- "⚠️ Companion DOCX/ODT documents missing (docx: {}, odt: {}), skipping",
700
- docx_path.exists(),
701
- odt_path.exists()
702
- );
703
- return;
704
- }
705
-
706
- let config = ExtractionConfig::default();
707
- let rtf_result = extract_file(&rtf_path, Some("application/rtf"), &config)
708
- .await
709
- .expect("RTF extraction should succeed for extraction_test.rtf");
710
- let docx_result = extract_file(&docx_path, None, &config)
711
- .await
712
- .expect("DOCX extraction should succeed for extraction_test.docx");
713
- let odt_result = extract_file(&odt_path, None, &config)
714
- .await
715
- .expect("ODT extraction should succeed for extraction_test.odt");
716
-
717
- assert!(
718
- rtf_result.content.contains("Comprehensive Extraction Test Document"),
719
- "Should include document heading"
720
- );
721
- assert!(
722
- rtf_result.content.contains("First Section"),
723
- "Should include first section heading"
724
- );
725
- assert!(
726
- rtf_result.content.contains("Second Section"),
727
- "Should include second section heading"
728
- );
729
- assert!(
730
- rtf_result.content.contains("Third Section"),
731
- "Should include third section heading"
732
- );
733
-
734
- // Table/text alignment with DOCX/ODT variants
735
- for expected in ["Header 1", "Cell 1A", "Product", "Apple"] {
736
- assert!(
737
- rtf_result.content.contains(expected),
738
- "Should include table content '{}'",
739
- expected
740
- );
741
- }
742
- assert!(
743
- rtf_result.content.contains("|"),
744
- "Should preserve table structure markers"
745
- );
746
- assert!(
747
- !rtf_result.tables.is_empty(),
748
- "Should extract structured tables from RTF"
749
- );
750
- assert!(
751
- rtf_result
752
- .tables
753
- .iter()
754
- .any(|t| t.markdown.contains("Header 1") || t.markdown.contains("Cell 1A")),
755
- "Table markdown should include header/data cells"
756
- );
757
- assert!(
758
- rtf_result.tables.len() >= docx_result.tables.len() && rtf_result.tables.len() >= odt_result.tables.len(),
759
- "RTF should capture at least as many tables as DOCX/ODT"
760
- );
761
-
762
- for (key, expected) in [
763
- ("page_count", 1),
764
- ("word_count", 83),
765
- ("character_count", 475),
766
- ("line_count", 12),
767
- ("paragraph_count", 8),
768
- ] {
769
- assert_eq!(
770
- rtf_result.metadata.additional.get(key).and_then(|v| v.as_i64()),
771
- Some(expected),
772
- "Metadata field {} should be populated",
773
- key
774
- );
775
- }
776
- }
1
+ //! Comprehensive TDD test suite for RTF extraction.
2
+ //!
3
+ //! This test suite validates RTF extraction capabilities.
4
+ //! Tests cover:
5
+ //! - Accent and Unicode handling
6
+ //! - Bookmarks and internal links
7
+ //! - Footnotes and references
8
+ //! - Text formatting (bold, italic, underline, strikeout, superscript, subscript, small caps)
9
+ //! - Headings and structure
10
+ //! - Image extraction
11
+ //! - External hyperlinks
12
+ //! - List extraction (simple and complex nested lists)
13
+ //! - Table extraction (simple and complex with special formatting)
14
+ //! - Unicode characters and special symbols
15
+ //!
16
+ //! Test Organization:
17
+ //! - Basic Content Extraction (unicode, accent)
18
+ //! - Structure Preservation (heading, list_simple, list_complex)
19
+ //! - Table Extraction (table_simple, table_error_codes)
20
+ //! - Formatting Detection (formatting)
21
+ //! - Special Features (footnote, bookmark, link)
22
+ //! - Integration Tests (deterministic extraction, no content loss)
23
+ //!
24
+ //! Success Criteria:
25
+ //! - All tests passing (100%)
26
+ //! - No content loss (should extract meaningful text from all files)
27
+ //! - Deterministic extraction (same input = same output)
28
+ //!
29
+ //! Note: These tests require the `office` feature to be enabled.
30
+
31
+ #![cfg(feature = "office")]
32
+ #![allow(clippy::doc_suspicious_footnotes)]
33
+
34
+ use kreuzberg::core::config::ExtractionConfig;
35
+ use kreuzberg::core::extractor::extract_file;
36
+ use std::path::PathBuf;
37
+
38
+ mod helpers;
39
+
40
+ /// Helper function to get path to RTF test document
41
+ fn get_rtf_path(filename: &str) -> PathBuf {
42
+ let manifest_dir = env!("CARGO_MANIFEST_DIR");
43
+ PathBuf::from(manifest_dir)
44
+ .parent()
45
+ .expect("kreuzberg crate should have a parent")
46
+ .parent()
47
+ .expect("parent should have a parent")
48
+ .join("test_documents")
49
+ .join("rtf")
50
+ .join(filename)
51
+ }
52
+
53
+ /// Helper for reaching the workspace root from the kreuzberg crate
54
+ fn workspace_root() -> PathBuf {
55
+ PathBuf::from(env!("CARGO_MANIFEST_DIR"))
56
+ .parent()
57
+ .expect("kreuzberg crate should have a parent")
58
+ .parent()
59
+ .expect("workspace root exists")
60
+ .to_path_buf()
61
+ }
62
+
63
+ /// Test extraction of RTF file with accent characters (accented vowels).
64
+ ///
65
+ /// File: accent.rtf
66
+ /// Content: "le café où on ne fume pas"
67
+ /// Expected: Correctly extracts French text with accented characters (é, ù)
68
+ /// Pandoc baseline: le café où on ne fume pas
69
+ #[tokio::test]
70
+ async fn test_rtf_accent_extraction() {
71
+ let config = ExtractionConfig::default();
72
+ let path = get_rtf_path("accent.rtf");
73
+
74
+ let result = extract_file(&path, Some("application/rtf"), &config).await;
75
+
76
+ assert!(result.is_ok(), "RTF extraction should succeed for accent.rtf");
77
+ let extraction = result.unwrap();
78
+
79
+ assert_eq!(extraction.mime_type, "application/rtf");
80
+
81
+ assert!(!extraction.content.is_empty(), "Content should not be empty");
82
+
83
+ let content = extraction.content.to_lowercase();
84
+
85
+ assert!(
86
+ extraction.content.contains("café") || content.contains("cafe"),
87
+ "Should extract French word 'café' or 'cafe'"
88
+ );
89
+
90
+ assert!(
91
+ extraction.content.contains("où") || content.contains("ou"),
92
+ "Should extract French word 'où' or 'ou'"
93
+ );
94
+
95
+ assert!(
96
+ content.contains("fume") || content.contains("smoking"),
97
+ "Should extract content about smoking"
98
+ );
99
+ }
100
+
101
+ /// Test extraction of RTF file with bookmarks (internal anchors/references).
102
+ ///
103
+ /// File: bookmark.rtf
104
+ /// Content: Bookmark anchor labeled "Bookmark_1" and link text "click me"
105
+ /// Expected: Extracts bookmark definition and link text
106
+ /// Pandoc baseline: [Bookmark_1]{#bookmark_1} and [click me](#bookmark_1)
107
+ #[tokio::test]
108
+ async fn test_rtf_bookmark_extraction() {
109
+ let config = ExtractionConfig::default();
110
+ let path = get_rtf_path("bookmark.rtf");
111
+
112
+ let result = extract_file(&path, Some("application/rtf"), &config).await;
113
+
114
+ assert!(result.is_ok(), "RTF extraction should succeed for bookmark.rtf");
115
+ let extraction = result.unwrap();
116
+
117
+ let content = extraction.content.to_lowercase();
118
+
119
+ assert!(
120
+ content.contains("bookmark") || content.contains("click") || content.contains("me"),
121
+ "Should extract bookmark or link text (found: {})",
122
+ extraction.content
123
+ );
124
+ }
125
+
126
+ /// Test extraction of RTF file with footnotes.
127
+ ///
128
+ /// File: footnote.rtf
129
+ /// Content: Academic text with footnote references and their content
130
+ /// Expected: Extracts both main text and footnote content
131
+ /// Pandoc baseline: Uses [^1] and [^2] syntax for footnotes
132
+ #[tokio::test]
133
+ async fn test_rtf_footnote_extraction() {
134
+ let config = ExtractionConfig::default();
135
+ let path = get_rtf_path("footnote.rtf");
136
+
137
+ let result = extract_file(&path, Some("application/rtf"), &config).await;
138
+
139
+ assert!(result.is_ok(), "RTF extraction should succeed for footnote.rtf");
140
+ let extraction = result.unwrap();
141
+
142
+ assert!(!extraction.content.is_empty(), "Content should not be empty");
143
+
144
+ let content = extraction.content.to_lowercase();
145
+
146
+ assert!(
147
+ content.contains("mead") || content.contains("landmark"),
148
+ "Should extract main text about Mead's study"
149
+ );
150
+
151
+ assert!(
152
+ content.contains("note")
153
+ || content.contains("annotated")
154
+ || content.contains("bibliography")
155
+ || content.contains("sahlins"),
156
+ "Should extract footnote content or references"
157
+ );
158
+
159
+ assert!(
160
+ content.contains("footnote") || extraction.content.contains("[^") || content.contains("annotated"),
161
+ "Should contain footnote indicators"
162
+ );
163
+ }
164
+
165
+ /// Test extraction of RTF file with various text formatting.
166
+ ///
167
+ /// File: formatting.rtf
168
+ /// Content: Text with bold, italic, underline, strikeout, superscript, subscript, small caps
169
+ /// Expected: Preserves or indicates all formatting types
170
+ /// Pandoc baseline: Detailed formatting in markdown syntax
171
+ #[tokio::test]
172
+ async fn test_rtf_formatting_extraction() {
173
+ let config = ExtractionConfig::default();
174
+ let path = get_rtf_path("formatting.rtf");
175
+
176
+ let result = extract_file(&path, Some("application/rtf"), &config).await;
177
+
178
+ assert!(result.is_ok(), "RTF extraction should succeed for formatting.rtf");
179
+ let extraction = result.unwrap();
180
+
181
+ assert!(!extraction.content.is_empty(), "Content should not be empty");
182
+
183
+ let content = extraction.content.to_lowercase();
184
+
185
+ assert!(
186
+ content.contains("formatting") || content.contains("test") || content.contains("bold"),
187
+ "Should extract formatting-related content"
188
+ );
189
+
190
+ assert!(
191
+ extraction.content.contains("**bold**") || content.contains("bold"),
192
+ "Should preserve or indicate bold text"
193
+ );
194
+
195
+ assert!(
196
+ extraction.content.contains("*italic") || content.contains("italic"),
197
+ "Should preserve or indicate italic text"
198
+ );
199
+
200
+ let has_formatting = extraction.content.contains("**")
201
+ || extraction.content.contains("*")
202
+ || extraction.content.contains("__")
203
+ || extraction.content.contains("_")
204
+ || extraction.content.contains("~~")
205
+ || extraction.content.contains("^")
206
+ || extraction.content.contains("~")
207
+ || content.contains("bold");
208
+
209
+ assert!(has_formatting, "Should preserve or indicate text formatting");
210
+ }
211
+
212
+ /// Test extraction of RTF file with heading hierarchy.
213
+ ///
214
+ /// File: heading.rtf
215
+ /// Content: Three levels of headings (H1, H2, H3) followed by paragraph
216
+ /// Expected: Extracts all headings and paragraph text
217
+ /// Pandoc baseline: Markdown heading syntax (# ## ###)
218
+ #[tokio::test]
219
+ async fn test_rtf_heading_extraction() {
220
+ let config = ExtractionConfig::default();
221
+ let path = get_rtf_path("heading.rtf");
222
+
223
+ let result = extract_file(&path, Some("application/rtf"), &config).await;
224
+
225
+ assert!(result.is_ok(), "RTF extraction should succeed for heading.rtf");
226
+ let extraction = result.unwrap();
227
+
228
+ assert!(!extraction.content.is_empty(), "Content should not be empty");
229
+
230
+ assert!(
231
+ extraction.content.contains("Heading 1"),
232
+ "Should extract Heading 1 text"
233
+ );
234
+
235
+ assert!(
236
+ extraction.content.contains("Heading 2"),
237
+ "Should extract Heading 2 text"
238
+ );
239
+
240
+ assert!(
241
+ extraction.content.contains("Heading 3"),
242
+ "Should extract Heading 3 text"
243
+ );
244
+
245
+ assert!(
246
+ extraction.content.contains("Paragraph"),
247
+ "Should extract paragraph text"
248
+ );
249
+
250
+ let content_lower = extraction.content.to_lowercase();
251
+ assert!(
252
+ extraction.content.contains("#")
253
+ || (content_lower.contains("heading 1") && content_lower.contains("heading 2")),
254
+ "Should preserve heading hierarchy"
255
+ );
256
+ }
257
+
258
+ /// Test extraction of RTF file with embedded or referenced image.
259
+ ///
260
+ /// File: image.rtf
261
+ /// Content: Image reference with dimensions (2.0in x 2.0in)
262
+ /// Expected: Extracts image reference and/or dimensions
263
+ /// Pandoc baseline: Markdown image syntax with dimensions
264
+ #[tokio::test]
265
+ async fn test_rtf_image_extraction() {
266
+ let config = ExtractionConfig::default();
267
+ let path = get_rtf_path("image.rtf");
268
+
269
+ let result = extract_file(&path, Some("application/rtf"), &config).await;
270
+
271
+ assert!(result.is_ok(), "RTF extraction should succeed for image.rtf");
272
+ let extraction = result.unwrap();
273
+
274
+ assert!(!extraction.content.is_empty(), "Content should not be empty");
275
+
276
+ let content = extraction.content.to_lowercase();
277
+
278
+ assert!(
279
+ extraction.content.contains("!")
280
+ || content.contains("image")
281
+ || extraction.content.contains(".jpg")
282
+ || content.contains("2.0")
283
+ || content.contains("width")
284
+ || content.contains("height"),
285
+ "Should contain image reference or dimension information (found: {})",
286
+ extraction.content
287
+ );
288
+ }
289
+
290
+ /// Test extraction of RTF file with external hyperlink.
291
+ ///
292
+ /// File: link.rtf
293
+ /// Content: Link to pandoc.org website
294
+ /// Expected: Extracts link text and/or URL
295
+ /// Pandoc baseline: Markdown link syntax [pandoc](http://pandoc.org)
296
+ #[tokio::test]
297
+ async fn test_rtf_link_extraction() {
298
+ let config = ExtractionConfig::default();
299
+ let path = get_rtf_path("link.rtf");
300
+
301
+ let result = extract_file(&path, Some("application/rtf"), &config).await;
302
+
303
+ assert!(result.is_ok(), "RTF extraction should succeed for link.rtf");
304
+ let extraction = result.unwrap();
305
+
306
+ assert!(!extraction.content.is_empty(), "Content should not be empty");
307
+
308
+ let content = extraction.content.to_lowercase();
309
+
310
+ assert!(
311
+ content.contains("pandoc") || content.contains("http"),
312
+ "Should extract link-related content (found: {})",
313
+ extraction.content
314
+ );
315
+ }
316
+
317
+ /// Test extraction of RTF file with complex nested list structure.
318
+ ///
319
+ /// File: list_complex.rtf
320
+ /// Content: Multi-level nested list with various numbering (numeric, alphabetic, roman)
321
+ /// Expected: Extracts all list items preserving or indicating hierarchy
322
+ /// Pandoc baseline: Markdown nested list with mixed numbering schemes
323
+ #[tokio::test]
324
+ async fn test_rtf_list_complex_extraction() {
325
+ let config = ExtractionConfig::default();
326
+ let path = get_rtf_path("list_complex.rtf");
327
+
328
+ let result = extract_file(&path, Some("application/rtf"), &config).await;
329
+
330
+ assert!(result.is_ok(), "RTF extraction should succeed for list_complex.rtf");
331
+ let extraction = result.unwrap();
332
+
333
+ assert!(!extraction.content.is_empty(), "Content should not be empty");
334
+
335
+ let content = extraction.content.to_lowercase();
336
+
337
+ assert!(content.contains("one"), "Should extract list item 'One'");
338
+
339
+ assert!(content.contains("two"), "Should extract list item 'Two'");
340
+
341
+ assert!(
342
+ content.contains("three") || content.contains("three"),
343
+ "Should extract nested list item 'Three'"
344
+ );
345
+
346
+ assert!(
347
+ content.contains("five") || content.contains("six"),
348
+ "Should extract deeply nested list items"
349
+ );
350
+
351
+ assert!(
352
+ extraction.content.contains("1")
353
+ || extraction.content.contains("-")
354
+ || extraction.content.contains("•")
355
+ || content.contains("one"),
356
+ "Should preserve list structure indicators"
357
+ );
358
+
359
+ assert!(
360
+ content.contains("out of list") || content.contains("out"),
361
+ "Should extract separator text 'Out of list'"
362
+ );
363
+
364
+ assert!(
365
+ content.contains("seven") || content.contains("eight") || content.contains("7") || content.contains("8"),
366
+ "Should extract restarted list numbering (7, 8)"
367
+ );
368
+ }
369
+
370
+ /// Test extraction of RTF file with simple bulleted list.
371
+ ///
372
+ /// File: list_simple.rtf
373
+ /// Content: Simple bullet list with one nested item and list break
374
+ /// Expected: Extracts all list items and indicates nesting
375
+ /// Pandoc baseline: Simple markdown bullet list with nesting
376
+ #[tokio::test]
377
+ async fn test_rtf_list_simple_extraction() {
378
+ let config = ExtractionConfig::default();
379
+ let path = get_rtf_path("list_simple.rtf");
380
+
381
+ let result = extract_file(&path, Some("application/rtf"), &config).await;
382
+
383
+ assert!(result.is_ok(), "RTF extraction should succeed for list_simple.rtf");
384
+ let extraction = result.unwrap();
385
+
386
+ assert!(!extraction.content.is_empty(), "Content should not be empty");
387
+
388
+ let content = extraction.content.to_lowercase();
389
+
390
+ assert!(content.contains("one"), "Should extract list item 'one'");
391
+
392
+ assert!(content.contains("two"), "Should extract list item 'two'");
393
+
394
+ assert!(content.contains("sub"), "Should extract nested list item 'sub'");
395
+
396
+ assert!(content.contains("new"), "Should extract 'new list' text");
397
+
398
+ assert!(
399
+ extraction.content.contains("-") || extraction.content.contains("•") || extraction.content.contains("*"),
400
+ "Should contain list markers"
401
+ );
402
+ }
403
+
404
+ /// Test extraction of RTF file with table containing error codes.
405
+ ///
406
+ /// File: table_error_codes.rtf
407
+ /// Content: Table with Code and Error columns, 23 rows of Pandoc error codes
408
+ /// Expected: Extracts table structure and all data cells
409
+ /// Pandoc baseline: Markdown table format with 2 columns and 23 rows
410
+ ///
411
+ /// Note: RTF table extraction via Pandoc markdown output may result in empty content
412
+ /// due to limitations in Pandoc's markdown table rendering. Tables are present
413
+ /// in Pandoc's internal JSON representation but may not render in text format.
414
+ #[tokio::test]
415
+ async fn test_rtf_table_error_codes_extraction() {
416
+ let config = ExtractionConfig::default();
417
+ let path = get_rtf_path("table_error_codes.rtf");
418
+
419
+ let result = extract_file(&path, Some("application/rtf"), &config).await;
420
+
421
+ assert!(
422
+ result.is_ok(),
423
+ "RTF extraction should succeed for table_error_codes.rtf"
424
+ );
425
+ let extraction = result.unwrap();
426
+
427
+ assert!(
428
+ extraction.mime_type == "application/rtf",
429
+ "MIME type should be preserved"
430
+ );
431
+ }
432
+
433
+ /// Test extraction of RTF file with simple 4-column, 2-row table.
434
+ ///
435
+ /// File: table_simple.rtf
436
+ /// Content: Table with headers A, B, C, D and data row E, F, G, H
437
+ /// Expected: Extracts all cells in correct table structure
438
+ /// Pandoc baseline: Markdown table format
439
+ ///
440
+ /// Note: RTF table extraction via Pandoc markdown output may result in empty content
441
+ /// due to limitations in Pandoc's markdown table rendering. Tables are present
442
+ /// in Pandoc's internal JSON representation but may not render in text format.
443
+ #[tokio::test]
444
+ async fn test_rtf_table_simple_extraction() {
445
+ let config = ExtractionConfig::default();
446
+ let path = get_rtf_path("table_simple.rtf");
447
+
448
+ let result = extract_file(&path, Some("application/rtf"), &config).await;
449
+
450
+ assert!(result.is_ok(), "RTF extraction should succeed for table_simple.rtf");
451
+ let extraction = result.unwrap();
452
+
453
+ assert!(
454
+ extraction.mime_type == "application/rtf",
455
+ "MIME type should be preserved"
456
+ );
457
+ }
458
+
459
+ /// Test extraction of RTF file with various Unicode characters.
460
+ ///
461
+ /// File: unicode.rtf
462
+ /// Content: Smart quotes, Greek letters (α, ä)
463
+ /// Expected: Correctly extracts and preserves Unicode characters
464
+ /// Pandoc baseline: "hi"'hi'αä
465
+ #[tokio::test]
466
+ async fn test_rtf_unicode_extraction() {
467
+ let config = ExtractionConfig::default();
468
+ let path = get_rtf_path("unicode.rtf");
469
+
470
+ let result = extract_file(&path, Some("application/rtf"), &config).await;
471
+
472
+ assert!(result.is_ok(), "RTF extraction should succeed for unicode.rtf");
473
+ let extraction = result.unwrap();
474
+
475
+ assert!(!extraction.content.is_empty(), "Content should not be empty");
476
+
477
+ assert!(
478
+ extraction.content.contains("hi") || extraction.content.contains("α") || extraction.content.contains("ä"),
479
+ "Should extract unicode content (found: {})",
480
+ extraction.content
481
+ );
482
+ }
483
+
484
+ /// Test that RTF extraction is deterministic
485
+ /// Same input should produce identical output
486
+ #[tokio::test]
487
+ async fn test_rtf_extraction_deterministic_unicode() {
488
+ let config = ExtractionConfig::default();
489
+ let path = get_rtf_path("unicode.rtf");
490
+
491
+ let result1 = extract_file(&path, Some("application/rtf"), &config).await;
492
+ let result2 = extract_file(&path, Some("application/rtf"), &config).await;
493
+
494
+ assert!(result1.is_ok() && result2.is_ok(), "Both extractions should succeed");
495
+
496
+ let extraction1 = result1.unwrap();
497
+ let extraction2 = result2.unwrap();
498
+
499
+ assert_eq!(
500
+ extraction1.content, extraction2.content,
501
+ "FAIL: Extraction is not deterministic. Same input produced different outputs."
502
+ );
503
+ }
504
+
505
+ /// Test that RTF extraction is deterministic for complex files
506
+ /// Same input should produce identical output
507
+ #[tokio::test]
508
+ async fn test_rtf_extraction_deterministic_list_complex() {
509
+ let config = ExtractionConfig::default();
510
+ let path = get_rtf_path("list_complex.rtf");
511
+
512
+ let result1 = extract_file(&path, Some("application/rtf"), &config).await;
513
+ let result2 = extract_file(&path, Some("application/rtf"), &config).await;
514
+
515
+ assert!(result1.is_ok() && result2.is_ok(), "Both extractions should succeed");
516
+
517
+ let extraction1 = result1.unwrap();
518
+ let extraction2 = result2.unwrap();
519
+
520
+ assert_eq!(
521
+ extraction1.content, extraction2.content,
522
+ "FAIL: Extraction is not deterministic. Same input produced different outputs."
523
+ );
524
+ }
525
+
526
+ /// Test no critical content loss
527
+ /// All RTF files should extract non-empty content (except possibly image-only files)
528
+ #[tokio::test]
529
+ async fn test_rtf_no_critical_content_loss() {
530
+ let config = ExtractionConfig::default();
531
+
532
+ let must_extract = vec![
533
+ "unicode.rtf",
534
+ "accent.rtf",
535
+ "heading.rtf",
536
+ "list_simple.rtf",
537
+ "list_complex.rtf",
538
+ "formatting.rtf",
539
+ "footnote.rtf",
540
+ "bookmark.rtf",
541
+ "link.rtf",
542
+ ];
543
+
544
+ for filename in must_extract {
545
+ let path = get_rtf_path(filename);
546
+ let result = extract_file(&path, Some("application/rtf"), &config).await;
547
+
548
+ assert!(
549
+ result.is_ok(),
550
+ "FAIL: Extraction failed for {} (critical file)",
551
+ filename
552
+ );
553
+
554
+ let extraction = result.unwrap();
555
+ assert!(
556
+ !extraction.content.is_empty(),
557
+ "FAIL: CRITICAL - Extracted 0 bytes from {}. RTF extractor lost all content.",
558
+ filename
559
+ );
560
+
561
+ assert!(
562
+ extraction.content.len() >= 5,
563
+ "FAIL: Extracted only {} bytes from {} (expected at least 5 characters). Content: '{}'",
564
+ extraction.content.len(),
565
+ filename,
566
+ extraction.content
567
+ );
568
+ }
569
+ }
570
+
571
+ /// Test MIME type preservation
572
+ /// All RTF extractions should preserve the application/rtf MIME type
573
+ #[tokio::test]
574
+ async fn test_rtf_mime_type_preservation() {
575
+ let config = ExtractionConfig::default();
576
+
577
+ let test_files = vec!["unicode.rtf", "accent.rtf", "heading.rtf", "list_simple.rtf"];
578
+
579
+ for filename in test_files {
580
+ let path = get_rtf_path(filename);
581
+ let result = extract_file(&path, Some("application/rtf"), &config).await;
582
+
583
+ assert!(result.is_ok(), "Extraction should succeed for {}", filename);
584
+
585
+ let extraction = result.unwrap();
586
+ assert_eq!(
587
+ extraction.mime_type, "application/rtf",
588
+ "FAIL: MIME type not preserved for {}",
589
+ filename
590
+ );
591
+ }
592
+ }
593
+
594
+ /// Parity check: RTF extracted from the DOCX `word_sample.docx` should
595
+ /// carry the same content signals and metadata as the DOCX extractor.
596
+ #[tokio::test]
597
+ async fn test_rtf_word_sample_matches_docx_metadata_and_content() {
598
+ let root = workspace_root();
599
+ let rtf_path = root.join("test_documents/rtf/word_sample.rtf");
600
+ let docx_path = root.join("test_documents/documents/word_sample.docx");
601
+
602
+ if !rtf_path.exists() || !docx_path.exists() {
603
+ println!("Skipping word_sample parity test: fixtures missing");
604
+ return;
605
+ }
606
+
607
+ let config = ExtractionConfig::default();
608
+ let rtf_result = extract_file(&rtf_path, Some("application/rtf"), &config)
609
+ .await
610
+ .expect("RTF extraction should succeed for word_sample");
611
+ let docx_result = extract_file(&docx_path, None, &config)
612
+ .await
613
+ .expect("DOCX extraction should succeed for word_sample");
614
+
615
+ let rtf_content_lower = rtf_result.content.to_lowercase();
616
+ assert!(
617
+ rtf_content_lower.contains("swim"),
618
+ "RTF content should include the same body text as DOCX"
619
+ );
620
+
621
+ for key in ["created_by", "modified_by", "created_at", "revision"] {
622
+ assert_eq!(
623
+ rtf_result.metadata.additional.get(key).and_then(|v| v.as_str()),
624
+ docx_result.metadata.additional.get(key).and_then(|v| v.as_str()),
625
+ "Metadata field {} should align with DOCX",
626
+ key
627
+ );
628
+ }
629
+
630
+ for (key, expected) in [
631
+ ("page_count", 2),
632
+ ("word_count", 108),
633
+ ("character_count", 620),
634
+ ("line_count", 5),
635
+ ("paragraph_count", 1),
636
+ ] {
637
+ assert_eq!(
638
+ rtf_result.metadata.additional.get(key).and_then(|v| v.as_i64()),
639
+ Some(expected),
640
+ "Metadata field {} should match DOCX values",
641
+ key
642
+ );
643
+ }
644
+ }
645
+
646
+ /// RTF generated from lorem_ipsum.docx should expose the same document statistics
647
+ /// we validate for the DOCX extractor.
648
+ #[tokio::test]
649
+ async fn test_rtf_lorem_ipsum_metadata_alignment() {
650
+ let root = workspace_root();
651
+ let rtf_path = root.join("test_documents/rtf/lorem_ipsum.rtf");
652
+
653
+ if !rtf_path.exists() {
654
+ println!("Skipping lorem_ipsum metadata test: fixture missing");
655
+ return;
656
+ }
657
+
658
+ let config = ExtractionConfig::default();
659
+ let result = extract_file(&rtf_path, Some("application/rtf"), &config)
660
+ .await
661
+ .expect("RTF extraction should succeed for lorem_ipsum");
662
+
663
+ assert!(
664
+ result.content.to_lowercase().contains("lorem ipsum"),
665
+ "Content should contain lorem ipsum text"
666
+ );
667
+
668
+ for (key, expected) in [
669
+ ("page_count", 1),
670
+ ("word_count", 520),
671
+ ("character_count", 2967),
672
+ ("line_count", 24),
673
+ ("paragraph_count", 6),
674
+ ] {
675
+ assert_eq!(
676
+ result.metadata.additional.get(key).and_then(|v| v.as_i64()),
677
+ Some(expected),
678
+ "Metadata field {} should match DOCX values",
679
+ key
680
+ );
681
+ }
682
+ }
683
+
684
+ /// The comprehensive extraction fixture should mirror the coverage of the ODT/DOCX variants:
685
+ /// headings, section text, table content, and metadata fields should all be present.
686
+ #[tokio::test]
687
+ async fn test_rtf_comprehensive_extraction_alignment() {
688
+ let root = workspace_root();
689
+ let rtf_path = root.join("test_documents/rtf/extraction_test.rtf");
690
+ let docx_path = root.join("test_documents/extraction_test.docx");
691
+ let odt_path = root.join("test_documents/extraction_test.odt");
692
+
693
+ if !rtf_path.exists() {
694
+ println!("⚠️ Test document not found at {:?}, skipping", rtf_path);
695
+ return;
696
+ }
697
+ if !docx_path.exists() || !odt_path.exists() {
698
+ println!(
699
+ "⚠️ Companion DOCX/ODT documents missing (docx: {}, odt: {}), skipping",
700
+ docx_path.exists(),
701
+ odt_path.exists()
702
+ );
703
+ return;
704
+ }
705
+
706
+ let config = ExtractionConfig::default();
707
+ let rtf_result = extract_file(&rtf_path, Some("application/rtf"), &config)
708
+ .await
709
+ .expect("RTF extraction should succeed for extraction_test.rtf");
710
+ let docx_result = extract_file(&docx_path, None, &config)
711
+ .await
712
+ .expect("DOCX extraction should succeed for extraction_test.docx");
713
+ let odt_result = extract_file(&odt_path, None, &config)
714
+ .await
715
+ .expect("ODT extraction should succeed for extraction_test.odt");
716
+
717
+ assert!(
718
+ rtf_result.content.contains("Comprehensive Extraction Test Document"),
719
+ "Should include document heading"
720
+ );
721
+ assert!(
722
+ rtf_result.content.contains("First Section"),
723
+ "Should include first section heading"
724
+ );
725
+ assert!(
726
+ rtf_result.content.contains("Second Section"),
727
+ "Should include second section heading"
728
+ );
729
+ assert!(
730
+ rtf_result.content.contains("Third Section"),
731
+ "Should include third section heading"
732
+ );
733
+
734
+ for expected in ["Header 1", "Cell 1A", "Product", "Apple"] {
735
+ assert!(
736
+ rtf_result.content.contains(expected),
737
+ "Should include table content '{}'",
738
+ expected
739
+ );
740
+ }
741
+ assert!(
742
+ rtf_result.content.contains("|"),
743
+ "Should preserve table structure markers"
744
+ );
745
+ assert!(
746
+ !rtf_result.tables.is_empty(),
747
+ "Should extract structured tables from RTF"
748
+ );
749
+ assert!(
750
+ rtf_result
751
+ .tables
752
+ .iter()
753
+ .any(|t| t.markdown.contains("Header 1") || t.markdown.contains("Cell 1A")),
754
+ "Table markdown should include header/data cells"
755
+ );
756
+ assert!(
757
+ rtf_result.tables.len() >= docx_result.tables.len() && rtf_result.tables.len() >= odt_result.tables.len(),
758
+ "RTF should capture at least as many tables as DOCX/ODT"
759
+ );
760
+
761
+ for (key, expected) in [
762
+ ("page_count", 1),
763
+ ("word_count", 83),
764
+ ("character_count", 475),
765
+ ("line_count", 12),
766
+ ("paragraph_count", 8),
767
+ ] {
768
+ assert_eq!(
769
+ rtf_result.metadata.additional.get(key).and_then(|v| v.as_i64()),
770
+ Some(expected),
771
+ "Metadata field {} should be populated",
772
+ key
773
+ );
774
+ }
775
+ }