kreuzberg 4.0.0.pre.rc.29 → 4.0.0.rc1

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 (321) hide show
  1. checksums.yaml +4 -4
  2. data/.gitignore +0 -6
  3. data/.rubocop.yaml +534 -1
  4. data/Gemfile +2 -1
  5. data/Gemfile.lock +28 -116
  6. data/README.md +269 -629
  7. data/Rakefile +0 -9
  8. data/Steepfile +4 -8
  9. data/examples/async_patterns.rb +58 -1
  10. data/ext/kreuzberg_rb/extconf.rb +5 -35
  11. data/ext/kreuzberg_rb/native/Cargo.toml +16 -55
  12. data/ext/kreuzberg_rb/native/build.rs +14 -12
  13. data/ext/kreuzberg_rb/native/include/ieeefp.h +1 -1
  14. data/ext/kreuzberg_rb/native/include/msvc_compat/strings.h +1 -1
  15. data/ext/kreuzberg_rb/native/include/strings.h +2 -2
  16. data/ext/kreuzberg_rb/native/include/unistd.h +1 -1
  17. data/ext/kreuzberg_rb/native/src/lib.rs +34 -897
  18. data/extconf.rb +6 -38
  19. data/kreuzberg.gemspec +20 -114
  20. data/lib/kreuzberg/api_proxy.rb +18 -2
  21. data/lib/kreuzberg/cache_api.rb +0 -22
  22. data/lib/kreuzberg/cli.rb +10 -2
  23. data/lib/kreuzberg/cli_proxy.rb +10 -0
  24. data/lib/kreuzberg/config.rb +22 -274
  25. data/lib/kreuzberg/errors.rb +7 -73
  26. data/lib/kreuzberg/extraction_api.rb +8 -237
  27. data/lib/kreuzberg/mcp_proxy.rb +11 -2
  28. data/lib/kreuzberg/ocr_backend_protocol.rb +73 -0
  29. data/lib/kreuzberg/post_processor_protocol.rb +71 -0
  30. data/lib/kreuzberg/result.rb +33 -151
  31. data/lib/kreuzberg/setup_lib_path.rb +2 -22
  32. data/lib/kreuzberg/validator_protocol.rb +73 -0
  33. data/lib/kreuzberg/version.rb +1 -1
  34. data/lib/kreuzberg.rb +13 -27
  35. data/pkg/kreuzberg-4.0.0.rc1.gem +0 -0
  36. data/sig/kreuzberg.rbs +12 -105
  37. data/spec/binding/cache_spec.rb +22 -22
  38. data/spec/binding/cli_proxy_spec.rb +4 -2
  39. data/spec/binding/cli_spec.rb +11 -12
  40. data/spec/binding/config_spec.rb +0 -74
  41. data/spec/binding/config_validation_spec.rb +6 -100
  42. data/spec/binding/error_handling_spec.rb +97 -283
  43. data/spec/binding/plugins/ocr_backend_spec.rb +8 -8
  44. data/spec/binding/plugins/postprocessor_spec.rb +11 -11
  45. data/spec/binding/plugins/validator_spec.rb +13 -12
  46. data/spec/examples.txt +104 -0
  47. data/spec/fixtures/config.toml +1 -0
  48. data/spec/fixtures/config.yaml +1 -0
  49. data/spec/fixtures/invalid_config.toml +1 -0
  50. data/spec/smoke/package_spec.rb +3 -2
  51. data/spec/spec_helper.rb +3 -1
  52. data/vendor/kreuzberg/Cargo.toml +67 -192
  53. data/vendor/kreuzberg/README.md +9 -97
  54. data/vendor/kreuzberg/build.rs +194 -516
  55. data/vendor/kreuzberg/src/api/handlers.rs +9 -130
  56. data/vendor/kreuzberg/src/api/mod.rs +3 -18
  57. data/vendor/kreuzberg/src/api/server.rs +71 -236
  58. data/vendor/kreuzberg/src/api/types.rs +7 -43
  59. data/vendor/kreuzberg/src/bin/profile_extract.rs +455 -0
  60. data/vendor/kreuzberg/src/cache/mod.rs +3 -27
  61. data/vendor/kreuzberg/src/chunking/mod.rs +79 -1705
  62. data/vendor/kreuzberg/src/core/batch_mode.rs +0 -60
  63. data/vendor/kreuzberg/src/core/config.rs +23 -905
  64. data/vendor/kreuzberg/src/core/extractor.rs +106 -403
  65. data/vendor/kreuzberg/src/core/io.rs +2 -4
  66. data/vendor/kreuzberg/src/core/mime.rs +12 -2
  67. data/vendor/kreuzberg/src/core/mod.rs +3 -22
  68. data/vendor/kreuzberg/src/core/pipeline.rs +78 -395
  69. data/vendor/kreuzberg/src/embeddings.rs +21 -169
  70. data/vendor/kreuzberg/src/error.rs +2 -2
  71. data/vendor/kreuzberg/src/extraction/archive.rs +31 -36
  72. data/vendor/kreuzberg/src/extraction/docx.rs +1 -365
  73. data/vendor/kreuzberg/src/extraction/email.rs +11 -12
  74. data/vendor/kreuzberg/src/extraction/excel.rs +129 -138
  75. data/vendor/kreuzberg/src/extraction/html.rs +170 -1447
  76. data/vendor/kreuzberg/src/extraction/image.rs +14 -138
  77. data/vendor/kreuzberg/src/extraction/libreoffice.rs +3 -13
  78. data/vendor/kreuzberg/src/extraction/mod.rs +5 -21
  79. data/vendor/kreuzberg/src/extraction/office_metadata/mod.rs +0 -2
  80. data/vendor/kreuzberg/src/extraction/pandoc/batch.rs +275 -0
  81. data/vendor/kreuzberg/src/extraction/pandoc/mime_types.rs +178 -0
  82. data/vendor/kreuzberg/src/extraction/pandoc/mod.rs +491 -0
  83. data/vendor/kreuzberg/src/extraction/pandoc/server.rs +496 -0
  84. data/vendor/kreuzberg/src/extraction/pandoc/subprocess.rs +1188 -0
  85. data/vendor/kreuzberg/src/extraction/pandoc/version.rs +162 -0
  86. data/vendor/kreuzberg/src/extraction/pptx.rs +94 -196
  87. data/vendor/kreuzberg/src/extraction/structured.rs +4 -5
  88. data/vendor/kreuzberg/src/extraction/table.rs +1 -2
  89. data/vendor/kreuzberg/src/extraction/text.rs +10 -18
  90. data/vendor/kreuzberg/src/extractors/archive.rs +0 -22
  91. data/vendor/kreuzberg/src/extractors/docx.rs +148 -69
  92. data/vendor/kreuzberg/src/extractors/email.rs +9 -37
  93. data/vendor/kreuzberg/src/extractors/excel.rs +40 -81
  94. data/vendor/kreuzberg/src/extractors/html.rs +173 -182
  95. data/vendor/kreuzberg/src/extractors/image.rs +8 -32
  96. data/vendor/kreuzberg/src/extractors/mod.rs +10 -171
  97. data/vendor/kreuzberg/src/extractors/pandoc.rs +201 -0
  98. data/vendor/kreuzberg/src/extractors/pdf.rs +64 -329
  99. data/vendor/kreuzberg/src/extractors/pptx.rs +34 -79
  100. data/vendor/kreuzberg/src/extractors/structured.rs +0 -16
  101. data/vendor/kreuzberg/src/extractors/text.rs +7 -30
  102. data/vendor/kreuzberg/src/extractors/xml.rs +8 -27
  103. data/vendor/kreuzberg/src/keywords/processor.rs +1 -9
  104. data/vendor/kreuzberg/src/keywords/rake.rs +1 -0
  105. data/vendor/kreuzberg/src/language_detection/mod.rs +51 -94
  106. data/vendor/kreuzberg/src/lib.rs +5 -17
  107. data/vendor/kreuzberg/src/mcp/mod.rs +1 -4
  108. data/vendor/kreuzberg/src/mcp/server.rs +21 -145
  109. data/vendor/kreuzberg/src/ocr/mod.rs +0 -2
  110. data/vendor/kreuzberg/src/ocr/processor.rs +8 -19
  111. data/vendor/kreuzberg/src/ocr/tesseract_backend.rs +0 -2
  112. data/vendor/kreuzberg/src/pdf/error.rs +1 -93
  113. data/vendor/kreuzberg/src/pdf/metadata.rs +100 -263
  114. data/vendor/kreuzberg/src/pdf/mod.rs +2 -33
  115. data/vendor/kreuzberg/src/pdf/rendering.rs +12 -12
  116. data/vendor/kreuzberg/src/pdf/table.rs +64 -61
  117. data/vendor/kreuzberg/src/pdf/text.rs +24 -416
  118. data/vendor/kreuzberg/src/plugins/extractor.rs +8 -40
  119. data/vendor/kreuzberg/src/plugins/mod.rs +0 -3
  120. data/vendor/kreuzberg/src/plugins/ocr.rs +14 -22
  121. data/vendor/kreuzberg/src/plugins/processor.rs +1 -10
  122. data/vendor/kreuzberg/src/plugins/registry.rs +0 -15
  123. data/vendor/kreuzberg/src/plugins/validator.rs +8 -20
  124. data/vendor/kreuzberg/src/stopwords/mod.rs +2 -2
  125. data/vendor/kreuzberg/src/text/mod.rs +0 -8
  126. data/vendor/kreuzberg/src/text/quality.rs +15 -28
  127. data/vendor/kreuzberg/src/text/string_utils.rs +10 -22
  128. data/vendor/kreuzberg/src/text/token_reduction/core.rs +50 -86
  129. data/vendor/kreuzberg/src/text/token_reduction/filters.rs +16 -37
  130. data/vendor/kreuzberg/src/text/token_reduction/simd_text.rs +1 -2
  131. data/vendor/kreuzberg/src/types.rs +67 -907
  132. data/vendor/kreuzberg/src/utils/mod.rs +0 -14
  133. data/vendor/kreuzberg/src/utils/quality.rs +3 -12
  134. data/vendor/kreuzberg/tests/api_tests.rs +0 -506
  135. data/vendor/kreuzberg/tests/archive_integration.rs +0 -2
  136. data/vendor/kreuzberg/tests/batch_orchestration.rs +12 -57
  137. data/vendor/kreuzberg/tests/batch_processing.rs +8 -32
  138. data/vendor/kreuzberg/tests/chunking_offset_demo.rs +92 -0
  139. data/vendor/kreuzberg/tests/concurrency_stress.rs +8 -40
  140. data/vendor/kreuzberg/tests/config_features.rs +1 -33
  141. data/vendor/kreuzberg/tests/config_loading_tests.rs +39 -16
  142. data/vendor/kreuzberg/tests/core_integration.rs +9 -35
  143. data/vendor/kreuzberg/tests/csv_integration.rs +81 -71
  144. data/vendor/kreuzberg/tests/docx_metadata_extraction_test.rs +25 -23
  145. data/vendor/kreuzberg/tests/email_integration.rs +1 -3
  146. data/vendor/kreuzberg/tests/error_handling.rs +34 -43
  147. data/vendor/kreuzberg/tests/format_integration.rs +1 -7
  148. data/vendor/kreuzberg/tests/helpers/mod.rs +0 -60
  149. data/vendor/kreuzberg/tests/image_integration.rs +0 -2
  150. data/vendor/kreuzberg/tests/mime_detection.rs +16 -17
  151. data/vendor/kreuzberg/tests/ocr_configuration.rs +0 -4
  152. data/vendor/kreuzberg/tests/ocr_errors.rs +0 -22
  153. data/vendor/kreuzberg/tests/ocr_quality.rs +0 -2
  154. data/vendor/kreuzberg/tests/pandoc_integration.rs +503 -0
  155. data/vendor/kreuzberg/tests/pdf_integration.rs +0 -2
  156. data/vendor/kreuzberg/tests/pipeline_integration.rs +2 -36
  157. data/vendor/kreuzberg/tests/plugin_ocr_backend_test.rs +0 -5
  158. data/vendor/kreuzberg/tests/plugin_postprocessor_test.rs +1 -17
  159. data/vendor/kreuzberg/tests/plugin_system.rs +0 -6
  160. data/vendor/kreuzberg/tests/registry_integration_tests.rs +22 -2
  161. data/vendor/kreuzberg/tests/security_validation.rs +1 -13
  162. data/vendor/kreuzberg/tests/test_fastembed.rs +23 -45
  163. metadata +25 -171
  164. data/.rubocop.yml +0 -543
  165. data/ext/kreuzberg_rb/native/.cargo/config.toml +0 -23
  166. data/ext/kreuzberg_rb/native/Cargo.lock +0 -7619
  167. data/lib/kreuzberg/error_context.rb +0 -136
  168. data/lib/kreuzberg/types.rb +0 -170
  169. data/lib/libpdfium.so +0 -0
  170. data/spec/binding/async_operations_spec.rb +0 -473
  171. data/spec/binding/batch_operations_spec.rb +0 -595
  172. data/spec/binding/batch_spec.rb +0 -359
  173. data/spec/binding/config_result_spec.rb +0 -377
  174. data/spec/binding/embeddings_spec.rb +0 -816
  175. data/spec/binding/error_recovery_spec.rb +0 -488
  176. data/spec/binding/font_config_spec.rb +0 -220
  177. data/spec/binding/images_spec.rb +0 -738
  178. data/spec/binding/keywords_extraction_spec.rb +0 -600
  179. data/spec/binding/metadata_types_spec.rb +0 -1228
  180. data/spec/binding/pages_extraction_spec.rb +0 -471
  181. data/spec/binding/tables_spec.rb +0 -641
  182. data/spec/unit/config/chunking_config_spec.rb +0 -213
  183. data/spec/unit/config/embedding_config_spec.rb +0 -343
  184. data/spec/unit/config/extraction_config_spec.rb +0 -438
  185. data/spec/unit/config/font_config_spec.rb +0 -285
  186. data/spec/unit/config/hierarchy_config_spec.rb +0 -314
  187. data/spec/unit/config/image_extraction_config_spec.rb +0 -209
  188. data/spec/unit/config/image_preprocessing_config_spec.rb +0 -249
  189. data/spec/unit/config/keyword_config_spec.rb +0 -229
  190. data/spec/unit/config/language_detection_config_spec.rb +0 -258
  191. data/spec/unit/config/ocr_config_spec.rb +0 -171
  192. data/spec/unit/config/page_config_spec.rb +0 -221
  193. data/spec/unit/config/pdf_config_spec.rb +0 -267
  194. data/spec/unit/config/postprocessor_config_spec.rb +0 -290
  195. data/spec/unit/config/tesseract_config_spec.rb +0 -181
  196. data/spec/unit/config/token_reduction_config_spec.rb +0 -251
  197. data/test/metadata_types_test.rb +0 -959
  198. data/vendor/Cargo.toml +0 -61
  199. data/vendor/kreuzberg/examples/bench_fixes.rs +0 -71
  200. data/vendor/kreuzberg/examples/test_pdfium_fork.rs +0 -62
  201. data/vendor/kreuzberg/src/chunking/processor.rs +0 -219
  202. data/vendor/kreuzberg/src/core/batch_optimizations.rs +0 -385
  203. data/vendor/kreuzberg/src/core/config_validation.rs +0 -949
  204. data/vendor/kreuzberg/src/core/formats.rs +0 -235
  205. data/vendor/kreuzberg/src/core/server_config.rs +0 -1220
  206. data/vendor/kreuzberg/src/extraction/capacity.rs +0 -263
  207. data/vendor/kreuzberg/src/extraction/markdown.rs +0 -216
  208. data/vendor/kreuzberg/src/extraction/office_metadata/odt_properties.rs +0 -284
  209. data/vendor/kreuzberg/src/extractors/bibtex.rs +0 -470
  210. data/vendor/kreuzberg/src/extractors/docbook.rs +0 -504
  211. data/vendor/kreuzberg/src/extractors/epub.rs +0 -696
  212. data/vendor/kreuzberg/src/extractors/fictionbook.rs +0 -492
  213. data/vendor/kreuzberg/src/extractors/jats.rs +0 -1054
  214. data/vendor/kreuzberg/src/extractors/jupyter.rs +0 -368
  215. data/vendor/kreuzberg/src/extractors/latex.rs +0 -653
  216. data/vendor/kreuzberg/src/extractors/markdown.rs +0 -701
  217. data/vendor/kreuzberg/src/extractors/odt.rs +0 -628
  218. data/vendor/kreuzberg/src/extractors/opml.rs +0 -635
  219. data/vendor/kreuzberg/src/extractors/orgmode.rs +0 -529
  220. data/vendor/kreuzberg/src/extractors/rst.rs +0 -577
  221. data/vendor/kreuzberg/src/extractors/rtf.rs +0 -809
  222. data/vendor/kreuzberg/src/extractors/security.rs +0 -484
  223. data/vendor/kreuzberg/src/extractors/security_tests.rs +0 -367
  224. data/vendor/kreuzberg/src/extractors/typst.rs +0 -651
  225. data/vendor/kreuzberg/src/language_detection/processor.rs +0 -218
  226. data/vendor/kreuzberg/src/ocr/language_registry.rs +0 -520
  227. data/vendor/kreuzberg/src/panic_context.rs +0 -154
  228. data/vendor/kreuzberg/src/pdf/bindings.rs +0 -306
  229. data/vendor/kreuzberg/src/pdf/bundled.rs +0 -408
  230. data/vendor/kreuzberg/src/pdf/fonts.rs +0 -358
  231. data/vendor/kreuzberg/src/pdf/hierarchy.rs +0 -903
  232. data/vendor/kreuzberg/src/text/quality_processor.rs +0 -231
  233. data/vendor/kreuzberg/src/text/utf8_validation.rs +0 -193
  234. data/vendor/kreuzberg/src/utils/pool.rs +0 -503
  235. data/vendor/kreuzberg/src/utils/pool_sizing.rs +0 -364
  236. data/vendor/kreuzberg/src/utils/string_pool.rs +0 -761
  237. data/vendor/kreuzberg/tests/api_embed.rs +0 -360
  238. data/vendor/kreuzberg/tests/api_extract_multipart.rs +0 -52
  239. data/vendor/kreuzberg/tests/api_large_pdf_extraction.rs +0 -471
  240. data/vendor/kreuzberg/tests/api_large_pdf_extraction_diagnostics.rs +0 -289
  241. data/vendor/kreuzberg/tests/batch_pooling_benchmark.rs +0 -154
  242. data/vendor/kreuzberg/tests/bibtex_parity_test.rs +0 -421
  243. data/vendor/kreuzberg/tests/config_integration_test.rs +0 -753
  244. data/vendor/kreuzberg/tests/data/hierarchy_ground_truth.json +0 -294
  245. data/vendor/kreuzberg/tests/docbook_extractor_tests.rs +0 -500
  246. data/vendor/kreuzberg/tests/docx_vs_pandoc_comparison.rs +0 -370
  247. data/vendor/kreuzberg/tests/epub_native_extractor_tests.rs +0 -275
  248. data/vendor/kreuzberg/tests/fictionbook_extractor_tests.rs +0 -228
  249. data/vendor/kreuzberg/tests/html_table_test.rs +0 -551
  250. data/vendor/kreuzberg/tests/instrumentation_test.rs +0 -139
  251. data/vendor/kreuzberg/tests/jats_extractor_tests.rs +0 -639
  252. data/vendor/kreuzberg/tests/jupyter_extractor_tests.rs +0 -704
  253. data/vendor/kreuzberg/tests/latex_extractor_tests.rs +0 -496
  254. data/vendor/kreuzberg/tests/markdown_extractor_tests.rs +0 -490
  255. data/vendor/kreuzberg/tests/ocr_language_registry.rs +0 -191
  256. data/vendor/kreuzberg/tests/odt_extractor_tests.rs +0 -674
  257. data/vendor/kreuzberg/tests/opml_extractor_tests.rs +0 -616
  258. data/vendor/kreuzberg/tests/orgmode_extractor_tests.rs +0 -822
  259. data/vendor/kreuzberg/tests/page_markers.rs +0 -297
  260. data/vendor/kreuzberg/tests/pdf_hierarchy_detection.rs +0 -301
  261. data/vendor/kreuzberg/tests/pdf_hierarchy_quality.rs +0 -589
  262. data/vendor/kreuzberg/tests/pdf_ocr_triggering.rs +0 -301
  263. data/vendor/kreuzberg/tests/pdf_text_merging.rs +0 -475
  264. data/vendor/kreuzberg/tests/pdfium_linking.rs +0 -340
  265. data/vendor/kreuzberg/tests/rst_extractor_tests.rs +0 -694
  266. data/vendor/kreuzberg/tests/rtf_extractor_tests.rs +0 -775
  267. data/vendor/kreuzberg/tests/typst_behavioral_tests.rs +0 -1260
  268. data/vendor/kreuzberg/tests/typst_extractor_tests.rs +0 -648
  269. data/vendor/kreuzberg-ffi/Cargo.toml +0 -67
  270. data/vendor/kreuzberg-ffi/README.md +0 -851
  271. data/vendor/kreuzberg-ffi/benches/result_view_benchmark.rs +0 -227
  272. data/vendor/kreuzberg-ffi/build.rs +0 -168
  273. data/vendor/kreuzberg-ffi/cbindgen.toml +0 -37
  274. data/vendor/kreuzberg-ffi/kreuzberg-ffi.pc.in +0 -12
  275. data/vendor/kreuzberg-ffi/kreuzberg.h +0 -3012
  276. data/vendor/kreuzberg-ffi/src/batch_streaming.rs +0 -588
  277. data/vendor/kreuzberg-ffi/src/config.rs +0 -1341
  278. data/vendor/kreuzberg-ffi/src/error.rs +0 -901
  279. data/vendor/kreuzberg-ffi/src/extraction.rs +0 -555
  280. data/vendor/kreuzberg-ffi/src/helpers.rs +0 -879
  281. data/vendor/kreuzberg-ffi/src/lib.rs +0 -977
  282. data/vendor/kreuzberg-ffi/src/memory.rs +0 -493
  283. data/vendor/kreuzberg-ffi/src/mime.rs +0 -329
  284. data/vendor/kreuzberg-ffi/src/panic_shield.rs +0 -265
  285. data/vendor/kreuzberg-ffi/src/plugins/document_extractor.rs +0 -442
  286. data/vendor/kreuzberg-ffi/src/plugins/mod.rs +0 -14
  287. data/vendor/kreuzberg-ffi/src/plugins/ocr_backend.rs +0 -628
  288. data/vendor/kreuzberg-ffi/src/plugins/post_processor.rs +0 -438
  289. data/vendor/kreuzberg-ffi/src/plugins/validator.rs +0 -329
  290. data/vendor/kreuzberg-ffi/src/result.rs +0 -510
  291. data/vendor/kreuzberg-ffi/src/result_pool.rs +0 -639
  292. data/vendor/kreuzberg-ffi/src/result_view.rs +0 -773
  293. data/vendor/kreuzberg-ffi/src/string_intern.rs +0 -568
  294. data/vendor/kreuzberg-ffi/src/types.rs +0 -363
  295. data/vendor/kreuzberg-ffi/src/util.rs +0 -210
  296. data/vendor/kreuzberg-ffi/src/validation.rs +0 -848
  297. data/vendor/kreuzberg-ffi/tests.disabled/README.md +0 -48
  298. data/vendor/kreuzberg-ffi/tests.disabled/config_loading_tests.rs +0 -299
  299. data/vendor/kreuzberg-ffi/tests.disabled/config_tests.rs +0 -346
  300. data/vendor/kreuzberg-ffi/tests.disabled/extractor_tests.rs +0 -232
  301. data/vendor/kreuzberg-ffi/tests.disabled/plugin_registration_tests.rs +0 -470
  302. data/vendor/kreuzberg-tesseract/.commitlintrc.json +0 -13
  303. data/vendor/kreuzberg-tesseract/.crate-ignore +0 -2
  304. data/vendor/kreuzberg-tesseract/Cargo.lock +0 -2933
  305. data/vendor/kreuzberg-tesseract/Cargo.toml +0 -57
  306. data/vendor/kreuzberg-tesseract/LICENSE +0 -22
  307. data/vendor/kreuzberg-tesseract/README.md +0 -399
  308. data/vendor/kreuzberg-tesseract/build.rs +0 -1127
  309. data/vendor/kreuzberg-tesseract/patches/README.md +0 -71
  310. data/vendor/kreuzberg-tesseract/patches/tesseract.diff +0 -199
  311. data/vendor/kreuzberg-tesseract/src/api.rs +0 -1371
  312. data/vendor/kreuzberg-tesseract/src/choice_iterator.rs +0 -77
  313. data/vendor/kreuzberg-tesseract/src/enums.rs +0 -297
  314. data/vendor/kreuzberg-tesseract/src/error.rs +0 -81
  315. data/vendor/kreuzberg-tesseract/src/lib.rs +0 -145
  316. data/vendor/kreuzberg-tesseract/src/monitor.rs +0 -57
  317. data/vendor/kreuzberg-tesseract/src/mutable_iterator.rs +0 -197
  318. data/vendor/kreuzberg-tesseract/src/page_iterator.rs +0 -253
  319. data/vendor/kreuzberg-tesseract/src/result_iterator.rs +0 -286
  320. data/vendor/kreuzberg-tesseract/src/result_renderer.rs +0 -183
  321. data/vendor/kreuzberg-tesseract/tests/integration_test.rs +0 -211
