ergotree_interpreter/eval/
extract_script_bytes.rs

1use ergotree_ir::mir::extract_script_bytes::ExtractScriptBytes;
2use ergotree_ir::mir::value::Value;
3
4use crate::eval::env::Env;
5use crate::eval::Context;
6use crate::eval::EvalError;
7use crate::eval::Evaluable;
8
9impl Evaluable for ExtractScriptBytes {
10    fn eval<'ctx>(
11        &self,
12        env: &mut Env<'ctx>,
13        ctx: &Context<'ctx>,
14    ) -> Result<Value<'ctx>, EvalError> {
15        let input_v = self.input.eval(env, ctx)?;
16        match input_v {
17            Value::CBox(b) => Ok(b.script_bytes()?.into()),
18            _ => Err(EvalError::UnexpectedValue(format!(
19                "Expected ExtractScriptBytes input to be Value::CBox, got {0:?}",
20                input_v
21            ))),
22        }
23    }
24}
25
26#[allow(clippy::unwrap_used)]
27#[cfg(test)]
28mod tests {
29    use super::*;
30    use crate::eval::context::Context;
31    use crate::eval::tests::eval_out;
32    use ergotree_ir::mir::expr::Expr;
33    use ergotree_ir::mir::global_vars::GlobalVars;
34    use sigma_test_util::force_any_val;
35
36    #[test]
37    fn eval() {
38        let e: Expr = ExtractScriptBytes {
39            input: Box::new(GlobalVars::SelfBox.into()),
40        }
41        .into();
42        let ctx = force_any_val::<Context>();
43        assert_eq!(
44            eval_out::<Vec<i8>>(&e, &ctx),
45            ctx.self_box.script_bytes().unwrap()
46        );
47    }
48}