keyhog-core 0.5.4

keyhog-core — shared data model and detector specifications for the KeyHog secret scanner
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

use keyhog_core::registry::{CustomVerifier, SourceRegistry, VerifierRegistry};
use keyhog_core::{Chunk, DedupedMatch, Source, SourceError, VerificationResult};
use std::collections::HashMap;
use std::sync::Arc;

struct MockSource;
impl Source for MockSource {
    fn name(&self) -> &str {
        "mock"
    }
    fn chunks(&self) -> Box<dyn Iterator<Item = Result<Chunk, SourceError>> + '_> {
        Box::new(std::iter::empty())
    }
    fn as_any(&self) -> &dyn std::any::Any {
        self
    }
}

#[test]
fn source_registry_register_and_get() {
    let registry = SourceRegistry::new(); // Assuming we added a new() for testing or use global
    let source = Arc::new(MockSource);
    registry.register(source.clone());

    let retrieved = registry.get("mock").unwrap();
    assert_eq!(retrieved.name(), "mock");
}

struct MockVerifier;
#[async_trait::async_trait]
impl CustomVerifier for MockVerifier {
    fn name(&self) -> &str {
        "mock-v"
    }
    async fn verify(&self, _m: &DedupedMatch) -> (VerificationResult, HashMap<String, String>) {
        (VerificationResult::Live, HashMap::new())
    }
}

#[tokio::test]
async fn verifier_registry_register_and_get() {
    let registry = VerifierRegistry::new();
    let verifier = Arc::new(MockVerifier);
    registry.register(verifier.clone());

    let retrieved = registry.get("mock-v").unwrap();
    assert_eq!(retrieved.name(), "mock-v");
}