keyhog 0.5.73

GPU-accelerated secret scanner for code, Git history, cloud, containers, browser assets, and live credential verification
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

//! Unit tests for daemon client timeouts (KH-1459).

#[cfg(test)]
use super::{
    request_timeout, DAEMON_HEALTH_TIMEOUT, DAEMON_REQUEST_TIMEOUT, DAEMON_SCAN_TEXT_TIMEOUT,
    DAEMON_SHUTDOWN_TIMEOUT,
};
#[cfg(test)]
use crate::daemon::protocol::Request;

#[cfg(test)]
#[test]
fn health_and_hello_use_short_timeout() {
    assert_eq!(request_timeout(&Request::Health), DAEMON_HEALTH_TIMEOUT);
    assert_eq!(request_timeout(&Request::Hello), DAEMON_HEALTH_TIMEOUT);
    assert_eq!(DAEMON_HEALTH_TIMEOUT.as_secs(), 5);
}

#[cfg(test)]
#[test]
fn shutdown_timeout_exceeds_the_bounded_server_drain() {
    assert_eq!(request_timeout(&Request::Shutdown), DAEMON_SHUTDOWN_TIMEOUT);
    assert_eq!(DAEMON_SHUTDOWN_TIMEOUT.as_secs(), 45);
}

#[cfg(test)]
#[test]
fn scan_text_uses_mid_timeout() {
    let req = Request::ScanText {
        path: None,
        text: "x".into(),
        dogfood: false,
        profile: false,
    };
    assert_eq!(request_timeout(&req), DAEMON_SCAN_TEXT_TIMEOUT);
    assert_eq!(DAEMON_SCAN_TEXT_TIMEOUT.as_secs(), 60);
}

#[cfg(test)]
#[test]
fn scan_path_uses_full_file_timeout() {
    let req = Request::ScanPath {
        path: "/tmp/a".into(),
        working_dir: None,
        dogfood: false,
        profile: false,
    };
    assert_eq!(request_timeout(&req), DAEMON_REQUEST_TIMEOUT);
    assert_eq!(DAEMON_REQUEST_TIMEOUT.as_secs(), 300);
}

#[cfg(test)]
#[test]
fn mass_filesystem_acquisition_uses_workload_timeout() {
    let request = Request::MassFilesystemBegin {
        root: "/workspace".into(),
        max_file_size: 1024,
        ignore_paths: Vec::new(),
        respect_default_excludes: true,
        reader_threads: None,
        incremental_cache: Some("/cache/keyhog/merkle.idx".into()),
    };
    assert_eq!(request_timeout(&request), DAEMON_REQUEST_TIMEOUT);
}

#[cfg(test)]
#[test]
fn mass_filesystem_drain_uses_full_workload_timeout() {
    assert_eq!(
        request_timeout(&Request::MassFilesystemDrain),
        DAEMON_REQUEST_TIMEOUT
    );
}