datafold 0.1.55

A personal database for data sovereignty with AI-powered ingestion
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70

//! Consolidated testing utilities for database setup and common test patterns
//!
//! This module eliminates duplicate database setup code found across 11+ files

use crate::db_operations::DbOperations;
use crate::fold_db_core::infrastructure::message_bus::AsyncMessageBus;
use sled::{Db, Tree};
use std::sync::Arc;

/// Consolidated temporary database creation - eliminates 11+ duplicates
pub struct TestDatabaseFactory;

impl TestDatabaseFactory {
    /// Create a temporary sled database for testing
    pub fn create_temp_sled_db() -> Result<Db, sled::Error> {
        sled::Config::new().temporary(true).open()
    }

    /// Create temporary DbOperations for testing - consolidates pattern from multiple files
    pub async fn create_temp_db_ops() -> Result<DbOperations, Box<dyn std::error::Error>> {
        let db = Self::create_temp_sled_db()?;
        Ok(DbOperations::from_sled(db).await?)
    }

    /// Create temporary tree for testing - consolidates pattern from orchestration files  
    pub fn create_temp_tree() -> Result<Tree, sled::Error> {
        let db = Self::create_temp_sled_db()?;
        db.open_tree("test_tree")
    }

    /// Create complete test environment with db_ops and message bus
    pub async fn create_test_environment(
    ) -> Result<(Arc<DbOperations>, Arc<AsyncMessageBus>), Box<dyn std::error::Error>> {
        let db_ops = Arc::new(Self::create_temp_db_ops().await?);
        let message_bus = Arc::new(AsyncMessageBus::new());
        Ok((db_ops, message_bus))
    }

    /// Create test schema (consolidates duplicate create_test_schema functions)
    pub fn create_test_schema(name: &str) -> crate::schema::types::Schema {
        // Create a simple declarative schema for testing
        crate::schema::types::Schema::new(
            name.to_string(),
            crate::schema::types::SchemaType::Single,
            None,
            Some(vec![]),
            None,
            None,
        )
    }

    /// Create test node config (consolidates create_test_config functions)
    pub fn create_test_node_config() -> crate::datafold_node::config::NodeConfig {
        let dir = tempfile::tempdir().unwrap();
        let path = dir.path().to_path_buf();
        crate::datafold_node::config::NodeConfig {
            database: crate::datafold_node::config::DatabaseConfig::Local { path: path.clone() },
            default_trust_distance: 1,
            network_listen_address: "/ip4/127.0.0.1/tcp/0".to_string(),
            security_config: crate::security::SecurityConfig::default(),
            schema_service_url: None,
        }
    }

    /// Create named test tree (consolidates multiple create_test_tree functions)
    pub fn create_named_test_tree(tree_name: &str) -> Tree {
        let db = Self::create_temp_sled_db().expect("Failed to create test database");
        db.open_tree(tree_name).expect("Failed to create test tree")
    }
}