geti 0.2.0

A tool for auditing Arbitrum stylus Rust smart contracts
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

// src/extraction_tests.rs

#[cfg(test)]
mod tests {
    use super::super::{init, extraction::extract_contracts};
    use std::fs;
    use std::path::Path;

    #[test]
    fn test_extract_contracts_current_dir() {
        init();
        let contract_files = extract_contracts(None);
        assert!(!contract_files.is_empty());
    }

    #[test]
    fn test_extract_contracts_specified_dir() {
        init();
        let test_dir = "test_contracts";
        fs::create_dir_all(test_dir).unwrap();
        fs::write(Path::new(test_dir).join("test1.rs"), "contract code").unwrap();
        fs::write(Path::new(test_dir).join("test2.txt"), "not a contract").unwrap();

        let contract_files = extract_contracts(Some(test_dir));
        assert_eq!(contract_files.len(), 1);
        assert_eq!(contract_files[0].file_name().unwrap(), "test1.rs");

        fs::remove_dir_all(test_dir).unwrap();
    }
}