raphtory-graphql 0.17.0

Raphtory GraphQL server
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

pub mod app_config;
pub mod auth_config;
pub mod cache_config;
#[cfg(feature = "search")]
pub mod index_config;
pub mod log_config;
pub mod otlp_config;

#[cfg(test)]
mod tests {
    use crate::config::{
        app_config::{load_config, AppConfigBuilder},
        otlp_config::TracingLevel,
    };
    use std::{fs, path::PathBuf};

    #[test]
    fn test_load_config_from_toml() {
        let config_toml = r#"
            [logging]
            log_level = "DEBUG"

            [tracing]
            tracing_enabled = true
            tracing_level = "Essential"

            [cache]
            tti_seconds = 1000

            [auth]
            public_key = "MCowBQYDK2VwAyEADdrWr1kTLj+wSHlr45eneXmOjlHo3N1DjLIvDa2ozno="
        "#;
        let config_path = PathBuf::from("test_config.toml");
        fs::write(&config_path, config_toml).unwrap();

        let result = load_config(None, Some(config_path.clone()));

        let expected_config = AppConfigBuilder::new()
            .with_log_level("DEBUG".to_string())
            .with_tracing(true)
            .with_tracing_level(TracingLevel::ESSENTIAL)
            .with_cache_capacity(30)
            .with_cache_tti_seconds(1000)
            .with_auth_public_key(Some(
                "MCowBQYDK2VwAyEADdrWr1kTLj+wSHlr45eneXmOjlHo3N1DjLIvDa2ozno=".to_owned(),
            ))
            .unwrap()
            .build();

        assert_eq!(result.unwrap(), expected_config);

        // Cleanup: delete the test TOML file
        fs::remove_file(config_path).unwrap();
    }

    #[test]
    fn test_load_config_with_custom_cache() {
        let app_config = AppConfigBuilder::new()
            .with_cache_capacity(50)
            .with_cache_tti_seconds(1200)
            .build();

        let result = load_config(Some(app_config.clone()), None);

        assert_eq!(result.unwrap(), app_config);
    }
}