atento-core 0.1.0

Core engine for the Atento Chained Script CLI
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

use crate::errors::Result;
use crate::executor::{CommandExecutor, ExecutionResult};
use crate::interpreter::Interpreter;
use std::cell::RefCell;
use std::collections::HashMap;

type CallRecord = (String, Interpreter, u64);

/// Mock implementation for unit tests
pub struct MockExecutor {
    responses: HashMap<String, ExecutionResult>,
    default_response: ExecutionResult,
    call_count: RefCell<usize>,
    last_call: RefCell<Option<CallRecord>>,
}

impl MockExecutor {
    pub fn new() -> Self {
        Self {
            responses: HashMap::new(),
            default_response: ExecutionResult {
                stdout: "mock output".to_string(),
                stderr: String::new(),
                exit_code: 0,
                duration_ms: 10,
            },
            call_count: RefCell::new(0),
            last_call: RefCell::new(None),
        }
    }

    pub fn expect_call(&mut self, script: &str, response: ExecutionResult) -> &mut Self {
        self.responses.insert(script.to_string(), response);
        self
    }

    pub fn expect_timeout(&mut self, script: &str) -> &mut Self {
        self.responses.insert(
            script.to_string(),
            ExecutionResult {
                stdout: String::new(),
                stderr: "Timeout".to_string(),
                exit_code: 124,
                duration_ms: 1000,
            },
        );
        self
    }

    pub fn expect_error(&mut self, script: &str, exit_code: i32, stderr: &str) -> &mut Self {
        self.responses.insert(
            script.to_string(),
            ExecutionResult {
                stdout: String::new(),
                stderr: stderr.to_string(),
                exit_code,
                duration_ms: 5,
            },
        );
        self
    }

    pub fn call_count(&self) -> usize {
        *self.call_count.borrow()
    }

    pub fn last_call(&self) -> Option<(String, Interpreter, u64)> {
        self.last_call.borrow().clone()
    }
}

impl CommandExecutor for MockExecutor {
    fn execute(
        &self,
        script: &str,
        interpreter: &Interpreter,
        timeout: u64,
    ) -> Result<ExecutionResult> {
        *self.call_count.borrow_mut() += 1;
        *self.last_call.borrow_mut() = Some((script.to_string(), interpreter.clone(), timeout));

        Ok(self
            .responses
            .get(script)
            .cloned()
            .unwrap_or_else(|| self.default_response.clone()))
    }
}

impl Default for MockExecutor {
    fn default() -> Self {
        Self::new()
    }
}