This commit is contained in:
114
e2e/swift_e2e/Tests/KreuzbergE2ETests/PluginApiTests.swift
generated
Normal file
114
e2e/swift_e2e/Tests/KreuzbergE2ETests/PluginApiTests.swift
generated
Normal file
@@ -0,0 +1,114 @@
|
||||
// This file is auto-generated by alef — DO NOT EDIT.
|
||||
// alef:hash:4e15143f4af1ae8bafbdb1506ef057da924484c66a19483966333558ad437e75
|
||||
// To regenerate: alef generate
|
||||
// To verify freshness: alef verify --exit-code
|
||||
// Issues & docs: https://github.com/kreuzberg-dev/alef
|
||||
// swift-format-ignore-file
|
||||
|
||||
import XCTest
|
||||
import Foundation
|
||||
#if canImport(FoundationNetworking)
|
||||
import FoundationNetworking
|
||||
#endif
|
||||
import Kreuzberg
|
||||
import RustBridge
|
||||
|
||||
/// E2e tests for category: plugin_api.
|
||||
final class PluginApiTests: XCTestCase {
|
||||
func testRegisterDocumentExtractorTraitBridge() throws {
|
||||
// register_document_extractor: trait bridge
|
||||
class TestStubRegisterDocumentExtractorTraitBridge: SwiftDocumentExtractorBridge {
|
||||
var name: String { "register_document_extractor_trait_bridge" }
|
||||
func extractBytes(content: Data, mimeType: String, config: ExtractionConfig) async throws -> InternalDocument { InternalDocument() }
|
||||
func supportedMimeTypes() -> [String] { [] }
|
||||
}
|
||||
|
||||
let result = try Kreuzberg.registerDocumentExtractor(TestStubRegisterDocumentExtractorTraitBridge())
|
||||
try? Kreuzberg.unregisterDocumentExtractor("swift-bridge-document_extractor")
|
||||
}
|
||||
|
||||
func testRegisterEmbeddingBackendTraitBridge() throws {
|
||||
// register_embedding_backend: trait bridge
|
||||
class TestStubRegisterEmbeddingBackendTraitBridge: SwiftEmbeddingBackendBridge {
|
||||
var name: String { "register_embedding_backend_trait_bridge" }
|
||||
func dimensions() -> UInt { 0 }
|
||||
func embed(texts: [String]) async throws -> [[Float]] { [] }
|
||||
}
|
||||
|
||||
let result = try Kreuzberg.registerEmbeddingBackend(TestStubRegisterEmbeddingBackendTraitBridge())
|
||||
try? Kreuzberg.unregisterEmbeddingBackend("swift-bridge-embedding_backend")
|
||||
}
|
||||
|
||||
func testRegisterOcrBackendTraitBridge() throws {
|
||||
// register_ocr_backend: trait bridge
|
||||
class TestStubRegisterOcrBackendTraitBridge: SwiftOcrBackendBridge {
|
||||
var name: String { "register_ocr_backend_trait_bridge" }
|
||||
func processImage(imageBytes: Data, config: OcrConfig) async throws -> ExtractionResult { ExtractionResult() }
|
||||
func supportsLanguage(lang: String) -> Bool { false }
|
||||
func backendType() -> OcrBackendType { OcrBackendType() }
|
||||
}
|
||||
|
||||
let result = try Kreuzberg.registerOcrBackend(TestStubRegisterOcrBackendTraitBridge())
|
||||
try? Kreuzberg.unregisterOcrBackend("swift-bridge-ocr_backend")
|
||||
}
|
||||
|
||||
func testRegisterPostProcessorTraitBridge() throws {
|
||||
// register_post_processor: trait bridge
|
||||
class TestStubRegisterPostProcessorTraitBridge: SwiftPostProcessorBridge {
|
||||
var name: String { "register_post_processor_trait_bridge" }
|
||||
func process(result: ExtractionResult, config: ExtractionConfig) async throws -> Void { () }
|
||||
func processingStage() -> ProcessingStage { ProcessingStage() }
|
||||
}
|
||||
|
||||
let result = try Kreuzberg.registerPostProcessor(TestStubRegisterPostProcessorTraitBridge())
|
||||
try? Kreuzberg.unregisterPostProcessor("swift-bridge-post_processor")
|
||||
}
|
||||
|
||||
func testRegisterRendererTraitBridge() throws {
|
||||
// register_renderer: trait bridge
|
||||
class TestStubRegisterRendererTraitBridge: SwiftRendererBridge {
|
||||
var name: String { "register_renderer_trait_bridge" }
|
||||
func render(doc: InternalDocument) -> String { "" }
|
||||
}
|
||||
|
||||
let result = try Kreuzberg.registerRenderer(TestStubRegisterRendererTraitBridge())
|
||||
try? Kreuzberg.unregisterRenderer("swift-bridge-renderer")
|
||||
}
|
||||
|
||||
func testRegisterValidatorTraitBridge() throws {
|
||||
// register_validator: trait bridge
|
||||
class TestStubRegisterValidatorTraitBridge: SwiftValidatorBridge {
|
||||
var name: String { "register_validator_trait_bridge" }
|
||||
func validate(result: ExtractionResult, config: ExtractionConfig) async throws -> Void { () }
|
||||
}
|
||||
|
||||
let result = try Kreuzberg.registerValidator(TestStubRegisterValidatorTraitBridge())
|
||||
try? Kreuzberg.unregisterValidator("swift-bridge-validator")
|
||||
}
|
||||
|
||||
func testUnregisterDocumentExtractorAfterRegister() throws {
|
||||
// unregister_document_extractor
|
||||
let result = try Kreuzberg.unregisterDocumentExtractor(name: "test-extractor")
|
||||
}
|
||||
|
||||
func testUnregisterEmbeddingBackendAfterRegister() throws {
|
||||
// unregister_embedding_backend
|
||||
let result = try Kreuzberg.unregisterEmbeddingBackend(name: "test-embedding-backend")
|
||||
}
|
||||
|
||||
func testUnregisterPostProcessorAfterRegister() throws {
|
||||
// unregister_post_processor
|
||||
let result = try Kreuzberg.unregisterPostProcessor(name: "test-processor")
|
||||
}
|
||||
|
||||
func testUnregisterRendererAfterRegister() throws {
|
||||
// unregister_renderer
|
||||
let result = try Kreuzberg.unregisterRenderer(name: "test-renderer")
|
||||
}
|
||||
|
||||
func testUnregisterValidatorAfterRegister() throws {
|
||||
// unregister_validator
|
||||
let result = try Kreuzberg.unregisterValidator(name: "test-validator")
|
||||
}
|
||||
|
||||
}
|
||||
Reference in New Issue
Block a user