fraiseql-wire 2.3.2

Streaming JSON query engine for Postgres 17
Documentation
#![allow(clippy::panic)] // Reason: test code, panics acceptable
use super::*;
use crate::util::oid::JSON_OID;

#[test]
fn test_valid_row_description() {
    let field = FieldDescription {
        name: "data".to_string(),
        table_oid: 0,
        column_attr: 0,
        type_oid: JSON_OID,
        type_size: -1,
        type_modifier: -1,
        format_code: 0,
    };

    let msg = BackendMessage::RowDescription(vec![field]);
    validate_row_description(&msg)
        .unwrap_or_else(|e| panic!("expected Ok for valid RowDescription: {e}"));
}

#[test]
fn test_wrong_column_name() {
    let field = FieldDescription {
        name: "wrong".to_string(),
        table_oid: 0,
        column_attr: 0,
        type_oid: JSON_OID,
        type_size: -1,
        type_modifier: -1,
        format_code: 0,
    };

    let msg = BackendMessage::RowDescription(vec![field]);
    let result = validate_row_description(&msg);
    assert!(
        matches!(result, Err(WireError::InvalidSchema(_))),
        "expected InvalidSchema error for wrong column name, got: {result:?}"
    );
}

#[test]
fn test_wrong_type() {
    let field = FieldDescription {
        name: "data".to_string(),
        table_oid: 0,
        column_attr: 0,
        type_oid: 23, // INT4
        type_size: 4,
        type_modifier: -1,
        format_code: 0,
    };

    let msg = BackendMessage::RowDescription(vec![field]);
    let result = validate_row_description(&msg);
    assert!(
        matches!(result, Err(WireError::InvalidSchema(_))),
        "expected InvalidSchema error for wrong type OID, got: {result:?}"
    );
}

#[test]
fn test_multiple_columns() {
    let field1 = FieldDescription {
        name: "data".to_string(),
        table_oid: 0,
        column_attr: 0,
        type_oid: JSON_OID,
        type_size: -1,
        type_modifier: -1,
        format_code: 0,
    };
    let field2 = field1.clone();

    let msg = BackendMessage::RowDescription(vec![field1, field2]);
    let result = validate_row_description(&msg);
    assert!(
        matches!(result, Err(WireError::InvalidSchema(_))),
        "expected InvalidSchema error for multiple columns, got: {result:?}"
    );
}