pmat 3.11.0

PMAT - Zero-config AI context generation and code quality toolkit (CLI, MCP, HTTP)
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

mod proptest_generation_tests {
    use super::*;

    #[test]
    fn test_generate_module_contains_import() {
        let config = CoverageImprovementConfig::default();
        let service = CoverageImprovementService::new(config);

        let code = "pub fn test_fn() {}";
        let syntax_tree = syn::parse_file(code).unwrap();
        let functions = service.extract_public_functions(&syntax_tree);
        let target = PathBuf::from("lib.rs");

        let module = service
            .generate_proptest_module(&target, &functions)
            .unwrap();

        assert!(module.contains("use crate::lib::*;"));
    }

    #[test]
    fn test_generate_module_multiple_functions() {
        let config = CoverageImprovementConfig::default();
        let service = CoverageImprovementService::new(config);

        let code = r#"
            pub fn func_a(x: i32) -> i32 { x }
            pub fn func_b(s: String) -> String { s }
            pub fn func_c() {}
        "#;
        let syntax_tree = syn::parse_file(code).unwrap();
        let functions = service.extract_public_functions(&syntax_tree);
        let target = PathBuf::from("multi.rs");

        let module = service
            .generate_proptest_module(&target, &functions)
            .unwrap();

        assert!(module.contains("fn proptest_func_a"));
        assert!(module.contains("fn proptest_func_b"));
        assert!(module.contains("fn proptest_func_c"));
    }

    #[test]
    fn test_generate_module_nested_path() {
        let config = CoverageImprovementConfig::default();
        let service = CoverageImprovementService::new(config);

        let code = "pub fn test_fn() {}";
        let syntax_tree = syn::parse_file(code).unwrap();
        let functions = service.extract_public_functions(&syntax_tree);
        let target = PathBuf::from("src/services/coverage.rs");

        let module = service
            .generate_proptest_module(&target, &functions)
            .unwrap();

        // Should use just the file stem
        assert!(module.contains("use crate::coverage::*;"));
    }
}