@@ -1,490 +0,0 @@
1
- //! Comprehensive Markdown Extractor Tests
2
- //!
3
- //! This test suite uses Pandoc as a baseline for validating markdown extraction capabilities.
4
- //! It tests:
5
- //! - YAML frontmatter metadata extraction (both standard and extended fields)
6
- //! - Table extraction from various markdown table formats
7
- //! - Complex formatting and structure preservation
8
- //! - Comparison with Pandoc's metadata extraction capabilities
9
-
10
- #![cfg(feature = "office")]
11
-
12
- use std::path::PathBuf;
13
-
14
- use kreuzberg::core::config::ExtractionConfig;
15
- use kreuzberg::extractors::markdown::MarkdownExtractor;
16
- use kreuzberg::plugins::DocumentExtractor;
17
-
18
- fn markdown_fixture_path(relative: &str) -> PathBuf {
19
- PathBuf::from(env!("CARGO_MANIFEST_DIR"))
20
- .join("../../test_documents/markdown")
21
- .join(relative)
22
- }
23
-
24
- fn read_markdown_fixture(relative: &str) -> Vec<u8> {
25
- let path = markdown_fixture_path(relative);
26
- std::fs::read(&path).unwrap_or_else(|err| panic!("Failed to read markdown fixture {}: {}", path.display(), err))
27
- }
28
-
29
- /// Test comprehensive YAML frontmatter with all Pandoc-recognized fields
30
- #[tokio::test]
31
- async fn test_pandoc_baseline_yaml_fields() {
32
- let markdown_with_yaml = b"---\ntitle: Test Document\nauthor: John Doe\ndate: 2024-01-15\nkeywords:\n - markdown\n - testing\n - rust\ndescription: A comprehensive test document\nabstract: This is an abstract\nsubject: Testing Subject\ncategory: Documentation\ntags:\n - important\n - draft\nlanguage: en\nversion: 1.0.0\n---\n\n# Content\n\nThis is the main content.";
33
-
34
- let extractor = MarkdownExtractor::new();
35
- let result = extractor
36
- .extract_bytes(markdown_with_yaml, "text/markdown", &ExtractionConfig::default())
37
- .await
38
- .expect("Should extract markdown with frontmatter");
39
-
40
- assert_eq!(
41
- result.metadata.additional.get("title").and_then(|v| v.as_str()),
42
- Some("Test Document")
43
- );
44
- assert_eq!(
45
- result.metadata.additional.get("author").and_then(|v| v.as_str()),
46
- Some("John Doe")
47
- );
48
- assert_eq!(result.metadata.created_at, Some("2024-01-15".to_string()));
49
-
50
- assert!(result.metadata.additional.contains_key("keywords"));
51
- let keywords = result
52
- .metadata
53
- .additional
54
- .get("keywords")
55
- .and_then(|v| v.as_str())
56
- .unwrap_or("");
57
- assert!(keywords.contains("markdown"));
58
- assert!(keywords.contains("testing"));
59
- assert!(keywords.contains("rust"));
60
-
61
- assert_eq!(
62
- result.metadata.additional.get("abstract").and_then(|v| v.as_str()),
63
- Some("This is an abstract")
64
- );
65
-
66
- assert_eq!(result.metadata.subject, Some("Testing Subject".to_string()));
67
-
68
- assert_eq!(
69
- result.metadata.additional.get("category").and_then(|v| v.as_str()),
70
- Some("Documentation")
71
- );
72
-
73
- assert!(result.metadata.additional.contains_key("tags"));
74
- let tags = result
75
- .metadata
76
- .additional
77
- .get("tags")
78
- .and_then(|v| v.as_str())
79
- .unwrap_or("");
80
- assert!(tags.contains("important"));
81
- assert!(tags.contains("draft"));
82
-
83
- assert_eq!(
84
- result.metadata.additional.get("language").and_then(|v| v.as_str()),
85
- Some("en")
86
- );
87
-
88
- assert_eq!(
89
- result.metadata.additional.get("version").and_then(|v| v.as_str()),
90
- Some("1.0.0")
91
- );
92
- }
93
-
94
- /// Test table extraction from pipe-format markdown tables
95
- #[tokio::test]
96
- async fn test_extract_simple_pipe_tables() {
97
- let markdown = b"# Tables Example\n\n| Header 1 | Header 2 | Header 3 |\n|----------|----------|----------|\n| Row1Col1 | Row1Col2 | Row1Col3 |\n| Row2Col1 | Row2Col2 | Row2Col3 |";
98
-
99
- let extractor = MarkdownExtractor::new();
100
- let result = extractor
101
- .extract_bytes(markdown, "text/markdown", &ExtractionConfig::default())
102
- .await
103
- .expect("Should extract tables");
104
-
105
- assert!(!result.tables.is_empty(), "Should extract at least one table");
106
- let table = &result.tables[0];
107
-
108
- assert_eq!(table.cells.len(), 3, "Should have 3 rows (header + 2 data rows)");
109
- assert_eq!(table.cells[0].len(), 3, "Should have 3 columns");
110
-
111
- assert_eq!(table.cells[0][0], "Header 1");
112
- assert_eq!(table.cells[0][1], "Header 2");
113
- assert_eq!(table.cells[0][2], "Header 3");
114
-
115
- assert_eq!(table.cells[1][0], "Row1Col1");
116
- assert_eq!(table.cells[2][0], "Row2Col1");
117
-
118
- assert!(table.markdown.contains("Header 1"));
119
- assert!(table.markdown.contains("Row1Col1"));
120
- assert!(table.markdown.contains("---"));
121
- }
122
-
123
- /// Test extraction of grid tables (as found in comprehensive.md)
124
- #[tokio::test]
125
- async fn test_extract_grid_tables() {
126
- let markdown = b"# Grid Table Example\n\n+--------+--------+\n| Cell 1 | Cell 2 |\n+========+========+\n| Cell 3 | Cell 4 |\n+--------+--------+\n| Cell 5 | Cell 6 |\n+--------+--------+";
127
-
128
- let extractor = MarkdownExtractor::new();
129
- let result = extractor
130
- .extract_bytes(markdown, "text/markdown", &ExtractionConfig::default())
131
- .await
132
- .expect("Should extract grid tables");
133
-
134
- let _ = result.tables;
135
- }
136
-
137
- /// Test extraction of tables with multiple blocks in cells
138
- #[tokio::test]
139
- async fn test_extract_complex_table_cells() {
140
- let markdown = b"# Complex Table\n\n| Header 1 | Header 2 |\n|----------|----------|\n| - bullet 1<br/>- bullet 2 | Simple text |\n| **Bold** *italic* | `code` |";
141
-
142
- let extractor = MarkdownExtractor::new();
143
- let result = extractor
144
- .extract_bytes(markdown, "text/markdown", &ExtractionConfig::default())
145
- .await
146
- .expect("Should extract tables with complex formatting");
147
-
148
- assert!(!result.tables.is_empty());
149
- assert!(!result.content.is_empty());
150
- }
151
-
152
- /// Test multiline table from tables.markdown
153
- #[tokio::test]
154
- async fn test_pandoc_style_multiline_table() {
155
- let markdown = b"Simple table with caption:\n\n Right Left Center Default\n ------- ------ -------- ---------\n 12 12 12 12\n 123 123 123 123\n 1 1 1 1\n\n : Demonstration of simple table syntax.";
156
-
157
- let extractor = MarkdownExtractor::new();
158
- let result = extractor
159
- .extract_bytes(markdown, "text/markdown", &ExtractionConfig::default())
160
- .await
161
- .expect("Should extract pandoc-style tables");
162
-
163
- assert!(result.content.contains("12") || result.content.contains("Demonstration"));
164
- }
165
-
166
- /// Test YAML frontmatter with author as list (Pandoc style)
167
- #[tokio::test]
168
- async fn test_pandoc_author_list() {
169
- let markdown = b"% Title\n% Author One; Author Two; Author Three\n\n# Content\n\nBody text.";
170
-
171
- let extractor = MarkdownExtractor::new();
172
- let result = extractor
173
- .extract_bytes(markdown, "text/markdown", &ExtractionConfig::default())
174
- .await
175
- .expect("Should extract markdown");
176
-
177
- assert!(!result.content.is_empty());
178
- }
179
-
180
- /// Test YAML with array keywords field (Pandoc format)
181
- #[tokio::test]
182
- async fn test_keywords_array_extraction() {
183
- let markdown =
184
- b"---\ntitle: Document\nkeywords:\n - rust\n - markdown\n - pandoc\n---\n\n# Main Content\n\nText here.";
185
-
186
- let extractor = MarkdownExtractor::new();
187
- let result = extractor
188
- .extract_bytes(markdown, "text/markdown", &ExtractionConfig::default())
189
- .await
190
- .expect("Should extract keywords array");
191
-
192
- assert!(result.metadata.additional.contains_key("keywords"));
193
- let keywords = result
194
- .metadata
195
- .additional
196
- .get("keywords")
197
- .and_then(|v| v.as_str())
198
- .unwrap_or("");
199
- assert!(keywords.contains("rust"));
200
- assert!(keywords.contains("markdown"));
201
- assert!(keywords.contains("pandoc"));
202
- }
203
-
204
- /// Test complex formatting in content (links, code, emphasis)
205
- #[tokio::test]
206
- async fn test_complex_markdown_formatting() {
207
- let markdown = b"# Document\n\nThis is a paragraph with [links](http://example.com) and `code blocks`.\n\n## Subsection\n\n- **Bold text**\n- *Italic text*\n- ***Bold italic***\n\n```rust\nfn main() {\n println!(\"Hello, world!\");\n}\n```";
208
-
209
- let extractor = MarkdownExtractor::new();
210
- let result = extractor
211
- .extract_bytes(markdown, "text/markdown", &ExtractionConfig::default())
212
- .await
213
- .expect("Should extract complex markdown");
214
-
215
- assert!(result.content.contains("links"));
216
- assert!(result.content.contains("code blocks"));
217
- assert!(result.content.contains("Bold text"));
218
- assert!(result.content.contains("println"));
219
- }
220
-
221
- /// Test extraction of raw HTML and LaTeX in markdown
222
- #[tokio::test]
223
- async fn test_raw_content_extraction() {
224
- let markdown = b"# Document\n\nSome text.\n\n<div>Raw HTML</div>\n\nMore text.\n\n\\\\begin{equation}\nx = y\n\\\\end{equation}";
225
-
226
- let extractor = MarkdownExtractor::new();
227
- let result = extractor
228
- .extract_bytes(markdown, "text/markdown", &ExtractionConfig::default())
229
- .await
230
- .expect("Should extract raw content");
231
-
232
- assert!(!result.content.is_empty());
233
- }
234
-
235
- /// Test comprehensive.md from test_documents
236
- #[tokio::test]
237
- async fn test_comprehensive_md_extraction() {
238
- let markdown = read_markdown_fixture("comprehensive.md");
239
-
240
- let extractor = MarkdownExtractor::new();
241
- let result = extractor
242
- .extract_bytes(&markdown, "text/markdown", &ExtractionConfig::default())
243
- .await
244
- .expect("Should extract comprehensive.md");
245
-
246
- assert!(!result.content.is_empty());
247
-
248
- let _has_title_or_author =
249
- result.metadata.additional.contains_key("title") || result.metadata.additional.contains_key("author");
250
-
251
- assert!(result.content.contains("Additional markdown reader tests") || result.content.contains("markdown"));
252
-
253
- let _ = result.tables;
254
- }
255
-
256
- /// Test tables.markdown from test_documents
257
- #[tokio::test]
258
- async fn test_tables_markdown_extraction() {
259
- let markdown = read_markdown_fixture("tables.markdown");
260
-
261
- let extractor = MarkdownExtractor::new();
262
- let result = extractor
263
- .extract_bytes(&markdown, "text/markdown", &ExtractionConfig::default())
264
- .await
265
- .expect("Should extract tables.markdown");
266
-
267
- assert!(!result.content.is_empty());
268
-
269
- assert!(result.content.contains("Right") || result.content.contains("Left") || result.content.contains("table"));
270
- }
271
-
272
- /// Test empty YAML frontmatter handling
273
- #[tokio::test]
274
- async fn test_empty_frontmatter() {
275
- let markdown = b"---\n---\n\n# Main Title\n\nContent here.";
276
-
277
- let extractor = MarkdownExtractor::new();
278
- let result = extractor
279
- .extract_bytes(markdown, "text/markdown", &ExtractionConfig::default())
280
- .await
281
- .expect("Should handle empty frontmatter");
282
-
283
- assert!(result.content.contains("Main Title"));
284
- assert!(result.content.contains("Content here"));
285
- }
286
-
287
- /// Test malformed YAML frontmatter fallback
288
- #[tokio::test]
289
- async fn test_malformed_frontmatter_graceful_fallback() {
290
- let markdown = b"---\ninvalid: yaml: syntax: here:\n---\n\nContent here.";
291
-
292
- let extractor = MarkdownExtractor::new();
293
- let result = extractor
294
- .extract_bytes(markdown, "text/markdown", &ExtractionConfig::default())
295
- .await
296
- .expect("Should handle malformed YAML gracefully");
297
-
298
- assert!(!result.content.is_empty());
299
- }
300
-
301
- /// Test metadata field extraction for standard YAML fields
302
- #[tokio::test]
303
- async fn test_standard_yaml_metadata_fields() {
304
- let markdown =
305
- b"---\ntitle: Standard Fields Test\nauthor: Test Author\ndate: 2024-12-06\n---\n\n# Content\n\nTest body.";
306
-
307
- let extractor = MarkdownExtractor::new();
308
- let result = extractor
309
- .extract_bytes(markdown, "text/markdown", &ExtractionConfig::default())
310
- .await
311
- .expect("Should extract standard fields");
312
-
313
- assert_eq!(
314
- result.metadata.additional.get("title").and_then(|v| v.as_str()),
315
- Some("Standard Fields Test")
316
- );
317
- assert_eq!(
318
- result.metadata.additional.get("author").and_then(|v| v.as_str()),
319
- Some("Test Author")
320
- );
321
- assert_eq!(result.metadata.created_at, Some("2024-12-06".to_string()));
322
- }
323
-
324
- /// Test extraction of description field (maps to subject)
325
- #[tokio::test]
326
- async fn test_description_to_subject_mapping() {
327
- let markdown = b"---\ntitle: Test\ndescription: This is the document description\n---\n\nContent.";
328
-
329
- let extractor = MarkdownExtractor::new();
330
- let result = extractor
331
- .extract_bytes(markdown, "text/markdown", &ExtractionConfig::default())
332
- .await
333
- .expect("Should extract description");
334
-
335
- assert_eq!(
336
- result.metadata.subject,
337
- Some("This is the document description".to_string())
338
- );
339
- }
340
-
341
- /// Test multi-line title extraction from YAML
342
- #[tokio::test]
343
- async fn test_multiline_title_in_yaml() {
344
- let markdown = b"---\ntitle: |\n This is a\n multi-line title\nauthor: Test\n---\n\n# Content\n\nBody.";
345
-
346
- let extractor = MarkdownExtractor::new();
347
- let result = extractor
348
- .extract_bytes(markdown, "text/markdown", &ExtractionConfig::default())
349
- .await
350
- .expect("Should extract multiline title");
351
-
352
- let title = result.metadata.additional.get("title").and_then(|v| v.as_str());
353
- assert!(title.is_some());
354
- }
355
-
356
- /// Test table page numbering
357
- #[tokio::test]
358
- async fn test_table_page_numbering() {
359
- let markdown = b"# Document\n\n| A | B |\n|---|---|\n| 1 | 2 |\n\nSome text between tables.\n\n| X | Y |\n|---|---|\n| 3 | 4 |";
360
-
361
- let extractor = MarkdownExtractor::new();
362
- let result = extractor
363
- .extract_bytes(markdown, "text/markdown", &ExtractionConfig::default())
364
- .await
365
- .expect("Should extract multiple tables");
366
-
367
- assert_eq!(result.tables.len(), 2);
368
- assert_eq!(result.tables[0].page_number, 1);
369
- assert_eq!(result.tables[1].page_number, 2);
370
- }
371
-
372
- /// Test unicode content extraction
373
- #[tokio::test]
374
- async fn test_unicode_markdown_extraction() {
375
- let markdown = "---\ntitle: Unicode Test\nauthor: 日本人\n---\n\n# こんにちは\n\nThis document has:\n- 中文 (Chinese)\n- 日本語 (Japanese)\n- Русский (Russian)\n- العربية (Arabic)".as_bytes();
376
-
377
- let extractor = MarkdownExtractor::new();
378
- let result = extractor
379
- .extract_bytes(markdown, "text/markdown", &ExtractionConfig::default())
380
- .await
381
- .expect("Should extract unicode content");
382
-
383
- assert!(result.content.contains("こんにちは") || result.content.contains("Chinese"));
384
- }
385
-
386
- /// Test YAML list to comma-separated conversion for keywords
387
- #[tokio::test]
388
- async fn test_keywords_list_comma_separation() {
389
- let markdown = b"---\nkeywords:\n - first\n - second\n - third\n---\n\nContent.";
390
-
391
- let extractor = MarkdownExtractor::new();
392
- let result = extractor
393
- .extract_bytes(markdown, "text/markdown", &ExtractionConfig::default())
394
- .await
395
- .expect("Should extract keywords list");
396
-
397
- let keywords = result
398
- .metadata
399
- .additional
400
- .get("keywords")
401
- .and_then(|v| v.as_str())
402
- .unwrap_or("");
403
-
404
- assert!(keywords.contains(","));
405
- assert!(keywords.contains("first"));
406
- assert!(keywords.contains("second"));
407
- assert!(keywords.contains("third"));
408
- }
409
-
410
- /// Test extraction without any frontmatter
411
- #[tokio::test]
412
- async fn test_no_frontmatter_extraction() {
413
- let markdown = b"# Document Title\n\nJust a document without frontmatter.\n\n## Section\n\nWith content.";
414
-
415
- let extractor = MarkdownExtractor::new();
416
- let result = extractor
417
- .extract_bytes(markdown, "text/markdown", &ExtractionConfig::default())
418
- .await
419
- .expect("Should extract markdown without frontmatter");
420
-
421
- assert!(result.content.contains("Document Title"));
422
- assert!(result.content.contains("document") || result.content.contains("Section"));
423
-
424
- let title = result.metadata.additional.get("title").and_then(|v| v.as_str());
425
- assert_eq!(title, Some("Document Title"));
426
- }
427
-
428
- /// Test code block extraction
429
- #[tokio::test]
430
- async fn test_code_block_extraction() {
431
- let markdown = b"# Code Examples\n\n```rust\nfn add(a: i32, b: i32) -> i32 {\n a + b\n}\n```\n\n```python\ndef add(a, b):\n return a + b\n```";
432
-
433
- let extractor = MarkdownExtractor::new();
434
- let result = extractor
435
- .extract_bytes(markdown, "text/markdown", &ExtractionConfig::default())
436
- .await
437
- .expect("Should extract code blocks");
438
-
439
- assert!(result.content.contains("add"));
440
- assert!(result.content.contains("return"));
441
- }
442
-
443
- /// Test extraction with various mime types
444
- #[tokio::test]
445
- async fn test_supported_mime_types() {
446
- let markdown = b"# Test\n\nContent.";
447
- let extractor = MarkdownExtractor::new();
448
-
449
- for mime_type in &["text/markdown", "text/x-markdown", "text/x-gfm", "text/x-commonmark"] {
450
- let result = extractor
451
- .extract_bytes(markdown, mime_type, &ExtractionConfig::default())
452
- .await
453
- .unwrap_or_else(|_| panic!("Should support {}", mime_type));
454
-
455
- assert_eq!(result.mime_type, *mime_type);
456
- assert!(result.content.contains("Test"));
457
- }
458
- }
459
-
460
- /// Test that metadata extraction handles nested YAML structures
461
- /// (Currently not fully supported - documents what's missing)
462
- #[tokio::test]
463
- async fn test_nested_yaml_awareness() {
464
- let markdown = b"---\ntitle: Test\nmetadata:\n organization: Test Corp\n location:\n city: San Francisco\n state: CA\n---\n\nContent.";
465
-
466
- let extractor = MarkdownExtractor::new();
467
- let result = extractor
468
- .extract_bytes(markdown, "text/markdown", &ExtractionConfig::default())
469
- .await
470
- .expect("Should extract document");
471
-
472
- let title = result.metadata.additional.get("title").and_then(|v| v.as_str());
473
- assert_eq!(title, Some("Test"));
474
- }
475
-
476
- /// Test extraction with special characters in metadata
477
- #[tokio::test]
478
- async fn test_special_characters_in_metadata() {
479
- let markdown = b"---\ntitle: \"Document: Part 1 & 2\"\nauthor: O'Brien\nkeywords: \"C++, C#, F#\"\n---\n\nContent.";
480
-
481
- let extractor = MarkdownExtractor::new();
482
- let result = extractor
483
- .extract_bytes(markdown, "text/markdown", &ExtractionConfig::default())
484
- .await
485
- .expect("Should extract with special characters");
486
-
487
- let title = result.metadata.additional.get("title").and_then(|v| v.as_str());
488
- assert!(title.is_some());
489
- assert!(title.unwrap().contains("&") || title.unwrap().contains("Part"));
490
- }
@@ -1,191 +0,0 @@
1
- #![cfg(feature = "ocr")]
2
- //! Integration tests for OCR language registry
3
- //!
4
- //! Tests the language registry functionality across all OCR backends.
5
-
6
- use kreuzberg::ocr::LanguageRegistry;
7
-
8
- #[test]
9
- fn test_registry_provides_easyocr_languages() {
10
- let registry = LanguageRegistry::new();
11
- let languages = registry.get_supported_languages("easyocr").expect("EasyOCR not found");
12
-
13
- assert_eq!(languages.len(), 83);
14
-
15
- let expected = vec!["en", "fr", "de", "es", "it", "pt", "ja", "ko", "ch_sim", "ch_tra", "ru"];
16
- for lang in expected {
17
- assert!(
18
- languages.contains(&lang.to_string()),
19
- "Expected language '{}' not found in EasyOCR",
20
- lang
21
- );
22
- }
23
- }
24
-
25
- #[test]
26
- fn test_registry_provides_paddleocr_languages() {
27
- let registry = LanguageRegistry::new();
28
- let languages = registry
29
- .get_supported_languages("paddleocr")
30
- .expect("PaddleOCR not found");
31
-
32
- assert_eq!(languages.len(), 14);
33
-
34
- let expected = vec!["en", "ch", "french", "german", "korean", "japan", "arabic"];
35
- for lang in expected {
36
- assert!(
37
- languages.contains(&lang.to_string()),
38
- "Expected language '{}' not found in PaddleOCR",
39
- lang
40
- );
41
- }
42
- }
43
-
44
- #[test]
45
- fn test_registry_provides_tesseract_languages() {
46
- let registry = LanguageRegistry::new();
47
- let languages = registry
48
- .get_supported_languages("tesseract")
49
- .expect("Tesseract not found");
50
-
51
- assert!(languages.len() >= 100, "Tesseract should support 100+ languages");
52
-
53
- let expected = vec![
54
- "eng", "fra", "deu", "spa", "ita", "por", "jpn", "kor", "chi_sim", "chi_tra", "rus",
55
- ];
56
- for lang in expected {
57
- assert!(
58
- languages.contains(&lang.to_string()),
59
- "Expected language '{}' not found in Tesseract",
60
- lang
61
- );
62
- }
63
- }
64
-
65
- #[test]
66
- fn test_language_support_checking() {
67
- let registry = LanguageRegistry::new();
68
-
69
- assert!(registry.is_language_supported("easyocr", "en"));
70
- assert!(registry.is_language_supported("easyocr", "fr"));
71
- assert!(!registry.is_language_supported("easyocr", "invalid_lang"));
72
-
73
- assert!(registry.is_language_supported("paddleocr", "en"));
74
- assert!(registry.is_language_supported("paddleocr", "ch"));
75
- assert!(!registry.is_language_supported("paddleocr", "en_US"));
76
-
77
- assert!(registry.is_language_supported("tesseract", "eng"));
78
- assert!(registry.is_language_supported("tesseract", "fra"));
79
- assert!(!registry.is_language_supported("tesseract", "en"));
80
-
81
- assert!(!registry.is_language_supported("invalid_backend", "en"));
82
- }
83
-
84
- #[test]
85
- fn test_backend_enumeration() {
86
- let registry = LanguageRegistry::new();
87
- let backends = registry.get_backends();
88
-
89
- assert_eq!(backends.len(), 3);
90
- assert!(backends.contains(&"easyocr".to_string()));
91
- assert!(backends.contains(&"paddleocr".to_string()));
92
- assert!(backends.contains(&"tesseract".to_string()));
93
- }
94
-
95
- #[test]
96
- fn test_language_count_per_backend() {
97
- let registry = LanguageRegistry::new();
98
-
99
- assert_eq!(registry.get_language_count("easyocr"), 83);
100
- assert_eq!(registry.get_language_count("paddleocr"), 14);
101
- assert!(registry.get_language_count("tesseract") >= 100);
102
- assert_eq!(registry.get_language_count("nonexistent"), 0);
103
- }
104
-
105
- #[test]
106
- fn test_registry_singleton_behavior() {
107
- let global1 = LanguageRegistry::global();
108
- let global2 = LanguageRegistry::global();
109
-
110
- assert_eq!(
111
- global1.get_language_count("easyocr"),
112
- global2.get_language_count("easyocr")
113
- );
114
- assert_eq!(
115
- global1.get_language_count("paddleocr"),
116
- global2.get_language_count("paddleocr")
117
- );
118
- }
119
-
120
- #[test]
121
- fn test_easyocr_special_languages() {
122
- let registry = LanguageRegistry::new();
123
- let languages = registry.get_supported_languages("easyocr").unwrap();
124
-
125
- let special_langs = vec!["ch_sim", "ch_tra", "rs_cyrillic", "rs_latin"];
126
-
127
- for lang in special_langs {
128
- assert!(
129
- languages.contains(&lang.to_string()),
130
- "EasyOCR should support special language '{}'",
131
- lang
132
- );
133
- }
134
- }
135
-
136
- #[test]
137
- fn test_registry_clone() {
138
- let registry1 = LanguageRegistry::new();
139
- let registry2 = registry1.clone();
140
-
141
- assert_eq!(
142
- registry1.get_language_count("easyocr"),
143
- registry2.get_language_count("easyocr")
144
- );
145
- assert_eq!(registry1.get_backends(), registry2.get_backends());
146
- }
147
-
148
- #[test]
149
- fn test_registry_default() {
150
- let registry_default = LanguageRegistry::default();
151
- let registry_new = LanguageRegistry::new();
152
-
153
- assert_eq!(registry_default.get_backends().len(), registry_new.get_backends().len());
154
- }
155
-
156
- #[test]
157
- fn test_registry_consistency() {
158
- let registries: Vec<_> = (0..5).map(|_| LanguageRegistry::new()).collect();
159
-
160
- let expected_backends = vec!["easyocr", "paddleocr", "tesseract"];
161
- let expected_counts = vec![("easyocr", 83), ("paddleocr", 14), ("tesseract", 100_usize)];
162
-
163
- for registry in &registries {
164
- let backends = registry.get_backends();
165
- assert_eq!(backends.len(), 3);
166
-
167
- for expected_backend in &expected_backends {
168
- assert!(backends.contains(&expected_backend.to_string()));
169
- }
170
-
171
- for (backend, min_count) in &expected_counts {
172
- let count = registry.get_language_count(backend);
173
- if backend == &"tesseract" {
174
- assert!(count >= *min_count);
175
- } else {
176
- assert_eq!(count, *min_count);
177
- }
178
- }
179
- }
180
- }
181
-
182
- #[test]
183
- fn test_language_case_sensitivity() {
184
- let registry = LanguageRegistry::new();
185
-
186
- assert!(registry.is_language_supported("easyocr", "en"));
187
- assert!(!registry.is_language_supported("easyocr", "EN"));
188
-
189
- assert!(registry.is_language_supported("easyocr", "en"));
190
- assert!(!registry.is_language_supported("EASYOCR", "en"));
191
- }