bgpkit-parser 0.16.0

MRT/BGP/BMP data processing library
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

#[cfg(test)]
mod tests {
    use std::collections::HashMap;
    use std::path::PathBuf;

    use bgpkit_parser::models::BgpElem;
    use bgpkit_parser::BgpkitParser;
    use tempfile::TempDir;

    fn setup_test_dir() -> (TempDir, PathBuf) {
        let temp_dir = TempDir::new().unwrap();
        let temp_path = temp_dir.path().to_path_buf();

        // Do common setup here
        std::fs::create_dir_all(temp_path.join("update-example")).unwrap();

        (temp_dir, temp_path)
    }

    #[test]
    fn test_parser_as_paths_conversion() {
        let (_, temp_path) = setup_test_dir();
        let url = "https://spaces.bgpkit.org/parser/update-example.gz";

        let parser = BgpkitParser::new_cached(url, temp_path.to_str().unwrap())
            .unwrap()
            .add_filter("peer_ip", "185.1.8.50")
            .unwrap()
            .add_filter("type", "a")
            .unwrap();

        let elems = parser.into_elem_iter().collect::<Vec<BgpElem>>();
        assert_eq!(
            vec![200612, 174, 1299, 31027, 198622],
            elems
                .first()
                .unwrap()
                .as_path
                .as_ref()
                .unwrap()
                .to_u32_vec_opt(false)
                .unwrap()
        );
    }

    // Iterate over the file twice
    #[test]
    fn test_record_iter_raw_record_iter_types() {
        let (_, temp_path) = setup_test_dir();
        let url = "https://spaces.bgpkit.org/parser/update-example.gz";

        let mut record_types = HashMap::new();
        let mut raw_record_types = HashMap::new();

        for record in BgpkitParser::new_cached(url, temp_path.to_str().unwrap())
            .unwrap()
            .into_record_iter()
        {
            let entry = record_types
                .entry((
                    record.common_header.entry_type,
                    record.common_header.entry_subtype,
                ))
                .or_insert(0);
            *entry += record.common_header.length;
        }

        for record in BgpkitParser::new_cached(url, temp_path.to_str().unwrap())
            .unwrap()
            .into_raw_record_iter()
        {
            let entry = raw_record_types
                .entry((
                    record.common_header.entry_type,
                    record.common_header.entry_subtype,
                ))
                .or_insert(0);
            *entry += record.common_header.length;
        }

        // Same file -> same records -> same sizes.
        assert_eq!(record_types, raw_record_types);
    }
}