ryo-executor 0.1.0

[experimental] Mutation execution engine for RYO - parallel execution, conflict detection, workspace management
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
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116

//! MatchArmConverter: Converts MutationSpec::AddMatchArm, RemoveMatchArm, and ReplaceMatchArm

use crate::engine::ASTRegApply;
use crate::executor::registry::converters::ResolveTargetSymbol;
use crate::executor::registry::{ConvertError, MutationConverter};
use crate::executor::spec::MutationSpec;
use ryo_analysis::AnalysisContext;
use ryo_mutations::{AddMatchArmMutation, RemoveMatchArmMutation, ReplaceMatchArmMutation};

/// Converter for AddMatchArm, RemoveMatchArm, and ReplaceMatchArm mutations
#[derive(Debug, Clone, Default)]
pub struct MatchArmConverter;

impl MatchArmConverter {
    pub fn new() -> Self {
        Self
    }
}

// MatchArmConverter uses the default implementation of ResolveTargetSymbol
impl ResolveTargetSymbol for MatchArmConverter {}

impl MutationConverter for MatchArmConverter {
    fn spec_kinds(&self) -> &'static [&'static str] {
        &["AddMatchArm", "RemoveMatchArm", "ReplaceMatchArm"]
    }

    fn convert_v2(
        &self,
        spec: &MutationSpec,
        ctx: &AnalysisContext,
    ) -> Result<Vec<Box<dyn ASTRegApply>>, ConvertError> {
        match spec {
            MutationSpec::AddMatchArm {
                target,
                enum_name,
                pattern,
                body,
            } => {
                let fn_id = self.resolve_target_symbol(target, ctx)?;

                let mutation = AddMatchArmMutation::new(
                    fn_id,
                    enum_name.clone(),
                    pattern.clone(),
                    body.clone(),
                );
                Ok(vec![Box::new(mutation)])
            }
            MutationSpec::RemoveMatchArm {
                target,
                enum_name,
                pattern,
            } => {
                let fn_id = self.resolve_target_symbol(target, ctx)?;

                let mutation =
                    RemoveMatchArmMutation::new(fn_id, enum_name.clone(), pattern.clone());
                Ok(vec![Box::new(mutation)])
            }
            MutationSpec::ReplaceMatchArm {
                target,
                enum_name,
                old_pattern,
                new_pattern,
                new_body,
            } => {
                let fn_id = self.resolve_target_symbol(target, ctx)?;

                let mutation = ReplaceMatchArmMutation::new(
                    fn_id,
                    enum_name.clone(),
                    old_pattern.clone(),
                    new_pattern.clone(),
                    new_body.clone(),
                );
                Ok(vec![Box::new(mutation)])
            }
            _ => Err(ConvertError::TypeMismatch {
                expected: "AddMatchArm, RemoveMatchArm, or ReplaceMatchArm",
                actual: spec.kind_name().to_string(),
            }),
        }
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::executor::spec::MutationTargetSymbol;
    use ryo_analysis::SymbolPath;

    #[test]
    fn test_match_arm_converter_spec_kinds() {
        let converter = MatchArmConverter::new();
        assert_eq!(
            converter.spec_kinds(),
            &["AddMatchArm", "RemoveMatchArm", "ReplaceMatchArm"]
        );
    }

    #[test]
    fn test_match_arm_converter_can_handle() {
        let converter = MatchArmConverter::new();

        let spec = MutationSpec::AddMatchArm {
            target: MutationTargetSymbol::ByPath(Box::new(
                SymbolPath::parse("test_crate::handlers::process_status").unwrap(),
            )),
            enum_name: "Status".to_string(),
            pattern: "Status::Cancelled".to_string(),
            body: "todo!()".to_string(),
        };
        assert!(converter.can_handle(&spec));
    }
}