nodedb-fts 0.6.1

// SPDX-License-Identifier: Apache-2.0

//! Shared test fixtures for this crate's inline `#[cfg(test)]` modules.

use std::sync::Arc;

use nodedb_mem::{EngineId, EngineLimits, GovernorConfig, MemoryGovernor, ScopedMemory};
use nodedb_types::{DatabaseId, TenantId};

/// Per-engine limit. The ceiling multiplies it back up, because the governor
/// rejects a config whose engine limits outgrow the global ceiling.
const PER_ENGINE: usize = usize::MAX / EngineId::COUNT;

/// A governor whose ceiling covers every engine's uniform per-engine limit.
pub(crate) fn test_governor() -> Arc<MemoryGovernor> {
    Arc::new(
        MemoryGovernor::new(GovernorConfig {
            global_ceiling: PER_ENGINE * EngineId::ALL.len(),
            engine_limits: EngineLimits::uniform(PER_ENGINE),
        })
        .expect("test governor"),
    )
}

/// A scope backed by [`test_governor`].
pub(crate) fn test_memory() -> ScopedMemory {
    ScopedMemory::new(
        test_governor(),
        DatabaseId::DEFAULT,
        TenantId::new(0),
        EngineId::Fts,
    )
}