psi_detector 0.1.3

Protocol detection and upgrade framework inspired by Yuri's PSI Detector
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
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103

//! 核心探测器模块测试

use psi_detector::core::detector::*;
use psi_detector::core::protocol::{ProtocolType, ProtocolInfo};
use std::time::Duration;

#[test]
fn test_detection_result() {
    let protocol_info = ProtocolInfo::new(ProtocolType::HTTP2, 0.95);
    let result = DetectionResult::new(
        protocol_info,
        Duration::from_millis(10),
        DetectionMethod::Passive,
        "test_detector".to_string(),
    );
    
    assert_eq!(result.protocol_type(), ProtocolType::HTTP2);
    assert_eq!(result.confidence(), 0.95);
    assert!(result.is_high_confidence());
    assert!(result.is_acceptable(0.8));
}

#[test]
fn test_detection_config() {
    let config = DetectionConfig::new()
        .with_min_confidence(0.8)
        .with_timeout(Duration::from_millis(500))
        .enable_heuristic()
        .disable_active_probing();
    
    assert_eq!(config.min_confidence, 0.8);
    assert_eq!(config.timeout, Duration::from_millis(500));
    assert!(config.enable_heuristic);
    assert!(!config.enable_active_probing);
}

#[test]
fn test_detection_stats() {
    let mut stats = DetectionStats::new();
    
    stats.record_success(ProtocolType::HTTP2, Duration::from_millis(10));
    stats.record_success(ProtocolType::HTTP2, Duration::from_millis(20));
    stats.record_failure(Duration::from_millis(5));
    
    assert_eq!(stats.total_detections, 3);
    assert_eq!(stats.successful_detections, 2);
    assert_eq!(stats.failed_detections, 1);
    assert!((stats.success_rate() - 0.6666666666666666).abs() < f64::EPSILON);
    assert_eq!(stats.most_common_protocol(), Some(ProtocolType::HTTP2));
}

#[test]
fn test_detection_config_builder() {
    let config = DetectionConfig::new()
        .with_min_confidence(0.9)
        .with_timeout(Duration::from_secs(1))
        .with_max_probe_size(4096)
        .enable_simd()
        .enable_heuristic()
        .enable_active_probing();
    
    assert_eq!(config.min_confidence, 0.9);
    assert_eq!(config.timeout, Duration::from_secs(1));
    assert_eq!(config.max_probe_size, 4096);
    assert!(config.enable_simd);
    assert!(config.enable_heuristic);
    assert!(config.enable_active_probing);
}

#[test]
fn test_detection_stats_empty() {
    let stats = DetectionStats::new();
    
    assert_eq!(stats.total_detections, 0);
    assert_eq!(stats.successful_detections, 0);
    assert_eq!(stats.failed_detections, 0);
    assert_eq!(stats.success_rate(), 0.0);
    assert_eq!(stats.most_common_protocol(), None);
}

#[test]
fn test_detection_stats_single_success() {
    let mut stats = DetectionStats::new();
    stats.record_success(ProtocolType::TLS, Duration::from_millis(15));
    
    assert_eq!(stats.total_detections, 1);
    assert_eq!(stats.successful_detections, 1);
    assert_eq!(stats.failed_detections, 0);
    assert_eq!(stats.success_rate(), 1.0);
    assert_eq!(stats.most_common_protocol(), Some(ProtocolType::TLS));
}

#[test]
fn test_detection_stats_single_failure() {
    let mut stats = DetectionStats::new();
    stats.record_failure(Duration::from_millis(5));
    
    assert_eq!(stats.total_detections, 1);
    assert_eq!(stats.successful_detections, 0);
    assert_eq!(stats.failed_detections, 1);
    assert_eq!(stats.success_rate(), 0.0);
    assert_eq!(stats.most_common_protocol(), None);
}