flowstdlib 0.142.0

use serde_json::{json, Value};

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

#[flow_function]
fn _sort(inputs: &[Value]) -> Result<(Option<Value>, RunAgain)> {
    if inputs[0].is_null() {
        return Ok((Some(Value::Null), RUN_AGAIN));
    }

    let array_num = inputs[0].as_array().ok_or("Could not get array")?;
    let mut array_of_numbers: Vec<Value> = array_num.clone();
    array_of_numbers.sort_by_key(|a| a.as_i64().unwrap_or(0));

    Ok((Some(json!(array_of_numbers)), RUN_AGAIN))
}

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

    use super::_sort;

    #[test]
    fn sort_null() {
        let (result, _) = _sort(&[Value::Null]).expect("_sort() failed");

        let output = result.expect("Could not get output value");
        assert_eq!(output, Value::Null);
    }

    #[test]
    fn sort_invalid() {
        assert!(_sort(&[json!("Hello World")]).is_err());
    }

    #[test]
    fn sort_one() {
        let (result, _) = _sort(&[json!([1])]).expect("_sort() failed");

        let output = result.expect("Could not get output value");
        assert_eq!(output, json!([1]));
    }

    #[test]
    fn sort_array() {
        let (result, _) = _sort(&[json!([7, 1, 4, 8, 3, 9])]).expect("_sort() failed");

        let output = result.expect("Could not get output value");
        assert_eq!(output, json!([1, 3, 4, 7, 8, 9]));
    }

    #[test]
    fn sort_array_repeats() {
        let (result, _) = _sort(&[json!([7, 1, 8, 4, 8, 3, 1, 9])]).expect("_sort() failed");

        let output = result.expect("Could not get output value");
        assert_eq!(output, json!([1, 1, 3, 4, 7, 8, 8, 9]));
    }
}