Arkasha
a3ee003947
1. Module Cycles: properly format cycle paths as A → B → C → A 2. Repository layout: group by top-level directory with file counts 3. Integration detection: match patterns against import names (substring), add Storage and AI/ML categories to all templates and summary 4. Usage examples: extract __init__ required params for class constructors Also fix golden test to use ends_with for module-prefixed symbol IDs.
89 lines
2.9 KiB
Rust
89 lines
2.9 KiB
Rust
//! Tests for the renderer functionality
|
|
|
|
use archdoc_core::{
|
|
model::{ProjectModel, Symbol, SymbolKind, IntegrationFlags, SymbolMetrics},
|
|
renderer::Renderer,
|
|
};
|
|
use std::collections::HashMap;
|
|
|
|
#[test]
|
|
fn test_render_with_integrations() {
|
|
// Create a mock project model with integration information
|
|
let mut project_model = ProjectModel::new();
|
|
|
|
// Add a symbol with database integration
|
|
let db_symbol = Symbol {
|
|
id: "DatabaseManager".to_string(),
|
|
kind: SymbolKind::Class,
|
|
module_id: "test_module".to_string(),
|
|
file_id: "test_file.py".to_string(),
|
|
qualname: "DatabaseManager".to_string(),
|
|
signature: "class DatabaseManager".to_string(),
|
|
annotations: None,
|
|
docstring_first_line: None,
|
|
purpose: "test".to_string(),
|
|
outbound_calls: vec![],
|
|
inbound_calls: vec![],
|
|
integrations_flags: IntegrationFlags {
|
|
db: true,
|
|
http: false,
|
|
queue: false,
|
|
storage: false,
|
|
ai: false,
|
|
},
|
|
metrics: SymbolMetrics {
|
|
fan_in: 0,
|
|
fan_out: 0,
|
|
is_critical: false,
|
|
cycle_participant: false,
|
|
},
|
|
};
|
|
|
|
// Add a symbol with HTTP integration
|
|
let http_symbol = Symbol {
|
|
id: "fetch_data".to_string(),
|
|
kind: SymbolKind::Function,
|
|
module_id: "test_module".to_string(),
|
|
file_id: "test_file.py".to_string(),
|
|
qualname: "fetch_data".to_string(),
|
|
signature: "def fetch_data()".to_string(),
|
|
annotations: None,
|
|
docstring_first_line: None,
|
|
purpose: "test".to_string(),
|
|
outbound_calls: vec![],
|
|
inbound_calls: vec![],
|
|
integrations_flags: IntegrationFlags {
|
|
db: false,
|
|
http: true,
|
|
queue: false,
|
|
storage: false,
|
|
ai: false,
|
|
},
|
|
metrics: SymbolMetrics {
|
|
fan_in: 0,
|
|
fan_out: 0,
|
|
is_critical: false,
|
|
cycle_participant: false,
|
|
},
|
|
};
|
|
|
|
project_model.symbols.insert("DatabaseManager".to_string(), db_symbol);
|
|
project_model.symbols.insert("fetch_data".to_string(), http_symbol);
|
|
|
|
// Initialize renderer
|
|
let renderer = Renderer::new();
|
|
|
|
// Render architecture documentation
|
|
let result = renderer.render_architecture_md(&project_model, None);
|
|
assert!(result.is_ok());
|
|
|
|
let rendered_content = result.unwrap();
|
|
println!("Rendered content:\n{}", rendered_content);
|
|
|
|
// Check that integration sections are present
|
|
assert!(rendered_content.contains("## Integrations"));
|
|
assert!(rendered_content.contains("### Database Integrations"));
|
|
assert!(rendered_content.contains("### HTTP/API Integrations"));
|
|
assert!(rendered_content.contains("DatabaseManager in test_file.py"));
|
|
assert!(rendered_content.contains("fetch_data in test_file.py"));
|
|
} |