flowstdlib 0.142.0

use serde_json::json;
use serde_json::Value;

use flowcore::{RUN_AGAIN, RunAgain};
use flowcore::errors::*;
use flowmacro::flow_function;

#[flow_function]
fn _zip(inputs: &[Value]) -> Result<(Option<Value>, RunAgain)> {
    let left = &inputs[0].as_array().ok_or("Could not get left array")?;
    let right = &inputs[1].as_array().ok_or("Could not get right array")?;
    let tuples = left.iter().zip(right.iter());
    let tuples_vec: Vec<(&Value, &Value)> = tuples.collect();
    Ok((Some(json!(tuples_vec)), RUN_AGAIN))
}

#[cfg(test)]
mod test {
    use serde_json::{json, Value};

    use super::_zip;

    #[test]
    fn zip_empty() {
        let left = Value::Array(vec![]);
        let right = Value::Array(vec![]);

        let inputs = vec![left, right];

        let (result, _) = _zip(&inputs).expect("_zip() failed");

        let zipped_array = result.expect("Could not get the value from the output");

        assert_eq!(zipped_array, Value::Array(vec!()));
    }

    #[test]
    fn zip_happy() {
        let left = json!(vec![1, 2]);
        let right = json!(vec![3, 4]);

        let inputs = vec![left, right];

        let (result, _) = _zip(&inputs).expect("_zip() failed");

        let zipped_array = result.expect("Could not get the value from the output");

        assert_eq!(zipped_array, json!(vec![(1,3), (2,4)]));
    }

    #[test]
    fn zip_invalid_left() {
        let left = json!(1);
        let right = json!(vec![3, 4]);

        let inputs = vec![left, right];

        assert!(_zip(&inputs).is_err());
    }

    #[test]
    fn zip_invalid_right() {
        let left = json!(vec![1, 2]);
        let right = json!(3);

        let inputs = vec![left, right];

        assert!(_zip(&inputs).is_err());
    }

    #[test]
    fn zip_unequal() {
        let left = json!(vec![1, 2]);
        let right = json!(vec![3]);

        let inputs = vec![left, right];

        let (result, _) = _zip(&inputs).expect("_zip() failed");

        let zipped_array = result.expect("Could not get the value from the output");

        assert_eq!(zipped_array, json!(vec![(1,3)]));
    }
}