Function test_parse_error_codes

src/environmentd/tests/sql.rs:2353–2407

()

#[mz_ore::test]
fn test_parse_error_codes() {
    // Parse/cast and overflow failures should report data-exception (class 22)
    // SQLSTATEs rather than the catch-all XX000 (INTERNAL_ERROR). Regression
    // test for the fix tracked in SQL-326.
    let server = test_util::TestHarness::default().start_blocking();
    let mut client = server.connect(postgres::NoTls).unwrap();

    let cases: &[(&str, &SqlState)] = &[
        // Out-of-range datetime value -> datetime_field_overflow (22008).
        (
            "SELECT '14714-12-31 18:01:00+00 BC'::date",
            &SqlState::DATETIME_FIELD_OVERFLOW,
        ),
        // Malformed datetime string -> invalid_datetime_format (22007).
        (
            "SELECT 'not a date'::date",
            &SqlState::INVALID_DATETIME_FORMAT,
        ),
        // Out-of-range float value -> numeric_value_out_of_range (22003).
        (
            "SELECT '1e400'::float8",
            &SqlState::NUMERIC_VALUE_OUT_OF_RANGE,
        ),
        // Malformed integer string -> invalid_text_representation (22P02).
        (
            "SELECT 'abc'::integer",
            &SqlState::INVALID_TEXT_REPRESENTATION,
        ),
        // Division by zero -> division_by_zero (22012). Previously XX000.
        ("SELECT 1 / 0", &SqlState::DIVISION_BY_ZERO),
        // Square root of a negative -> invalid_argument_for_power_function
        // (2201F). Previously XX000.
        (
            "SELECT sqrt(-1.0)",
            &SqlState::INVALID_ARGUMENT_FOR_POWER_FUNCTION,
        ),
        // More than one row from a scalar subquery -> cardinality_violation
        // (21000). Previously XX000.
        (
            "SELECT (SELECT generate_series(1, 2))",
            &SqlState::CARDINALITY_VIOLATION,
        ),
    ];

    for (query, expected) in cases {
        let err = client.query_one(*query, &[]).unwrap_err().unwrap_db_error();
        assert_eq!(
            err.code(),
            *expected,
            "unexpected SQLSTATE {} for query `{query}`: {}",
            err.code().code(),
            err.message(),
        );
    }
}

#[mz_ore::test]
#[allow(clippy::disallowed_methods)]