dolt/integration-tests/bats/import-append-tables.bats

#!/usr/bin/env bats
load $BATS_TEST_DIRNAME/helper/common.bash

setup() {
    setup_common
}

teardown() {
    assert_feature_version
    teardown_common
}

@test "import-append-tables: disallow overwriting row during append" {
    dolt sql -q "CREATE TABLE t (pk int primary key, col1 int);"
    dolt table import -a t <<CSV
pk, col1
1, 1
CSV
    run dolt table import -a t <<CSV
pk, col1
1, 2
CSV

    [ "$status" -eq 1 ]
    [[ "$output" =~ "An error occurred while moving data" ]] || false
    [[ "$output" =~ "row [1,1] would be overwritten by [1,2]" ]] || false

    run dolt sql -q "select * from t"
    [ "$status" -eq 0 ]
    [[   "$output" =~ "| 1  | 1    |" ]] || false
    [[ ! "$output" =~ "| 1  | 2    |" ]] || false
    [ "${#lines[@]}" -eq 5 ]
}

@test "import-append-tables: disallow multiple keys with different values during append" {
    dolt sql -q "CREATE TABLE t (pk int primary key, col1 int);"
    run dolt table import -a t <<CSV
pk, col1
1, 1
1, 2
CSV

    [ "$status" -eq 1 ]
    [[ "$output" =~ "An error occurred while moving data" ]] || false
    [[ "$output" =~ "row [1,1] would be overwritten by [1,2]" ]] || false

    run dolt sql -q "select * from t"
    [ "$status" -eq 0 ]
    [[ ! "$output" =~ "| 1  | 1    |" ]] || false
    [[ ! "$output" =~ "| 1  | 2    |" ]] || false
    [ "${#lines[@]}" -eq 0 ]
}

@test "import-append-tables: ignore rows that would have no effect on import" {
    dolt sql -q "CREATE TABLE t (pk int primary key, col1 int);"
    dolt table import -a t <<CSV
pk, col1
1, 1
CSV
    run dolt table import -a t --continue <<CSV
pk, col1
1, 1
2, 3
CSV

    [ "$status" -eq 0 ]
    [[ "$output" =~ "The following rows were skipped:" ]] || false
    [[ "$output" =~ "[1,1]" ]] || false
    [[ "$output" =~ "Rows Processed: 1, Additions: 1, Modifications: 0, Had No Effect: 0" ]] || false
    [[ "$output" =~ "Lines skipped: 1" ]] || false

    run dolt sql -q "select * from t"
    [ "$status" -eq 0 ]
    [[ "$output" =~ "| 1  | 1    |" ]] || false
    [[ "$output" =~ "| 2  | 3    |" ]] || false
    [ "${#lines[@]}" -eq 6 ]
}

@test "import-append-tables: reject rows in source that would modify rows in destination, but continue if --continue is supplied" {
    dolt sql -q "CREATE TABLE t (pk int primary key, col1 int);"
    dolt table import -a t <<CSV
pk, col1
1, 1
CSV
    run dolt table import -a t --continue <<CSV
pk, col1
1, 2
2, 3
CSV

    [ "$status" -eq 0 ]
    [[ "$output" =~ "The following rows were skipped:" ]] || false
    [[ "$output" =~ "[1,2]" ]] || false
    [[ "$output" =~ "Rows Processed: 1, Additions: 1, Modifications: 0, Had No Effect: 0" ]] || false
    [[ "$output" =~ "Lines skipped: 1" ]] || false

    run dolt sql -q "select * from t"
    [ "$status" -eq 0 ]
    [[   "$output" =~ "| 1  | 1    |" ]] || false
    [[   "$output" =~ "| 2  | 3    |" ]] || false
    [[ ! "$output" =~ "| 1  | 2    |" ]] || false
    [ "${#lines[@]}" -eq 6 ]
}

@test "import-append-tables: import error message contains useful information" {
    dolt sql -q "CREATE TABLE shirts (name VARCHAR(40), size ENUM('x-small', 'small', 'medium', 'large', 'x-large'), color ENUM('red', 'blue'));"
    run dolt table import -a shirts <<CSV
name, size, color
"shirt1", "x-small", "red"
"shirt2", "other", "green"
CSV
    [ "$status" -eq 1 ]
    [[ "$output" =~ "An error occurred while moving data" ]] || false
    [[ "$output" =~ "cause: value other is not valid for this Enum" ]] || false
    [[ "$output" =~ "A bad row was encountered inserting into table shirts (on line 3):" ]] || false    # table name
    [[ "$output" =~ "name: shirt2" ]] || false                                              # column names
    [[ "$output" =~ "size: other" ]] || false
    [[ "$output" =~ "color: green" ]] || false
    [[ "$output" =~ "Errors during import can be ignored using '--continue'" ]] || false
}

@test "import-append-tables: different schema warning lists differing columns" {
    dolt sql -q "CREATE TABLE t (pk int primary key, col1 int);"
    run dolt table import -a t <<CSV
pk, col2
1, 1
CSV
    [ "$status" -eq 0 ]
    [[ "$output" =~ "Warning: The import file's schema does not match the table's schema" ]] || false
    [[ "$output" =~ "If unintentional, check for any typos in the import file's header" ]] || false
    [[ "$output" =~ "Missing columns in t:" ]] || false
    [[ "$output" =~ "	col1" ]] || false
    [[ "$output" =~ "Extra columns in import file:" ]] || false
    [[ "$output" =~ "	col2" ]] || false
}

@test "import-append-tables: can't use --all-text with -a" {
    dolt sql -q "CREATE TABLE t (pk int primary key, col1 int);"
    run dolt table import -a t --all-text <<CSV
pk, col1
1, 1
CSV
    [ "$status" -eq 1 ]
    [[ "$output" =~ "fatal: --all-text is only supported for create operations" ]] || false
}