nushell/crates/nu-command/tests/commands/take/while_.rs

use nu_test_support::{nu, pipeline};

#[test]
fn condition_is_met() {
    let sample = r#"
                    [["Chicken Collection", "29/04/2020", "30/04/2020", "31/04/2020"];
                     ["Yellow Chickens", "", "", ""],
                     [Andrés, 1, 1, 1],
                     [JT, 1, 1, 1],
                     [Jason, 1, 1, 1],
                     [Yehuda, 1, 1, 1],
                     ["Blue Chickens", "", "", ""],
                     [Andrés, 1, 1, 2],
                     [JT, 1, 1, 2],
                     [Jason, 1, 1, 2],
                     [Yehuda, 1, 1, 2],
                     ["Red Chickens", "", "", ""],
                     [Andrés, 1, 1, 3],
                     [JT, 1, 1, 3],
                     [Jason, 1, 1, 3],
                     [Yehuda, 1, 1, 3]]
            "#;

    let actual = nu!(pipeline(&format!(
        r#"
                {sample}
                | skip 1
                | take while {{|row| $row."Chicken Collection" != "Blue Chickens"}}
                | into int "31/04/2020"
                | get "31/04/2020"
                | math sum
                "#
    )));

    assert_eq!(actual.out, "4");
}

#[test]
fn fail_on_non_iterator() {
    let actual = nu!("1 | take while {|row| $row == 2}");

    assert!(actual.err.contains("command doesn't support"));
}