nushell/crates/nu-command/tests/commands/default.rs

use nu_test_support::{fs::Stub::EmptyFile, nu, pipeline, playground::Playground};

#[test]
fn adds_row_data_if_column_missing() {
    let sample = r#"
                {
                    "amigos": [
                        {"name": "Yehuda"},
                        {"name": "JT", "rusty_luck": 0},
                        {"name": "Andres", "rusty_luck": 0},
                        {"name": "Michael", "rusty_luck": []},
                        {"name": "Darren", "rusty_luck": {}},
                        {"name": "Stefan", "rusty_luck": ""},
                        {"name": "GorbyPuff"}
                    ]
                }
            "#;

    let actual = nu!(pipeline(&format!(
        "
                {sample}
                | get amigos
                | default 1 rusty_luck
                | where rusty_luck == 1
                | length
            "
    )));

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

#[test]
fn default_after_empty_filter() {
    let actual = nu!("[a b] | where $it == 'c' | get -i 0 | default 'd'");

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

#[test]
fn keeps_nulls_in_lists() {
    let actual = nu!(r#"[null, 2, 3] | default [] | to json -r"#);
    assert_eq!(actual.out, "[null,2,3]");
}

#[test]
fn replaces_null() {
    let actual = nu!(r#"null | default 1"#);
    assert_eq!(actual.out, "1");
}

#[test]
fn adds_row_data_if_column_missing_or_empty() {
    let sample = r#"
                {
                    "amigos": [
                        {"name": "Yehuda"},
                        {"name": "JT", "rusty_luck": 0},
                        {"name": "Andres", "rusty_luck": 0},
                        {"name": "Michael", "rusty_luck": []},
                        {"name": "Darren", "rusty_luck": {}},
                        {"name": "Stefan", "rusty_luck": ""},
                        {"name": "GorbyPuff"}
                    ]
                }
            "#;

    let actual = nu!(pipeline(&format!(
        "
                {sample}
                | get amigos
                | default -e 1 rusty_luck
                | where rusty_luck == 1
                | length
            "
    )));

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

#[test]
fn replace_empty_string() {
    let actual = nu!(r#"'' | default -e foo"#);
    assert_eq!(actual.out, "foo");
}

#[test]
fn do_not_replace_empty_string() {
    let actual = nu!(r#"'' | default 1"#);
    assert_eq!(actual.out, "");
}

#[test]
fn replace_empty_list() {
    let actual = nu!(r#"[] | default -e foo"#);
    assert_eq!(actual.out, "foo");
}

#[test]
fn do_not_replace_empty_list() {
    let actual = nu!(r#"[] | default 1 | length"#);
    assert_eq!(actual.out, "0");
}

#[test]
fn replace_empty_record() {
    let actual = nu!(r#"{} | default -e foo"#);
    assert_eq!(actual.out, "foo");
}

#[test]
fn do_not_replace_empty_record() {
    let actual = nu!(r#"{} | default {a:5} | columns | length"#);
    assert_eq!(actual.out, "0");
}

#[test]
fn replace_empty_list_stream() {
    // This is specific for testing ListStreams when empty behave like other empty values
    Playground::setup("glob_empty_list", |dirs, sandbox| {
        sandbox.with_files(&[
            EmptyFile("yehuda.txt"),
            EmptyFile("jttxt"),
            EmptyFile("andres.txt"),
        ]);

        let actual = nu!(
            cwd: dirs.test(),
            "glob ? | default -e void",
        );

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

#[test]
fn do_not_replace_non_empty_list_stream() {
    // This is specific for testing ListStreams when empty behave like other empty values
    Playground::setup("glob_non_empty_list", |dirs, sandbox| {
        sandbox.with_files(&[
            EmptyFile("yehuda.txt"),
            EmptyFile("jt.rs"),
            EmptyFile("andres.txt"),
        ]);

        let actual = nu!(
            cwd: dirs.test(),
            "glob '*.txt' | default -e void | length",
        );

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

#[test]
fn closure_eval_simple() {
    let actual = nu!(r#"null | default { 1 }"#);
    assert_eq!(actual.out, "1");
}

#[test]
fn closure_eval_complex() {
    let actual = nu!(r#"null | default { seq 1 5 | math sum }"#);
    assert_eq!(actual.out, "15");
}

#[test]
fn closure_eval_is_lazy() {
    let actual = nu!(r#"1 | default { error make -u {msg: foo} }"#);
    assert_eq!(actual.out, "1");
}

#[test]
fn column_closure_eval_is_lazy() {
    let actual = nu!(r#"{a: 1} | default { error make -u {msg: foo} } a | get a"#);
    assert_eq!(actual.out, "1");
}

#[test]
fn closure_eval_replace_empty_string() {
    let actual = nu!(r#"'' | default --empty { 1 }"#);
    assert_eq!(actual.out, "1");
}

#[test]
fn closure_eval_do_not_replace_empty_string() {
    let actual = nu!(r#"'' | default { 1 }"#);
    assert_eq!(actual.out, "");
}

#[test]
fn closure_eval_replace_empty_list() {
    let actual = nu!(r#"[] | default --empty { 1 }"#);
    assert_eq!(actual.out, "1");
}

#[test]
fn closure_eval_do_not_replace_empty_list() {
    let actual = nu!(r#"[] | default { 1 } | length"#);
    assert_eq!(actual.out, "0");
}

#[test]
fn closure_eval_replace_empty_record() {
    let actual = nu!(r#"{} | default --empty { 1 }"#);
    assert_eq!(actual.out, "1");
}

#[test]
fn closure_eval_do_not_replace_empty_record() {
    let actual = nu!(r#"{} | default { 1 } | columns | length"#);
    assert_eq!(actual.out, "0");
}

#[test]
fn closure_eval_add_missing_column_record() {
    let actual = nu!(r#"
        {a: 1} | default { 2 } b | get b
    "#);
    assert_eq!(actual.out, "2");
}

#[test]
fn closure_eval_add_missing_column_table() {
    let actual = nu!(r#"
        [{a: 1, b: 2}, {b: 4}] | default { 3 } a | get a | to json -r
    "#);
    assert_eq!(actual.out, "[1,3]");
}

#[test]
fn closure_eval_replace_empty_column() {
    let actual = nu!(r#"{a: ''} | default -e { 1 } a | get a"#);
    assert_eq!(actual.out, "1");
}

#[test]
fn replace_multiple_columns() {
    let actual = nu!(r#"{a: ''} | default -e 1 a b | values | to json -r"#);
    assert_eq!(actual.out, "[1,1]");
}

#[test]
fn return_closure_value() {
    let actual = nu!(r#"null | default { {||} }"#);
    assert!(actual.out.starts_with("closure"));
}