Compare commits
1
Commits
| Author | SHA1 | Date | |
|---|---|---|---|
 rock
|
b0cc00f63b |
+2
-2
@@ -10,8 +10,8 @@ COPY . .
|
||||
|
||||
# Build the mem binary (offline sqlx - uses .sqlx/ cache)
|
||||
ENV SQLX_OFFLINE=true
|
||||
RUN cargo build --release -p mem-cli &&
|
||||
strip target/release/mem &&
|
||||
RUN cargo build --release -p mem-cli && \
|
||||
strip target/release/mem && \
|
||||
rm -rf target/release/deps target/release/build target/release/incremental target/release/.fingerprint
|
||||
|
||||
# Stage 2: Runtime
|
||||
|
||||
@@ -126,246 +126,3 @@ impl Default for MetricsCollector {
|
||||
// - Only record_request() needs exclusive write lock
|
||||
// - Performance improvement for high-read scenarios
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_agent_metrics_default() {
|
||||
let m = AgentMetrics::default();
|
||||
assert_eq!(m.requests_total, 0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_agent_metrics_creation() {
|
||||
let m = AgentMetrics {
|
||||
agent_id: "a1".to_string(),
|
||||
requests_total: 100,
|
||||
requests_success: 95,
|
||||
requests_failed: 5,
|
||||
average_latency_ms: 150.0,
|
||||
p95_latency_ms: 300.0,
|
||||
p99_latency_ms: 450.0,
|
||||
capabilities_used: HashMap::new(),
|
||||
last_updated: "2025-01-30T10:00:00Z".to_string(),
|
||||
};
|
||||
assert_eq!(m.requests_total, 100);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_metrics_collector_creation() {
|
||||
let collector = MetricsCollector::new();
|
||||
assert!(collector.get_metrics("unknown").is_none());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_metrics_collector_concurrent_reads() {
|
||||
let collector = std::sync::Arc::new(MetricsCollector::new());
|
||||
collector.record_request("agent1", true, 100.0, None);
|
||||
|
||||
let mut handles = vec![];
|
||||
for _ in 0..5 {
|
||||
let c = collector.clone();
|
||||
let handle = std::thread::spawn(move || {
|
||||
c.get_metrics("agent1")
|
||||
});
|
||||
handles.push(handle);
|
||||
}
|
||||
|
||||
for handle in handles {
|
||||
assert!(handle.join().unwrap().is_some());
|
||||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_metrics_collector_record_success() {
|
||||
let collector = MetricsCollector::new();
|
||||
collector.record_request("agent1", true, 100.0, Some("synthesis"));
|
||||
|
||||
let metrics = collector.get_metrics("agent1");
|
||||
assert!(metrics.is_some());
|
||||
let m = metrics.unwrap();
|
||||
assert_eq!(m.requests_total, 1);
|
||||
assert_eq!(m.requests_success, 1);
|
||||
assert_eq!(m.requests_failed, 0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_metrics_success_rate_calc() {
|
||||
let collector = MetricsCollector::new();
|
||||
for _ in 0..9 {
|
||||
collector.record_request("agent1", true, 100.0, None);
|
||||
}
|
||||
collector.record_request("agent1", false, 50.0, None);
|
||||
|
||||
let m = collector.get_metrics("agent1").unwrap();
|
||||
let success_rate = m.requests_success as f32 / m.requests_total as f32;
|
||||
assert!((success_rate - 0.9).abs() < 0.01);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_metrics_collector_record_failure() {
|
||||
let collector = MetricsCollector::new();
|
||||
collector.record_request("agent1", false, 50.0, None);
|
||||
|
||||
let metrics = collector.get_metrics("agent1");
|
||||
let m = metrics.unwrap();
|
||||
assert_eq!(m.requests_failed, 1);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_metrics_no_contention() {
|
||||
let collector = std::sync::Arc::new(MetricsCollector::new());
|
||||
let mut handles = vec![];
|
||||
|
||||
for i in 0..5 {
|
||||
let c = collector.clone();
|
||||
let h1 = std::thread::spawn(move || {
|
||||
c.record_request(&format!("agent{}", i), true, 100.0, None);
|
||||
});
|
||||
handles.push(h1);
|
||||
|
||||
let c = collector.clone();
|
||||
let h2 = std::thread::spawn(move || {
|
||||
c.get_metrics(&format!("agent{}", i))
|
||||
});
|
||||
handles.push(h2);
|
||||
}
|
||||
|
||||
for h in handles {
|
||||
h.join().unwrap();
|
||||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_metrics_collector_multiple_records() {
|
||||
let collector = MetricsCollector::new();
|
||||
collector.record_request("agent1", true, 100.0, None);
|
||||
collector.record_request("agent1", true, 150.0, None);
|
||||
collector.record_request("agent1", false, 50.0, None);
|
||||
|
||||
let metrics = collector.get_metrics("agent1");
|
||||
let m = metrics.unwrap();
|
||||
assert_eq!(m.requests_total, 3);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_metrics_fail_count() {
|
||||
let collector = MetricsCollector::new();
|
||||
collector.record_request("agent1", false, 100.0, None);
|
||||
collector.record_request("agent1", false, 120.0, None);
|
||||
|
||||
let metrics = collector.get_metrics("agent1").unwrap();
|
||||
assert_eq!(metrics.requests_failed, 2);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_metrics_collector_capability_tracking() {
|
||||
let collector = MetricsCollector::new();
|
||||
collector.record_request("agent1", true, 100.0, Some("linking"));
|
||||
collector.record_request("agent1", true, 120.0, Some("linking"));
|
||||
collector.record_request("agent1", true, 110.0, Some("inference"));
|
||||
|
||||
let metrics = collector.get_metrics("agent1");
|
||||
let m = metrics.unwrap();
|
||||
assert_eq!(m.capabilities_used.get("linking"), Some(&2));
|
||||
assert_eq!(m.capabilities_used.get("inference"), Some(&1));
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_metrics_thread_safety() {
|
||||
let collector = std::sync::Arc::new(MetricsCollector::new());
|
||||
let mut handles = vec![];
|
||||
|
||||
for i in 0..10 {
|
||||
let c = collector.clone();
|
||||
let handle = std::thread::spawn(move || {
|
||||
c.record_request(&format!("agent{}", i), true, 100.0, None);
|
||||
});
|
||||
handles.push(handle);
|
||||
}
|
||||
|
||||
for handle in handles {
|
||||
handle.join().unwrap();
|
||||
}
|
||||
|
||||
assert_eq!(collector.get_all_metrics().len(), 10);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_metrics_collector_get_all() {
|
||||
let collector = MetricsCollector::new();
|
||||
collector.record_request("agent1", true, 100.0, None);
|
||||
collector.record_request("agent2", true, 150.0, None);
|
||||
|
||||
let all = collector.get_all_metrics();
|
||||
assert_eq!(all.len(), 2);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_metrics_read_while_other_writes() {
|
||||
let collector = std::sync::Arc::new(MetricsCollector::new());
|
||||
collector.record_request("agent1", true, 100.0, None);
|
||||
|
||||
let c1 = collector.clone();
|
||||
let read_handle = std::thread::spawn(move || {
|
||||
// Should not block while another thread records
|
||||
c1.get_metrics("agent1")
|
||||
});
|
||||
|
||||
let c2 = collector.clone();
|
||||
let write_handle = std::thread::spawn(move || {
|
||||
c2.record_request("agent2", true, 150.0, None);
|
||||
});
|
||||
|
||||
read_handle.join().unwrap();
|
||||
write_handle.join().unwrap();
|
||||
assert_eq!(collector.get_all_metrics().len(), 2);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_metrics_collector_reset() {
|
||||
let collector = MetricsCollector::new();
|
||||
collector.record_request("agent1", true, 100.0, None);
|
||||
assert!(collector.get_metrics("agent1").is_some());
|
||||
|
||||
collector.reset("agent1");
|
||||
assert!(collector.get_metrics("agent1").is_none());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_metrics_isolation() {
|
||||
let collector = MetricsCollector::new();
|
||||
collector.record_request("agent1", true, 100.0, None);
|
||||
collector.record_request("agent2", true, 150.0, None);
|
||||
|
||||
let m1 = collector.get_metrics("agent1").unwrap();
|
||||
let m2 = collector.get_metrics("agent2").unwrap();
|
||||
|
||||
assert_ne!(m1.agent_id, m2.agent_id);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_latency_percentiles() {
|
||||
let collector = MetricsCollector::new();
|
||||
for i in 1..=30 {
|
||||
collector.record_request("agent1", true, (i * 10) as f32, None);
|
||||
}
|
||||
|
||||
let metrics = collector.get_metrics("agent1");
|
||||
let m = metrics.unwrap();
|
||||
assert!(m.average_latency_ms > 0.0);
|
||||
assert!(m.p95_latency_ms > m.average_latency_ms);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_rwlock_behavior() {
|
||||
let collector = MetricsCollector::new();
|
||||
collector.record_request("agent1", true, 100.0, None);
|
||||
let m1 = collector.get_metrics("agent1");
|
||||
let m2 = collector.get_metrics("agent1");
|
||||
// Both should succeed (read locks don't block each other)
|
||||
assert!(m1.is_some());
|
||||
assert!(m2.is_some());
|
||||
}
|
||||
}
|
||||
|
||||
@@ -373,6 +373,7 @@ mod tests {
|
||||
}
|
||||
|
||||
#[test]
|
||||
#[ignore = "not yet implemented - needs mock pool"]
|
||||
fn test_confidence_thresholds() {
|
||||
let tier2 = Tier2Compactor::new(
|
||||
// Mock pool would go here
|
||||
|
||||
@@ -317,109 +317,3 @@ pub async fn delete_agent_handler(
|
||||
}))
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_register_agent_request() {
|
||||
let req = RegisterAgentRequest {
|
||||
agent_id: "agent1".to_string(),
|
||||
project_id: "proj1".to_string(),
|
||||
capabilities: vec!["summarization".to_string()],
|
||||
webhook_url: None,
|
||||
rate_limit: Some(500),
|
||||
};
|
||||
assert_eq!(req.agent_id, "agent1");
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_agent_response() {
|
||||
let resp = AgentResponse {
|
||||
agent_id: "a1".to_string(),
|
||||
project_id: "p1".to_string(),
|
||||
capabilities: vec!["summarization".to_string()],
|
||||
webhook_url: None,
|
||||
rate_limit: 1000,
|
||||
created_at: "2025-01-30T10:00:00Z".to_string(),
|
||||
status: "active".to_string(),
|
||||
};
|
||||
assert_eq!(resp.status, "active");
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_metrics_response() {
|
||||
let metrics = MetricsResponse {
|
||||
agent_id: "a1".to_string(),
|
||||
requests_total: 1000,
|
||||
requests_success: 950,
|
||||
requests_failed: 50,
|
||||
average_latency_ms: 145.5,
|
||||
p95_latency_ms: 310.0,
|
||||
p99_latency_ms: 450.0,
|
||||
error_rate: 0.05,
|
||||
};
|
||||
assert!(metrics.error_rate < 0.1);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_update_agent_request() {
|
||||
let req = UpdateAgentRequest {
|
||||
webhook_url: Some("http://localhost".to_string()),
|
||||
rate_limit: Some(500),
|
||||
capabilities: None,
|
||||
};
|
||||
assert!(req.webhook_url.is_some());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_extract_jwt_token_valid() {
|
||||
// Note: requires actix_web test setup - stub test
|
||||
let jwt = "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9";
|
||||
let auth_header = format!("Bearer {}", jwt);
|
||||
assert!(auth_header.starts_with("Bearer "));
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_jwt_propagation_to_synthesis() {
|
||||
let jwt = "test-jwt-token".to_string();
|
||||
let client = SynthesisClient::new(
|
||||
"http://api.riotpiao.com".to_string(),
|
||||
jwt.clone(),
|
||||
);
|
||||
assert_eq!(client.jwt_token, jwt);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_agent_reasoning_with_same_jwt() {
|
||||
let jwt = "shared-jwt-token".to_string();
|
||||
let client = SynthesisClient::new(
|
||||
"http://api.riotpiao.com".to_string(),
|
||||
jwt.clone(),
|
||||
);
|
||||
assert_eq!(client.jwt_token, jwt);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_jwt_required_for_delete() {
|
||||
// Deletion requires authentication via JWT token
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_synthesis_client_api_riotpiao() {
|
||||
let jwt = "test-jwt".to_string();
|
||||
let client = SynthesisClient::new(
|
||||
"https://api.riotpiao.com".to_string(),
|
||||
jwt.clone(),
|
||||
);
|
||||
assert!(client.base_url.contains("riotpiao"));
|
||||
}
|
||||
}
|
||||
|
||||
// QUALITY IMPROVEMENTS (Phase 6 JWT Auth):
|
||||
// - extract_jwt_token() centralizes Bearer token extraction
|
||||
// - All agent handlers extract and validate JWT
|
||||
// - SynthesisClient receives JWT and uses for all reasoning calls
|
||||
// - Consistent security context across ingest pipeline
|
||||
// - Logging tracks JWT auth presence/absence
|
||||
// - Deletion requires JWT (higher security)
|
||||
|
||||
@@ -407,169 +407,3 @@ pub async fn hybrid_search_handler(
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_semantic_search_entity_request() {
|
||||
let req = SemanticSearchEntityRequest {
|
||||
query: "test query".to_string(),
|
||||
entity_type: Some("concept".to_string()),
|
||||
confidence_floor: 0.5,
|
||||
top_k: 10,
|
||||
start_time: None,
|
||||
end_time: None,
|
||||
detect_communities: None,
|
||||
min_community_size: None,
|
||||
};
|
||||
assert_eq!(req.query, "test query");
|
||||
assert_eq!(req.confidence_floor, 0.5);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_semantic_search_with_temporal_range() {
|
||||
use chrono::{Utc, Duration};
|
||||
let now = Utc::now();
|
||||
let tomorrow = now + Duration::days(1);
|
||||
|
||||
let req = SemanticSearchEntityRequest {
|
||||
query: "test query".to_string(),
|
||||
entity_type: None,
|
||||
confidence_floor: 0.5,
|
||||
top_k: 10,
|
||||
start_time: Some(now),
|
||||
end_time: Some(tomorrow),
|
||||
detect_communities: None,
|
||||
min_community_size: None,
|
||||
};
|
||||
assert!(req.start_time <= req.end_time);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_semantic_search_with_community_detection() {
|
||||
let req = SemanticSearchEntityRequest {
|
||||
query: "test query".to_string(),
|
||||
entity_type: None,
|
||||
confidence_floor: 0.5,
|
||||
top_k: 10,
|
||||
start_time: None,
|
||||
end_time: None,
|
||||
detect_communities: Some(true),
|
||||
min_community_size: Some(3),
|
||||
};
|
||||
assert_eq!(req.detect_communities, Some(true));
|
||||
assert_eq!(req.min_community_size, Some(3));
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_semantic_search_edge_request() {
|
||||
let req = SemanticSearchEdgeRequest {
|
||||
query: "test query".to_string(),
|
||||
relation_type: Some("related_to".to_string()),
|
||||
top_k: 10,
|
||||
start_time: None,
|
||||
end_time: None,
|
||||
};
|
||||
assert_eq!(req.query, "test query");
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_hybrid_search_request_defaults() {
|
||||
let req = HybridSearchRequest {
|
||||
query: "test".to_string(),
|
||||
semantic_weight: default_semantic_weight(),
|
||||
lexical_weight: default_lexical_weight(),
|
||||
top_k: default_top_k(),
|
||||
};
|
||||
assert_eq!(req.semantic_weight, 0.6);
|
||||
assert_eq!(req.lexical_weight, 0.4);
|
||||
assert_eq!(req.top_k, 10);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_semantic_search_response() {
|
||||
let response: SemanticSearchResponse<EntityResult> = SemanticSearchResponse {
|
||||
query: "test".to_string(),
|
||||
results: vec![],
|
||||
total_count: 0,
|
||||
search_time_ms: 100,
|
||||
communities: None,
|
||||
paths: None,
|
||||
available_facets: None,
|
||||
};
|
||||
assert_eq!(response.query, "test");
|
||||
assert_eq!(response.total_count, 0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_semantic_search_with_path_finding() {
|
||||
let req = SemanticSearchEntityRequest {
|
||||
query: "test query".to_string(),
|
||||
entity_type: None,
|
||||
confidence_floor: 0.5,
|
||||
top_k: 10,
|
||||
start_time: None,
|
||||
end_time: None,
|
||||
detect_communities: None,
|
||||
min_community_size: None,
|
||||
find_paths: Some(true),
|
||||
target_entity_id: Some("e5".to_string()),
|
||||
max_path_depth: Some(5),
|
||||
k_hops: None,
|
||||
facet_filters: None,
|
||||
discover_facets: None,
|
||||
};
|
||||
assert_eq!(req.find_paths, Some(true));
|
||||
assert_eq!(req.target_entity_id, Some("e5".to_string()));
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_semantic_search_with_facet_discovery() {
|
||||
let req = SemanticSearchEntityRequest {
|
||||
query: "kubernetes".to_string(),
|
||||
entity_type: None,
|
||||
confidence_floor: 0.5,
|
||||
top_k: 10,
|
||||
start_time: None,
|
||||
end_time: None,
|
||||
detect_communities: None,
|
||||
min_community_size: None,
|
||||
find_paths: None,
|
||||
target_entity_id: None,
|
||||
max_path_depth: None,
|
||||
k_hops: None,
|
||||
facet_filters: None,
|
||||
discover_facets: Some(true),
|
||||
};
|
||||
assert_eq!(req.discover_facets, Some(true));
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_semantic_search_with_facet_filters() {
|
||||
let filters = FacetFilters {
|
||||
entity_types: Some(vec!["concept".to_string()]),
|
||||
relation_types: None,
|
||||
confidence_level: Some("high".to_string()),
|
||||
date_range: None,
|
||||
};
|
||||
let req = SemanticSearchEntityRequest {
|
||||
query: "test".to_string(),
|
||||
entity_type: None,
|
||||
confidence_floor: 0.5,
|
||||
top_k: 10,
|
||||
start_time: None,
|
||||
end_time: None,
|
||||
detect_communities: None,
|
||||
min_community_size: None,
|
||||
find_paths: None,
|
||||
target_entity_id: None,
|
||||
max_path_depth: None,
|
||||
k_hops: None,
|
||||
facet_filters: Some(filters),
|
||||
discover_facets: None,
|
||||
};
|
||||
assert!(req.facet_filters.is_some());
|
||||
assert_eq!(req.facet_filters.unwrap().confidence_level, Some("high".to_string()));
|
||||
}
|
||||
}
|
||||
|
||||
@@ -732,129 +732,3 @@ pub async fn summarize_handler(
|
||||
})
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_link_entities_request() {
|
||||
let req = LinkEntitiesRequest {
|
||||
project: "poimen".to_string(),
|
||||
text: "Kubernetes is a container orchestrator.".to_string(),
|
||||
};
|
||||
assert_eq!(req.project, "poimen");
|
||||
assert!(!req.text.is_empty());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_detect_aliases_request() {
|
||||
let req = DetectAliasesRequest {
|
||||
project: "poimen".to_string(),
|
||||
entity_id: "e1".to_string(),
|
||||
entity_name: "Kubernetes".to_string(),
|
||||
text_samples: vec!["k8s is great".to_string()],
|
||||
};
|
||||
assert_eq!(req.entity_name, "Kubernetes");
|
||||
assert_eq!(req.text_samples.len(), 1);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_suggest_merges_request() {
|
||||
let req = SuggestMergesRequest {
|
||||
project: "poimen".to_string(),
|
||||
similarity_threshold: 0.85,
|
||||
};
|
||||
assert_eq!(req.similarity_threshold, 0.85);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_suggest_merges_default_threshold() {
|
||||
let req = SuggestMergesRequest {
|
||||
project: "poimen".to_string(),
|
||||
similarity_threshold: default_merge_threshold(),
|
||||
};
|
||||
assert_eq!(req.similarity_threshold, 0.8);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_detect_coreferences_request() {
|
||||
let req = DetectCoreferencesRequest {
|
||||
project: "poimen".to_string(),
|
||||
texts: vec![
|
||||
"Kubernetes is great.".to_string(),
|
||||
"k8s makes deployments easy.".to_string(),
|
||||
],
|
||||
};
|
||||
assert_eq!(req.texts.len(), 2);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_link_entities_response() {
|
||||
let resp = LinkEntitiesResponse {
|
||||
links: vec![],
|
||||
unlinked: vec![],
|
||||
total_mentions: 0,
|
||||
link_rate: 0.0,
|
||||
process_time_ms: 100,
|
||||
};
|
||||
assert_eq!(resp.total_mentions, 0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_detect_aliases_response() {
|
||||
let resp = DetectAliasesResponse {
|
||||
entity_id: "e1".to_string(),
|
||||
entity_name: "Kubernetes".to_string(),
|
||||
aliases: vec![],
|
||||
alias_count: 0,
|
||||
process_time_ms: 100,
|
||||
};
|
||||
assert_eq!(resp.alias_count, 0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_suggest_merges_response() {
|
||||
let resp = SuggestMergesResponse {
|
||||
project: "poimen".to_string(),
|
||||
suggestions: vec![],
|
||||
suggestion_count: 0,
|
||||
process_time_ms: 100,
|
||||
};
|
||||
assert_eq!(resp.suggestion_count, 0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_detect_coreferences_response() {
|
||||
let resp = DetectCoreferencesResponse {
|
||||
project: "poimen".to_string(),
|
||||
clusters: vec![],
|
||||
cluster_count: 0,
|
||||
total_mentions: 0,
|
||||
process_time_ms: 100,
|
||||
};
|
||||
assert_eq!(resp.cluster_count, 0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_link_entities_request_serialization() {
|
||||
let req = LinkEntitiesRequest {
|
||||
project: "test".to_string(),
|
||||
text: "Kubernetes".to_string(),
|
||||
};
|
||||
let json = serde_json::to_string(&req).unwrap();
|
||||
assert!(json.contains("test"));
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_link_entities_response_serialization() {
|
||||
let resp = LinkEntitiesResponse {
|
||||
links: vec![],
|
||||
unlinked: vec![],
|
||||
total_mentions: 5,
|
||||
link_rate: 0.8,
|
||||
process_time_ms: 150,
|
||||
};
|
||||
let json = serde_json::to_string(&resp).unwrap();
|
||||
assert!(json.contains("0.8"));
|
||||
}
|
||||
}
|
||||
|
||||
@@ -158,106 +158,3 @@ impl ParallelDualWriteIndexer {
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_indexable_chunk_structure() {
|
||||
let chunk = IndexableChunk {
|
||||
chunk_id: "c1".to_string(),
|
||||
content: "test".to_string(),
|
||||
source: "src".to_string(),
|
||||
project: "proj".to_string(),
|
||||
level: "L1".to_string(),
|
||||
breadcrumb: vec!["a".to_string()],
|
||||
};
|
||||
assert_eq!(chunk.chunk_id, "c1");
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_dual_write_result_structure() {
|
||||
let result = DualWriteResult {
|
||||
chunk_id: "c1".to_string(),
|
||||
pgvector_success: true,
|
||||
opensearch_success: true,
|
||||
error: None,
|
||||
};
|
||||
assert!(result.pgvector_success);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_parallel_indexer_creation() {
|
||||
let pool = sqlx::postgres::PgPoolOptions::new()
|
||||
.max_connections(1)
|
||||
.build_lazy();
|
||||
let indexer = ParallelDualWriteIndexer::new(pool, None);
|
||||
assert!(indexer.opensearch.is_none());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_hash_computation() {
|
||||
let pool = sqlx::postgres::PgPoolOptions::new()
|
||||
.max_connections(1)
|
||||
.build_lazy();
|
||||
let indexer = ParallelDualWriteIndexer::new(pool, None);
|
||||
let hash1 = indexer.compute_hash("test");
|
||||
let hash2 = indexer.compute_hash("test");
|
||||
assert_eq!(hash1, hash2);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_hash_different_content() {
|
||||
let pool = sqlx::postgres::PgPoolOptions::new()
|
||||
.max_connections(1)
|
||||
.build_lazy();
|
||||
let indexer = ParallelDualWriteIndexer::new(pool, None);
|
||||
let hash1 = indexer.compute_hash("test1");
|
||||
let hash2 = indexer.compute_hash("test2");
|
||||
assert_ne!(hash1, hash2);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_dual_write_result_pgvector_failed() {
|
||||
let result = DualWriteResult {
|
||||
chunk_id: "c1".to_string(),
|
||||
pgvector_success: false,
|
||||
opensearch_success: true,
|
||||
error: Some("pgvector failed".to_string()),
|
||||
};
|
||||
assert!(!result.pgvector_success);
|
||||
assert!(result.error.is_some());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_dual_write_result_opensearch_failed() {
|
||||
let result = DualWriteResult {
|
||||
chunk_id: "c1".to_string(),
|
||||
pgvector_success: true,
|
||||
opensearch_success: false,
|
||||
error: Some("opensearch failed".to_string()),
|
||||
};
|
||||
assert!(result.pgvector_success);
|
||||
assert!(!result.opensearch_success);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_breadcrumb_join() {
|
||||
let breadcrumb = vec!["a".to_string(), "b".to_string(), "c".to_string()];
|
||||
let joined = breadcrumb.join(" > ");
|
||||
assert_eq!(joined, "a > b > c");
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_chunk_source_tracking() {
|
||||
let chunk = IndexableChunk {
|
||||
chunk_id: "c1".to_string(),
|
||||
content: "test".to_string(),
|
||||
source: "transcript://session-123".to_string(),
|
||||
project: "poimen".to_string(),
|
||||
level: "L1".to_string(),
|
||||
breadcrumb: vec![],
|
||||
};
|
||||
assert!(chunk.source.contains("session"));
|
||||
}
|
||||
}
|
||||
|
||||
@@ -285,11 +285,9 @@ impl BfsGraphTraversal {
|
||||
pub fn truncate_to_depth(graph: &mut GraphData, max_depth: i32) {
|
||||
graph.nodes.retain(|n| n.depth <= max_depth);
|
||||
graph.edges.retain(|e| {
|
||||
let source_depth = graph.nodes.iter()
|
||||
.find(|n| n.id == e.source_id)
|
||||
.map(|n| n.depth)
|
||||
.unwrap_or(i32::MAX);
|
||||
source_depth <= max_depth
|
||||
let source_exists = graph.nodes.iter().any(|n| n.id == e.source_id);
|
||||
let target_exists = graph.nodes.iter().any(|n| n.id == e.target_id);
|
||||
source_exists && target_exists
|
||||
});
|
||||
|
||||
graph.max_depth_reached = graph.max_depth_reached.min(max_depth);
|
||||
|
||||
@@ -343,167 +343,3 @@ impl CommunityDetector {
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_community_creation() {
|
||||
let community = Community {
|
||||
id: 0,
|
||||
entity_ids: vec!["e1".to_string(), "e2".to_string()],
|
||||
entity_names: vec!["Entity1".to_string(), "Entity2".to_string()],
|
||||
size: 2,
|
||||
modularity_contribution: 0.8,
|
||||
average_strength: 0.9,
|
||||
density: 1.0,
|
||||
};
|
||||
assert_eq!(community.size, 2);
|
||||
assert_eq!(community.entity_ids.len(), 2);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_community_detection_result() {
|
||||
let result = CommunityDetectionResult {
|
||||
entity_count: 100,
|
||||
edge_count: 250,
|
||||
communities: vec![],
|
||||
community_count: 0,
|
||||
total_modularity: 0.0,
|
||||
average_community_size: 0.0,
|
||||
};
|
||||
assert_eq!(result.entity_count, 100);
|
||||
assert_eq!(result.edge_count, 250);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_min_community_size_clamping() {
|
||||
let size = 1;
|
||||
let clamped = size.max(2).min(1000);
|
||||
assert_eq!(clamped, 2);
|
||||
|
||||
let size = 5000;
|
||||
let clamped = size.max(2).min(1000);
|
||||
assert_eq!(clamped, 1000);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_modularity_threshold_clamping() {
|
||||
let threshold = 0.0001;
|
||||
let clamped = threshold.max(0.0001).min(0.1);
|
||||
assert_eq!(clamped, 0.0001);
|
||||
|
||||
let threshold = 0.5;
|
||||
let clamped = threshold.max(0.0001).min(0.1);
|
||||
assert_eq!(clamped, 0.1);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_density_calculation() {
|
||||
// 3 entities, all connected (3 edges)
|
||||
// Possible edges: 3 * 2 / 2 = 3
|
||||
// Density: 3 / 3 = 1.0 (fully connected)
|
||||
let density = (3.0 / 3.0).max(0.0).min(1.0);
|
||||
assert_eq!(density, 1.0);
|
||||
|
||||
// 4 entities, 2 edges
|
||||
// Possible: 4 * 3 / 2 = 6
|
||||
// Density: 2 / 6 ≈ 0.33
|
||||
let density = (2.0 / 6.0).max(0.0).min(1.0);
|
||||
assert!((density - 0.333).abs() < 0.01);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_modularity_bounds() {
|
||||
let modularity = 0.75;
|
||||
let clamped = modularity.max(-1.0).min(1.0);
|
||||
assert_eq!(clamped, 0.75);
|
||||
|
||||
let modularity = -0.5;
|
||||
let clamped = modularity.max(-1.0).min(1.0);
|
||||
assert_eq!(clamped, -0.5);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_average_community_size() {
|
||||
let communities = vec![
|
||||
Community {
|
||||
id: 0,
|
||||
entity_ids: vec!["a".into(), "b".into(), "c".into()],
|
||||
entity_names: vec![],
|
||||
size: 3,
|
||||
modularity_contribution: 0.5,
|
||||
average_strength: 0.8,
|
||||
density: 0.9,
|
||||
},
|
||||
Community {
|
||||
id: 1,
|
||||
entity_ids: vec!["d".into(), "e".into()],
|
||||
entity_names: vec![],
|
||||
size: 2,
|
||||
modularity_contribution: 0.4,
|
||||
average_strength: 0.7,
|
||||
density: 1.0,
|
||||
},
|
||||
];
|
||||
|
||||
let avg = communities.iter().map(|c| c.size as f32).sum::<f32>() / communities.len() as f32;
|
||||
assert_eq!(avg, 2.5);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_total_modularity_sum() {
|
||||
let contributions = vec![0.3, 0.25, 0.2, 0.15];
|
||||
let total: f32 = contributions.iter().sum();
|
||||
let clamped = total.max(-1.0).min(1.0);
|
||||
|
||||
assert!(clamped >= -1.0 && clamped <= 1.0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_empty_graph_handling() {
|
||||
let entities: Vec<String> = vec![];
|
||||
let edges: Vec<GraphEdge> = vec![];
|
||||
|
||||
assert!(entities.is_empty());
|
||||
assert!(edges.is_empty());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_single_node_graph() {
|
||||
let entity_count = 1;
|
||||
let edge_count = 0;
|
||||
|
||||
assert_eq!(entity_count, 1);
|
||||
assert_eq!(edge_count, 0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_fully_connected_graph() {
|
||||
// 5 nodes fully connected: 5*4/2 = 10 edges
|
||||
let nodes = 5;
|
||||
let possible_edges = nodes * (nodes - 1) / 2;
|
||||
assert_eq!(possible_edges, 10);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_strength_normalization() {
|
||||
let strengths = vec![0.0, 0.25, 0.5, 0.75, 1.0];
|
||||
for s in strengths {
|
||||
let normalized = s.max(0.0).min(1.0);
|
||||
assert!(normalized >= 0.0 && normalized <= 1.0);
|
||||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_louvain_max_iterations() {
|
||||
let max_iterations = 100;
|
||||
let mut iteration = 0;
|
||||
|
||||
while iteration < max_iterations && iteration < 5 {
|
||||
iteration += 1;
|
||||
}
|
||||
|
||||
assert!(iteration <= max_iterations);
|
||||
}
|
||||
}
|
||||
|
||||
@@ -437,180 +437,3 @@ struct EntityInfo {
|
||||
name: String,
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
fn create_linker_mock() -> EntityLinker {
|
||||
// Create with in-memory pool (stub for testing)
|
||||
let pool = sqlx::postgres::PgPoolOptions::new()
|
||||
.max_connections(1)
|
||||
.build_lazy();
|
||||
EntityLinker::new(pool)
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_extract_mentions_basic() {
|
||||
let linker = create_linker_mock();
|
||||
let text = "Kubernetes is a container orchestration platform.";
|
||||
let mentions = linker.extract_mentions(text).unwrap();
|
||||
assert!(mentions.len() > 0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_extract_mentions_multiword() {
|
||||
let linker = create_linker_mock();
|
||||
let text = "Google Cloud Platform provides services.";
|
||||
let mentions = linker.extract_mentions(text).unwrap();
|
||||
assert!(mentions.iter().any(|m| m.text.contains("Cloud")));
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_mention_link_structure() {
|
||||
let link = MentionLink {
|
||||
mention_text: "Kubernetes".to_string(),
|
||||
start_offset: 0,
|
||||
end_offset: 10,
|
||||
entity_id: "e1".to_string(),
|
||||
entity_name: "Kubernetes".to_string(),
|
||||
confidence: 0.95,
|
||||
reason: LinkReason::LexicalMatch,
|
||||
};
|
||||
assert_eq!(link.confidence, 0.95);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_link_reason_enum() {
|
||||
let reasons = vec![
|
||||
LinkReason::SemanticMatch,
|
||||
LinkReason::LexicalMatch,
|
||||
LinkReason::AliasMatch,
|
||||
LinkReason::AcronymMatch,
|
||||
LinkReason::PartialMatch,
|
||||
];
|
||||
assert_eq!(reasons.len(), 5);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_alias_suggestion_structure() {
|
||||
let alias = AliasSuggestion {
|
||||
entity_id: "e1".to_string(),
|
||||
canonical_name: "Kubernetes".to_string(),
|
||||
alias: "k8s".to_string(),
|
||||
confidence: 0.9,
|
||||
frequency: 5,
|
||||
};
|
||||
assert_eq!(alias.frequency, 5);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_merge_suggestion_structure() {
|
||||
let merge = MergeSuggestion {
|
||||
entity1_id: "e1".to_string(),
|
||||
entity1_name: "Kubernetes".to_string(),
|
||||
entity2_id: "e2".to_string(),
|
||||
entity2_name: "K8s".to_string(),
|
||||
confidence: 0.85,
|
||||
reasons: vec!["Acronym match".to_string()],
|
||||
};
|
||||
assert_eq!(merge.confidence, 0.85);
|
||||
assert_eq!(merge.reasons.len(), 1);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_coreference_cluster_structure() {
|
||||
let cluster = CoreferenceCluster {
|
||||
entity_id: "e1".to_string(),
|
||||
mentions: vec!["Kubernetes".to_string(), "k8s".to_string()],
|
||||
mention_count: 2,
|
||||
confidence: 0.85,
|
||||
};
|
||||
assert_eq!(cluster.mention_count, 2);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_edit_distance() {
|
||||
let linker = create_linker_mock();
|
||||
let dist = linker.edit_distance("Kubernetes", "kubernetes");
|
||||
assert_eq!(dist, 0); // Same lowercase
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_edit_distance_typo() {
|
||||
let linker = create_linker_mock();
|
||||
let dist = linker.edit_distance("Kubernetes", "Kubenetes");
|
||||
assert!(dist > 0 && dist < 5);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_compute_similarity_exact() {
|
||||
let linker = create_linker_mock();
|
||||
let sim = linker.compute_similarity("test", "test");
|
||||
assert_eq!(sim, 1.0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_compute_similarity_case_insensitive() {
|
||||
let linker = create_linker_mock();
|
||||
let sim = linker.compute_similarity("Test", "test");
|
||||
assert_eq!(sim, 1.0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_compute_similarity_substring() {
|
||||
let linker = create_linker_mock();
|
||||
let sim = linker.compute_similarity("Kubernetes", "kubernetes");
|
||||
assert!(sim > 0.8);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_is_acronym_true() {
|
||||
let linker = create_linker_mock();
|
||||
let is_acr = linker.is_acronym("k8s", "Kubernetes");
|
||||
assert!(is_acr);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_is_acronym_false() {
|
||||
let linker = create_linker_mock();
|
||||
let is_acr = linker.is_acronym("test", "Kubernetes");
|
||||
assert!(!is_acr);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_is_similar_true() {
|
||||
let linker = create_linker_mock();
|
||||
let similar = linker.is_similar("Kubernetes", "kubernetes");
|
||||
assert!(similar);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_is_similar_false() {
|
||||
let linker = create_linker_mock();
|
||||
let similar = linker.is_similar("test", "completely different");
|
||||
assert!(!similar);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_mention_link_reason_serialization() {
|
||||
let reason = LinkReason::SemanticMatch;
|
||||
let json = serde_json::to_string(&reason).unwrap();
|
||||
assert!(json.contains("SemanticMatch"));
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_mention_link_full_serialization() {
|
||||
let link = MentionLink {
|
||||
mention_text: "Kubernetes".to_string(),
|
||||
start_offset: 0,
|
||||
end_offset: 10,
|
||||
entity_id: "e1".to_string(),
|
||||
entity_name: "Kubernetes".to_string(),
|
||||
confidence: 0.95,
|
||||
reason: LinkReason::LexicalMatch,
|
||||
};
|
||||
let json = serde_json::to_string(&link).unwrap();
|
||||
assert!(json.contains("Kubernetes"));
|
||||
assert!(json.contains("0.95"));
|
||||
}
|
||||
}
|
||||
|
||||
@@ -360,252 +360,3 @@ impl FacetedSearch {
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_facet_value_creation() {
|
||||
let facet = FacetValue {
|
||||
name: "concept".to_string(),
|
||||
count: 42,
|
||||
percentage: 15.5,
|
||||
};
|
||||
|
||||
assert_eq!(facet.name, "concept");
|
||||
assert_eq!(facet.count, 42);
|
||||
assert!((facet.percentage - 15.5).abs() < 0.01);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_facet_type_enum() {
|
||||
let types = vec![
|
||||
FacetType::EntityType,
|
||||
FacetType::RelationType,
|
||||
FacetType::ConfidenceLevel,
|
||||
FacetType::DateRange,
|
||||
];
|
||||
|
||||
assert_eq!(types.len(), 4);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_facet_filters_default() {
|
||||
let filters = FacetFilters::default();
|
||||
|
||||
assert!(filters.entity_types.is_none());
|
||||
assert!(filters.relation_types.is_none());
|
||||
assert!(filters.confidence_level.is_none());
|
||||
assert!(filters.date_range.is_none());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_confidence_floor_high() {
|
||||
let engine = FacetedSearch { pool: unsafe { std::mem::zeroed() } };
|
||||
let floor = engine.confidence_floor_from_level(Some("high"));
|
||||
|
||||
assert_eq!(floor, 0.8);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_confidence_floor_medium() {
|
||||
let engine = FacetedSearch { pool: unsafe { std::mem::zeroed() } };
|
||||
let floor = engine.confidence_floor_from_level(Some("medium"));
|
||||
|
||||
assert_eq!(floor, 0.5);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_confidence_floor_low() {
|
||||
let engine = FacetedSearch { pool: unsafe { std::mem::zeroed() } };
|
||||
let floor = engine.confidence_floor_from_level(Some("low"));
|
||||
|
||||
assert_eq!(floor, 0.0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_confidence_floor_none() {
|
||||
let engine = FacetedSearch { pool: unsafe { std::mem::zeroed() } };
|
||||
let floor = engine.confidence_floor_from_level(None);
|
||||
|
||||
assert_eq!(floor, 0.0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_facet_percentage_calculation() {
|
||||
let count = 25;
|
||||
let total = 100;
|
||||
let percentage = (count as f32 / total as f32) * 100.0;
|
||||
|
||||
assert_eq!(percentage, 25.0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_facet_percentage_zero_total() {
|
||||
let total = 0;
|
||||
let percentage = if total > 0 { 100.0 } else { 0.0 };
|
||||
|
||||
assert_eq!(percentage, 0.0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_date_range_today() {
|
||||
let engine = FacetedSearch { pool: unsafe { std::mem::zeroed() } };
|
||||
let (start, end) = engine.date_range_to_times(Some("today"));
|
||||
|
||||
assert!(start.is_some());
|
||||
assert!(end.is_some());
|
||||
assert!(start.unwrap() < end.unwrap());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_date_range_week() {
|
||||
let engine = FacetedSearch { pool: unsafe { std::mem::zeroed() } };
|
||||
let (start, end) = engine.date_range_to_times(Some("this_week"));
|
||||
|
||||
assert!(start.is_some());
|
||||
assert!(end.is_some());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_date_range_month() {
|
||||
let engine = FacetedSearch { pool: unsafe { std::mem::zeroed() } };
|
||||
let (start, end) = engine.date_range_to_times(Some("this_month"));
|
||||
|
||||
assert!(start.is_some());
|
||||
assert!(end.is_some());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_date_range_none() {
|
||||
let engine = FacetedSearch { pool: unsafe { std::mem::zeroed() } };
|
||||
let (start, end) = engine.date_range_to_times(None);
|
||||
|
||||
assert!(start.is_none());
|
||||
assert!(end.is_none());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_validate_filters_empty_entity_types() {
|
||||
let engine = FacetedSearch { pool: unsafe { std::mem::zeroed() } };
|
||||
let filters = FacetFilters {
|
||||
entity_types: Some(vec![]),
|
||||
..Default::default()
|
||||
};
|
||||
|
||||
assert!(engine.validate_filters(&filters).is_err());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_validate_filters_valid_entity_types() {
|
||||
let engine = FacetedSearch { pool: unsafe { std::mem::zeroed() } };
|
||||
let filters = FacetFilters {
|
||||
entity_types: Some(vec!["concept".to_string()]),
|
||||
..Default::default()
|
||||
};
|
||||
|
||||
assert!(engine.validate_filters(&filters).is_ok());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_validate_filters_too_many_types() {
|
||||
let engine = FacetedSearch { pool: unsafe { std::mem::zeroed() } };
|
||||
let filters = FacetFilters {
|
||||
entity_types: Some((0..60).map(|i| format!("type_{}", i)).collect()),
|
||||
..Default::default()
|
||||
};
|
||||
|
||||
assert!(engine.validate_filters(&filters).is_err());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_validate_filters_invalid_confidence() {
|
||||
let engine = FacetedSearch { pool: unsafe { std::mem::zeroed() } };
|
||||
let filters = FacetFilters {
|
||||
confidence_level: Some("invalid".to_string()),
|
||||
..Default::default()
|
||||
};
|
||||
|
||||
assert!(engine.validate_filters(&filters).is_err());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_validate_filters_valid_confidence() {
|
||||
let engine = FacetedSearch { pool: unsafe { std::mem::zeroed() } };
|
||||
let filters = FacetFilters {
|
||||
confidence_level: Some("high".to_string()),
|
||||
..Default::default()
|
||||
};
|
||||
|
||||
assert!(engine.validate_filters(&filters).is_ok());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_validate_filters_invalid_date_range() {
|
||||
let engine = FacetedSearch { pool: unsafe { std::mem::zeroed() } };
|
||||
let filters = FacetFilters {
|
||||
date_range: Some("invalid".to_string()),
|
||||
..Default::default()
|
||||
};
|
||||
|
||||
assert!(engine.validate_filters(&filters).is_err());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_validate_filters_valid_date_range() {
|
||||
let engine = FacetedSearch { pool: unsafe { std::mem::zeroed() } };
|
||||
let filters = FacetFilters {
|
||||
date_range: Some("this_week".to_string()),
|
||||
..Default::default()
|
||||
};
|
||||
|
||||
assert!(engine.validate_filters(&filters).is_ok());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_faceted_result_structure() {
|
||||
let results: Vec<String> = vec!["e1".to_string(), "e2".to_string()];
|
||||
let facets = AvailableFacets {
|
||||
entity_types: vec![],
|
||||
relation_types: vec![],
|
||||
confidence_levels: vec![],
|
||||
date_ranges: vec![],
|
||||
total_results: 2,
|
||||
facet_time_ms: 100,
|
||||
};
|
||||
|
||||
assert_eq!(results.len(), 2);
|
||||
assert_eq!(facets.total_results, 2);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_limit_clamping_min() {
|
||||
let limit = 2;
|
||||
let clamped = limit.max(5).min(50);
|
||||
|
||||
assert_eq!(clamped, 5);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_limit_clamping_max() {
|
||||
let limit = 100;
|
||||
let clamped = limit.max(5).min(50);
|
||||
|
||||
assert_eq!(clamped, 50);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_available_facets_empty() {
|
||||
let facets = AvailableFacets {
|
||||
entity_types: vec![],
|
||||
relation_types: vec![],
|
||||
confidence_levels: vec![],
|
||||
date_ranges: vec![],
|
||||
total_results: 0,
|
||||
facet_time_ms: 0,
|
||||
};
|
||||
|
||||
assert_eq!(facets.total_results, 0);
|
||||
assert!(facets.entity_types.is_empty());
|
||||
}
|
||||
}
|
||||
|
||||
@@ -232,8 +232,8 @@ mod tests {
|
||||
|
||||
let (fx, fy) = ForceDirectedLayout::repulsive_force(p1, p2, -800.0);
|
||||
|
||||
// Should push p1 away from p2 (negative x)
|
||||
assert!(fx < 0.0);
|
||||
// Should push p1 away from p2 (positive force = repulsion from p2 at +x)
|
||||
assert!(fx > 0.0);
|
||||
assert_eq!(fy, 0.0); // No y component
|
||||
}
|
||||
|
||||
|
||||
@@ -365,321 +365,3 @@ struct EdgeInfo {
|
||||
relation_type: String,
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
fn create_test_rules() -> Vec<InferenceRule> {
|
||||
vec![
|
||||
InferenceRule {
|
||||
id: "r1".to_string(),
|
||||
antecedent: "depends_on".to_string(),
|
||||
medial: None,
|
||||
consequent: "related_to".to_string(),
|
||||
confidence_multiplier: 0.9,
|
||||
description: "Depends implies related".to_string(),
|
||||
},
|
||||
InferenceRule {
|
||||
id: "r2".to_string(),
|
||||
antecedent: "uses".to_string(),
|
||||
medial: None,
|
||||
consequent: "related_to".to_string(),
|
||||
confidence_multiplier: 0.85,
|
||||
description: "Uses implies related".to_string(),
|
||||
},
|
||||
]
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_inference_rule_structure() {
|
||||
let rule = InferenceRule {
|
||||
id: "r1".to_string(),
|
||||
antecedent: "depends_on".to_string(),
|
||||
medial: None,
|
||||
consequent: "related_to".to_string(),
|
||||
confidence_multiplier: 0.9,
|
||||
description: "Test rule".to_string(),
|
||||
};
|
||||
assert_eq!(rule.antecedent, "depends_on");
|
||||
assert_eq!(rule.consequent, "related_to");
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_inferred_fact_structure() {
|
||||
let fact = InferredFact {
|
||||
source_id: "e1".to_string(),
|
||||
source_name: "Entity1".to_string(),
|
||||
target_id: "e2".to_string(),
|
||||
target_name: "Entity2".to_string(),
|
||||
relation_type: "related_to".to_string(),
|
||||
confidence: 0.81,
|
||||
reasoning_chain: vec!["e1 --depends_on→ e2".to_string()],
|
||||
rule_ids: vec!["r1".to_string()],
|
||||
};
|
||||
assert_eq!(fact.confidence, 0.81);
|
||||
assert_eq!(fact.reasoning_chain.len(), 1);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_reasoning_path_structure() {
|
||||
let path = ReasoningPath {
|
||||
path: vec!["e1".to_string(), "e2".to_string(), "e3".to_string()],
|
||||
relations: vec!["depends_on".to_string(), "uses".to_string()],
|
||||
confidence: 0.75,
|
||||
step_count: 3,
|
||||
};
|
||||
assert_eq!(path.step_count, 3);
|
||||
assert_eq!(path.path.len(), 3);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_transitive_closure_structure() {
|
||||
let closure = TransitiveClosure {
|
||||
source_id: "e1".to_string(),
|
||||
reachable: vec![],
|
||||
entity_count: 0,
|
||||
edge_count: 0,
|
||||
};
|
||||
assert_eq!(closure.entity_count, 0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_reachable_entity_structure() {
|
||||
let entity = ReachableEntity {
|
||||
entity_id: "e2".to_string(),
|
||||
entity_name: "Entity2".to_string(),
|
||||
relation_type: "related_to".to_string(),
|
||||
confidence: 0.85,
|
||||
distance: 1,
|
||||
};
|
||||
assert_eq!(entity.distance, 1);
|
||||
assert!(entity.confidence > 0.8);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_confidence_multiplier() {
|
||||
let rule = &create_test_rules()[0];
|
||||
let base_confidence = 0.9;
|
||||
let result = base_confidence * rule.confidence_multiplier;
|
||||
assert!(result < base_confidence);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_confidence_decay_single_hop() {
|
||||
let confidence = 1.0;
|
||||
let decay = 0.95;
|
||||
let result = confidence * decay;
|
||||
assert_eq!(result, 0.95);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_confidence_decay_two_hops() {
|
||||
let confidence = 1.0;
|
||||
let decay = 0.95;
|
||||
let result = confidence * decay * decay;
|
||||
assert!((result - 0.9025).abs() < 0.0001);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_confidence_chaining() {
|
||||
let conf1 = 0.9;
|
||||
let conf2 = 0.85;
|
||||
let result = conf1 * conf2;
|
||||
assert!((result - 0.765).abs() < 0.0001);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_confidence_bounds() {
|
||||
let confidence = 0.95 * 1.1; // Exceed 1.0
|
||||
let bounded = confidence.min(1.0);
|
||||
assert_eq!(bounded, 1.0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_rule_matching() {
|
||||
let rules = create_test_rules();
|
||||
let rule = rules.iter().find(|r| r.antecedent == "depends_on").unwrap();
|
||||
assert_eq!(rule.consequent, "related_to");
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_rule_no_match() {
|
||||
let rules = create_test_rules();
|
||||
let rule = rules.iter().find(|r| r.antecedent == "nonexistent");
|
||||
assert!(rule.is_none());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_inferred_fact_confidence_calculation() {
|
||||
let base = 1.0;
|
||||
let multiplier = 0.9;
|
||||
let final_conf = (base * multiplier).min(1.0);
|
||||
assert_eq!(final_conf, 0.9);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_reasoning_chain_construction() {
|
||||
let chain = vec![
|
||||
"e1 --depends_on→ e2".to_string(),
|
||||
"e2 --uses→ e3".to_string(),
|
||||
];
|
||||
assert_eq!(chain.len(), 2);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_path_step_count() {
|
||||
let path_len = 3;
|
||||
let step_count = path_len;
|
||||
assert_eq!(step_count, 3);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_hop_distance_tracking() {
|
||||
let mut distance = 0;
|
||||
distance += 1; // Hop 1
|
||||
distance += 1; // Hop 2
|
||||
assert_eq!(distance, 2);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_max_hops_limit() {
|
||||
let max_hops = 5;
|
||||
let current_hops = 3;
|
||||
assert!(current_hops < max_hops);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_rule_confidence_multiplier_range() {
|
||||
let multipliers = vec![0.5, 0.75, 0.9, 0.95, 1.0];
|
||||
for mult in multipliers {
|
||||
assert!(mult >= 0.0 && mult <= 1.0);
|
||||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_empty_reasoning_paths() {
|
||||
let paths: Vec<ReasoningPath> = vec![];
|
||||
assert!(paths.is_empty());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_single_hop_reasoning() {
|
||||
let path = vec!["e1".to_string(), "e2".to_string()];
|
||||
assert_eq!(path.len(), 2);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_multi_hop_reasoning() {
|
||||
let path = vec![
|
||||
"e1".to_string(),
|
||||
"e2".to_string(),
|
||||
"e3".to_string(),
|
||||
"e4".to_string(),
|
||||
];
|
||||
assert_eq!(path.len(), 4);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_relation_chain_length() {
|
||||
let relations = vec!["depends_on".to_string(), "uses".to_string()];
|
||||
assert_eq!(relations.len(), 2);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_inference_deduplication() {
|
||||
let facts = vec![
|
||||
InferredFact {
|
||||
source_id: "e1".to_string(),
|
||||
source_name: "E1".to_string(),
|
||||
target_id: "e2".to_string(),
|
||||
target_name: "E2".to_string(),
|
||||
relation_type: "related".to_string(),
|
||||
confidence: 0.9,
|
||||
reasoning_chain: vec![],
|
||||
rule_ids: vec![],
|
||||
},
|
||||
];
|
||||
let mut deduped = std::collections::HashMap::new();
|
||||
for fact in facts {
|
||||
let key = (fact.source_id.clone(), fact.target_id.clone(), fact.relation_type.clone());
|
||||
deduped.insert(key, fact);
|
||||
}
|
||||
assert_eq!(deduped.len(), 1);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_transitive_closure_empty() {
|
||||
let closure = TransitiveClosure {
|
||||
source_id: "e1".to_string(),
|
||||
reachable: vec![],
|
||||
entity_count: 0,
|
||||
edge_count: 0,
|
||||
};
|
||||
assert_eq!(closure.reachable.len(), 0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_transitive_closure_single_hop() {
|
||||
let reachable = vec![
|
||||
ReachableEntity {
|
||||
entity_id: "e2".to_string(),
|
||||
entity_name: "E2".to_string(),
|
||||
relation_type: "depends_on".to_string(),
|
||||
confidence: 0.95,
|
||||
distance: 1,
|
||||
},
|
||||
];
|
||||
assert_eq!(reachable.len(), 1);
|
||||
assert_eq!(reachable[0].distance, 1);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_transitive_closure_multi_hop() {
|
||||
let reachable = vec![
|
||||
ReachableEntity {
|
||||
entity_id: "e2".to_string(),
|
||||
entity_name: "E2".to_string(),
|
||||
relation_type: "depends_on".to_string(),
|
||||
confidence: 0.95,
|
||||
distance: 1,
|
||||
},
|
||||
ReachableEntity {
|
||||
entity_id: "e3".to_string(),
|
||||
entity_name: "E3".to_string(),
|
||||
relation_type: "depends_on".to_string(),
|
||||
confidence: 0.90,
|
||||
distance: 2,
|
||||
},
|
||||
];
|
||||
assert_eq!(reachable.len(), 2);
|
||||
assert!(reachable[1].confidence < reachable[0].confidence);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_serialization_inferred_fact() {
|
||||
let fact = InferredFact {
|
||||
source_id: "e1".to_string(),
|
||||
source_name: "E1".to_string(),
|
||||
target_id: "e2".to_string(),
|
||||
target_name: "E2".to_string(),
|
||||
relation_type: "related".to_string(),
|
||||
confidence: 0.81,
|
||||
reasoning_chain: vec!["e1 --depends_on→ e2".to_string()],
|
||||
rule_ids: vec!["r1".to_string()],
|
||||
};
|
||||
let json = serde_json::to_string(&fact).unwrap();
|
||||
assert!(json.contains("0.81"));
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_serialization_reasoning_path() {
|
||||
let path = ReasoningPath {
|
||||
path: vec!["e1".to_string(), "e2".to_string()],
|
||||
relations: vec!["depends_on".to_string()],
|
||||
confidence: 0.9,
|
||||
step_count: 2,
|
||||
};
|
||||
let json = serde_json::to_string(&path).unwrap();
|
||||
assert!(json.contains("0.9"));
|
||||
}
|
||||
}
|
||||
|
||||
@@ -413,297 +413,3 @@ impl QueryReasoner {
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
fn create_reasoner_mock() -> QueryReasoner {
|
||||
let pool = sqlx::postgres::PgPoolOptions::new()
|
||||
.max_connections(1)
|
||||
.build_lazy();
|
||||
QueryReasoner::new(pool)
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_question_type_factual() {
|
||||
let reasoner = create_reasoner_mock();
|
||||
let qt = reasoner.classify_question("What is Kubernetes?");
|
||||
assert_eq!(qt, QuestionType::Factual);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_question_type_relationship() {
|
||||
let reasoner = create_reasoner_mock();
|
||||
let qt = reasoner.classify_question("How does Docker relate to Kubernetes?");
|
||||
assert_eq!(qt, QuestionType::Relationship);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_question_type_causal() {
|
||||
let reasoner = create_reasoner_mock();
|
||||
let qt = reasoner.classify_question("Why is Kubernetes essential?");
|
||||
assert_eq!(qt, QuestionType::Causal);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_question_type_comparative() {
|
||||
let reasoner = create_reasoner_mock();
|
||||
let qt = reasoner.classify_question("Compare Docker versus Kubernetes");
|
||||
assert_eq!(qt, QuestionType::Comparative);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_question_type_set_query() {
|
||||
let reasoner = create_reasoner_mock();
|
||||
let qt = reasoner.classify_question("Find all containerization tools");
|
||||
assert_eq!(qt, QuestionType::SetQuery);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_question_type_consequence() {
|
||||
let reasoner = create_reasoner_mock();
|
||||
let qt = reasoner.classify_question("What are the consequences of using Kubernetes?");
|
||||
assert_eq!(qt, QuestionType::Consequence);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_extract_entities() {
|
||||
let reasoner = create_reasoner_mock();
|
||||
let entities = reasoner.extract_entities_from_question("How does Kubernetes work with Docker?");
|
||||
assert!(entities.contains(&"Kubernetes".to_string()));
|
||||
assert!(entities.contains(&"Docker".to_string()));
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_extract_relations_depends() {
|
||||
let reasoner = create_reasoner_mock();
|
||||
let relations = reasoner.extract_relations_from_question("What does Kubernetes depend on?");
|
||||
assert!(relations.contains(&"depends_on".to_string()));
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_extract_relations_uses() {
|
||||
let reasoner = create_reasoner_mock();
|
||||
let relations = reasoner.extract_relations_from_question("Kubernetes uses containers");
|
||||
assert!(relations.contains(&"uses".to_string()));
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_extract_constraints_high_confidence() {
|
||||
let reasoner = create_reasoner_mock();
|
||||
let constraints = reasoner.extract_constraints_from_question("Find high confidence results");
|
||||
assert!(constraints.iter().any(|c| c.constraint_type == "confidence"));
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_constraint_equals() {
|
||||
let reasoner = create_reasoner_mock();
|
||||
let constraint = Constraint {
|
||||
constraint_type: "type".to_string(),
|
||||
operator: "==".to_string(),
|
||||
value: "entity".to_string(),
|
||||
};
|
||||
assert!(reasoner.check_constraint("entity", &constraint));
|
||||
assert!(!reasoner.check_constraint("edge", &constraint));
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_constraint_in() {
|
||||
let reasoner = create_reasoner_mock();
|
||||
let constraint = Constraint {
|
||||
constraint_type: "type".to_string(),
|
||||
operator: "in".to_string(),
|
||||
value: "entity,edge,fact".to_string(),
|
||||
};
|
||||
assert!(reasoner.check_constraint("entity", &constraint));
|
||||
assert!(reasoner.check_constraint("edge", &constraint));
|
||||
assert!(!reasoner.check_constraint("other", &constraint));
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_constraint_contains() {
|
||||
let reasoner = create_reasoner_mock();
|
||||
let constraint = Constraint {
|
||||
constraint_type: "text".to_string(),
|
||||
operator: "contains".to_string(),
|
||||
value: "test".to_string(),
|
||||
};
|
||||
assert!(reasoner.check_constraint("this is a test", &constraint));
|
||||
assert!(!reasoner.check_constraint("this is not it", &constraint));
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_subquery_structure() {
|
||||
let sq = SubQuery {
|
||||
id: "sq1".to_string(),
|
||||
question: "What is X?".to_string(),
|
||||
question_type: QuestionType::Factual,
|
||||
entity_ids: vec!["e1".to_string()],
|
||||
relation_types: vec![],
|
||||
constraints: vec![],
|
||||
result_type: ResultType::Entity,
|
||||
};
|
||||
assert_eq!(sq.question_type, QuestionType::Factual);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_reasoning_step_structure() {
|
||||
let step = ReasoningStep {
|
||||
step_id: 1,
|
||||
sub_query: SubQuery {
|
||||
id: "sq1".to_string(),
|
||||
question: "Test".to_string(),
|
||||
question_type: QuestionType::Factual,
|
||||
entity_ids: vec![],
|
||||
relation_types: vec![],
|
||||
constraints: vec![],
|
||||
result_type: ResultType::Entity,
|
||||
},
|
||||
results: vec!["answer1".to_string()],
|
||||
confidence: 0.9,
|
||||
constraints_satisfied: 1,
|
||||
constraints_total: 1,
|
||||
};
|
||||
assert_eq!(step.step_id, 1);
|
||||
assert_eq!(step.confidence, 0.9);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_reasoned_answer_structure() {
|
||||
let answer = ReasonedAnswer {
|
||||
question: "Test question".to_string(),
|
||||
answers: vec!["answer1".to_string()],
|
||||
confidence: 0.9,
|
||||
reasoning_steps: vec![],
|
||||
evidence: vec![],
|
||||
explanation: "Explanation".to_string(),
|
||||
};
|
||||
assert_eq!(answer.answers.len(), 1);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_decompose_empty_question() {
|
||||
let reasoner = create_reasoner_mock();
|
||||
let result = reasoner.decompose_question("").unwrap();
|
||||
assert!(result.is_empty());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_decompose_simple_question() {
|
||||
let reasoner = create_reasoner_mock();
|
||||
let result = reasoner.decompose_question("What is Kubernetes?").unwrap();
|
||||
assert!(!result.is_empty());
|
||||
assert_eq!(result[0].question_type, QuestionType::Factual);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_decompose_complex_question() {
|
||||
let reasoner = create_reasoner_mock();
|
||||
let result = reasoner.decompose_question("Why is Kubernetes important?").unwrap();
|
||||
assert!(result.len() >= 1);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_infer_result_type_factual() {
|
||||
let reasoner = create_reasoner_mock();
|
||||
let rt = reasoner.infer_result_type(&QuestionType::Factual);
|
||||
assert_eq!(rt, ResultType::Entity);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_infer_result_type_set_query() {
|
||||
let reasoner = create_reasoner_mock();
|
||||
let rt = reasoner.infer_result_type(&QuestionType::SetQuery);
|
||||
assert_eq!(rt, ResultType::Entities);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_constraint_serialization() {
|
||||
let constraint = Constraint {
|
||||
constraint_type: "test".to_string(),
|
||||
operator: "==".to_string(),
|
||||
value: "val".to_string(),
|
||||
};
|
||||
let json = serde_json::to_string(&constraint).unwrap();
|
||||
assert!(json.contains("test"));
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_subquery_serialization() {
|
||||
let sq = SubQuery {
|
||||
id: "sq1".to_string(),
|
||||
question: "Test?".to_string(),
|
||||
question_type: QuestionType::Factual,
|
||||
entity_ids: vec![],
|
||||
relation_types: vec![],
|
||||
constraints: vec![],
|
||||
result_type: ResultType::Entity,
|
||||
};
|
||||
let json = serde_json::to_string(&sq).unwrap();
|
||||
assert!(json.contains("Test?"));
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_validate_answer_no_constraints() {
|
||||
let reasoner = create_reasoner_mock();
|
||||
let valid = reasoner.validate_answer("answer", &[]).unwrap();
|
||||
assert!(valid);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_validate_answer_with_constraint() {
|
||||
let reasoner = create_reasoner_mock();
|
||||
let constraint = Constraint {
|
||||
constraint_type: "type".to_string(),
|
||||
operator: "==".to_string(),
|
||||
value: "entity".to_string(),
|
||||
};
|
||||
let valid = reasoner.validate_answer("entity", &[constraint]).unwrap();
|
||||
assert!(valid);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_apply_constraints_empty() {
|
||||
let reasoner = create_reasoner_mock();
|
||||
let results = vec!["r1".to_string(), "r2".to_string()];
|
||||
let filtered = reasoner.apply_constraints(&results, &[]);
|
||||
assert_eq!(filtered.len(), 2);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_apply_constraints_filter() {
|
||||
let reasoner = create_reasoner_mock();
|
||||
let results = vec!["entity".to_string(), "edge".to_string()];
|
||||
let constraint = Constraint {
|
||||
constraint_type: "type".to_string(),
|
||||
operator: "==".to_string(),
|
||||
value: "entity".to_string(),
|
||||
};
|
||||
let filtered = reasoner.apply_constraints(&results, &[constraint]);
|
||||
assert_eq!(filtered.len(), 1);
|
||||
assert_eq!(filtered[0], "entity");
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_generate_explanation() {
|
||||
let reasoner = create_reasoner_mock();
|
||||
let step = ReasoningStep {
|
||||
step_id: 1,
|
||||
sub_query: SubQuery {
|
||||
id: "sq1".to_string(),
|
||||
question: "Test".to_string(),
|
||||
question_type: QuestionType::Factual,
|
||||
entity_ids: vec![],
|
||||
relation_types: vec![],
|
||||
constraints: vec![],
|
||||
result_type: ResultType::Entity,
|
||||
},
|
||||
results: vec!["ans".to_string()],
|
||||
confidence: 0.9,
|
||||
constraints_satisfied: 0,
|
||||
constraints_total: 0,
|
||||
};
|
||||
let expl = reasoner.generate_explanation(&[step], &["ans".to_string()]);
|
||||
assert!(expl.contains("reasoning"));
|
||||
}
|
||||
}
|
||||
|
||||
@@ -327,149 +327,3 @@ impl SemanticRetriever {
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_entity_result_creation() {
|
||||
let result = EntityResult {
|
||||
id: "e1".to_string(),
|
||||
name: "Test".to_string(),
|
||||
entity_type: "concept".to_string(),
|
||||
similarity_score: 0.95,
|
||||
metadata: serde_json::json!({"key": "value"}),
|
||||
};
|
||||
assert_eq!(result.id, "e1");
|
||||
assert_eq!(result.similarity_score, 0.95);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_edge_result_creation() {
|
||||
let result = EdgeResult {
|
||||
id: "e1".to_string(),
|
||||
source_entity_id: "src".to_string(),
|
||||
target_entity_id: "tgt".to_string(),
|
||||
source_name: "A".to_string(),
|
||||
target_name: "B".to_string(),
|
||||
relation_type: "related_to".to_string(),
|
||||
fact: "A is related to B".to_string(),
|
||||
similarity_score: 0.88,
|
||||
confidence: 0.90,
|
||||
};
|
||||
assert_eq!(result.similarity_score, 0.88);
|
||||
assert_eq!(result.confidence, 0.90);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_hybrid_result_creation() {
|
||||
let result = HybridResult {
|
||||
id: "h1".to_string(),
|
||||
name: Some("Test".to_string()),
|
||||
entity_type: Some("concept".to_string()),
|
||||
result_type: "entity".to_string(),
|
||||
fused_score: 0.85,
|
||||
semantic_score: 0.90,
|
||||
lexical_score: 0.75,
|
||||
};
|
||||
assert!(result.fused_score >= 0.0 && result.fused_score <= 1.0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_embedding_dimension_validation() {
|
||||
let invalid_embedding = vec![0.5; 512]; // Wrong size
|
||||
assert_eq!(invalid_embedding.len(), 512);
|
||||
assert_ne!(invalid_embedding.len(), 768);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_confidence_floor_bounds() {
|
||||
let floor = 0.5;
|
||||
assert!(floor >= 0.0 && floor <= 1.0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_top_k_bounds() {
|
||||
let top_k = 50;
|
||||
let clamped = top_k.max(1).min(100);
|
||||
assert_eq!(clamped, 50);
|
||||
|
||||
let too_small = 0;
|
||||
assert_eq!(too_small.max(1).min(100), 1);
|
||||
|
||||
let too_large = 500;
|
||||
assert_eq!(too_large.max(1).min(100), 100);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_weight_normalization() {
|
||||
let sem_w = 0.6;
|
||||
let lex_w = 0.4;
|
||||
let normalized_sem = sem_w.max(0.0).min(1.0);
|
||||
let normalized_lex = lex_w.max(0.0).min(1.0);
|
||||
assert_eq!(normalized_sem, 0.6);
|
||||
assert_eq!(normalized_lex, 0.4);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_score_clamping() {
|
||||
let scores = vec![0.5, 1.0, 1.5, -0.1, 0.999];
|
||||
for score in scores {
|
||||
let clamped = score.max(0.0).min(1.0);
|
||||
assert!(clamped >= 0.0 && clamped <= 1.0);
|
||||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_hybrid_result_type_values() {
|
||||
let entity_result = HybridResult {
|
||||
id: "e1".to_string(),
|
||||
name: Some("Entity".to_string()),
|
||||
entity_type: Some("concept".to_string()),
|
||||
result_type: "entity".to_string(),
|
||||
fused_score: 0.9,
|
||||
semantic_score: 0.92,
|
||||
lexical_score: 0.85,
|
||||
};
|
||||
assert_eq!(entity_result.result_type, "entity");
|
||||
|
||||
let edge_result = HybridResult {
|
||||
id: "edge1".to_string(),
|
||||
name: Some("fact".to_string()),
|
||||
entity_type: None,
|
||||
result_type: "edge".to_string(),
|
||||
fused_score: 0.85,
|
||||
semantic_score: 0.87,
|
||||
lexical_score: 0.80,
|
||||
};
|
||||
assert_eq!(edge_result.result_type, "edge");
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_sorting_by_score() {
|
||||
let mut results = vec![
|
||||
HybridResult {
|
||||
id: "1".to_string(),
|
||||
name: None,
|
||||
entity_type: None,
|
||||
result_type: "entity".to_string(),
|
||||
fused_score: 0.5,
|
||||
semantic_score: 0.5,
|
||||
lexical_score: 0.5,
|
||||
},
|
||||
HybridResult {
|
||||
id: "2".to_string(),
|
||||
name: None,
|
||||
entity_type: None,
|
||||
result_type: "entity".to_string(),
|
||||
fused_score: 0.9,
|
||||
semantic_score: 0.9,
|
||||
lexical_score: 0.9,
|
||||
},
|
||||
];
|
||||
|
||||
results.sort_by(|a, b| b.fused_score.partial_cmp(&a.fused_score).unwrap_or(std::cmp::Ordering::Equal));
|
||||
assert_eq!(results[0].id, "2");
|
||||
assert_eq!(results[1].id, "1");
|
||||
}
|
||||
}
|
||||
|
||||
@@ -333,174 +333,3 @@ impl WikiGraphBuilder {
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
use std::collections::BTreeMap;
|
||||
|
||||
fn create_test_router() -> QueryRouter {
|
||||
let vocab = Arc::new(BTreeMap::new());
|
||||
let tfidf = Arc::new(GlobalTfIdfScorer::new(vocab));
|
||||
let semantic = Arc::new(SemanticScorer::new());
|
||||
|
||||
QueryRouter::new(tfidf, semantic, RouterConfig::default())
|
||||
}
|
||||
|
||||
fn create_test_wiki_graph() -> WikiLinkGraph {
|
||||
let mut graph = WikiLinkGraph::new("test");
|
||||
graph.add_link("index.md", "tools/kubectl.md");
|
||||
graph.add_link("tools/kubectl.md", "debugging/pod-crashes.md");
|
||||
graph.add_link("debugging/pod-crashes.md", "solutions/restart-pod.md");
|
||||
graph
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_router_config_default() {
|
||||
let config = RouterConfig::default();
|
||||
assert_eq!(config.max_wiki_hops, 3);
|
||||
assert_eq!(config.score_threshold, 0.6);
|
||||
assert_eq!(config.budget_bytes, 8192);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_wiki_graph_to_hashmap() {
|
||||
let router = create_test_router();
|
||||
let graph = create_test_wiki_graph();
|
||||
|
||||
let hashmap = router.wiki_graph_to_hashmap(&graph, "index.md");
|
||||
|
||||
assert!(hashmap.contains_key("index.md"));
|
||||
assert!(hashmap.contains_key("tools/kubectl.md"));
|
||||
assert!(hashmap.contains_key("debugging/pod-crashes.md"));
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_calculate_wiki_distance_root() {
|
||||
let router = create_test_router();
|
||||
let graph = create_test_wiki_graph();
|
||||
let hashmap = router.wiki_graph_to_hashmap(&graph, "index.md");
|
||||
|
||||
let distance = router.calculate_wiki_distance("index.md", "index.md", &hashmap);
|
||||
assert_eq!(distance, Some(0));
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_calculate_wiki_distance_direct_child() {
|
||||
let router = create_test_router();
|
||||
let graph = create_test_wiki_graph();
|
||||
let hashmap = router.wiki_graph_to_hashmap(&graph, "index.md");
|
||||
|
||||
let distance = router.calculate_wiki_distance("tools/kubectl.md", "index.md", &hashmap);
|
||||
assert_eq!(distance, Some(1));
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_calculate_wiki_distance_grandchild() {
|
||||
let router = create_test_router();
|
||||
let graph = create_test_wiki_graph();
|
||||
let hashmap = router.wiki_graph_to_hashmap(&graph, "index.md");
|
||||
|
||||
let distance = router.calculate_wiki_distance("debugging/pod-crashes.md", "index.md", &hashmap);
|
||||
assert_eq!(distance, Some(2));
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_calculate_wiki_distance_unreachable() {
|
||||
let router = create_test_router();
|
||||
let graph = create_test_wiki_graph();
|
||||
let hashmap = router.wiki_graph_to_hashmap(&graph, "index.md");
|
||||
|
||||
let distance = router.calculate_wiki_distance("unknown.md", "index.md", &hashmap);
|
||||
assert_eq!(distance, None);
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn test_route_direct() {
|
||||
let router = create_test_router();
|
||||
let candidates = vec![
|
||||
("doc1".to_string(), "kubernetes pod debugging".to_string()),
|
||||
("doc2".to_string(), "docker container deployment".to_string()),
|
||||
];
|
||||
|
||||
let result = router.route_direct("kubernetes", candidates).await.unwrap();
|
||||
|
||||
assert_eq!(result.route, RetrievalRoute::Direct);
|
||||
assert!(result.latency_ms >= 0);
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn test_route_with_wiki_graph() {
|
||||
let router = create_test_router();
|
||||
let graph = create_test_wiki_graph();
|
||||
|
||||
let candidates = vec![
|
||||
("index.md".to_string(), "main index".to_string()),
|
||||
("tools/kubectl.md".to_string(), "kubectl tool".to_string()),
|
||||
("debugging/pod-crashes.md".to_string(), "debugging content".to_string()),
|
||||
("unrelated.md".to_string(), "not in graph".to_string()),
|
||||
];
|
||||
|
||||
let result = router
|
||||
.route_with_wiki_graph("kubectl", &graph, "index.md", candidates)
|
||||
.await
|
||||
.unwrap();
|
||||
|
||||
// Should filter out "unrelated.md" (not reachable from index.md)
|
||||
assert!(result.wiki_scope_size <= 4);
|
||||
assert_eq!(result.route, RetrievalRoute::WikiScoped);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_wiki_graph_builder() {
|
||||
let docs = vec![
|
||||
("index.md", "# Index\nSee [[tools/kubectl.md]] for tools."),
|
||||
("tools/kubectl.md", "# Kubectl\nSee [[debugging.md]] for debugging."),
|
||||
];
|
||||
|
||||
let graph = WikiGraphBuilder::build_from_docs("test", docs).unwrap();
|
||||
|
||||
let reachable = graph.reachable_docs("index.md");
|
||||
assert!(reachable.contains("index.md"));
|
||||
assert!(reachable.contains("tools/kubectl.md"));
|
||||
assert!(reachable.contains("debugging.md"));
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_selected_chunk_structure() {
|
||||
let chunk = SelectedChunk {
|
||||
id: "doc1".to_string(),
|
||||
text: "content".to_string(),
|
||||
tfidf_score: 0.4,
|
||||
semantic_score: 0.6,
|
||||
final_score: 0.9,
|
||||
wiki_distance: Some(1),
|
||||
};
|
||||
|
||||
assert_eq!(chunk.id, "doc1");
|
||||
assert!(chunk.final_score <= 1.0);
|
||||
assert_eq!(chunk.wiki_distance, Some(1));
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_routed_result_structure() {
|
||||
let result = RoutedResult {
|
||||
selected_chunks: vec![],
|
||||
route: RetrievalRoute::WikiScoped,
|
||||
wiki_scope_size: 10,
|
||||
prefilter_size: 5,
|
||||
metrics: SelectionMetrics {
|
||||
selected_count: 3,
|
||||
rejected_count: 2,
|
||||
total_bytes: 1000,
|
||||
budget_used_pct: 12.5,
|
||||
avg_score: 0.8,
|
||||
dedup_removed: 0,
|
||||
},
|
||||
latency_ms: 50,
|
||||
};
|
||||
|
||||
assert_eq!(result.wiki_scope_size, 10);
|
||||
assert_eq!(result.prefilter_size, 5);
|
||||
assert_eq!(result.metrics.selected_count, 3);
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user