fil/e2e/swift_e2e/Tests/KreuzbergE2ETests/PluginApiTests.swift

// 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")
    }

}