Function test_information_schema_columns_basic

tests/information_schema_test.rs:140–215

()

#[tokio::test]
async fn test_information_schema_columns_basic() {
    let _ = env_logger::builder().is_test(true).try_init();

    let server = common::setup_test_server_with_init(|db| {
        Box::pin(async move {
            // Create a test table with various column types
            db.execute(r#"
                CREATE TABLE test_table (
                    id INTEGER PRIMARY KEY,
                    name VARCHAR(100) NOT NULL,
                    age INT,
                    salary DECIMAL(10,2),
                    is_active BOOLEAN DEFAULT true,
                    created_at TIMESTAMP
                )
            "#).await?;
            Ok(())
        })
    }).await;
    let client = &server.client;

    // Query information_schema.columns
    let rows = client.query(
        "SELECT table_name, column_name, data_type, is_nullable, column_default FROM information_schema.columns WHERE table_name = 'test_table' ORDER BY ordinal_position",
        &[]
    ).await.unwrap();

    // Verify we got the expected number of columns
    assert_eq!(rows.len(), 6, "Should have 6 columns");

    // Check that we get the expected column names and types
    let mut found_columns = std::collections::HashSet::new();
    for row in &rows {
        let table_name: &str = row.get(0);
        let column_name: &str = row.get(1);
        let data_type: &str = row.get(2);
        let is_nullable: &str = row.get(3);

        assert_eq!(table_name, "test_table");
        found_columns.insert(column_name.to_string());

        match column_name {
            "id" => {
                assert_eq!(data_type, "integer");
                assert_eq!(is_nullable, "NO"); // PRIMARY KEY is NOT NULL
            },
            "name" => {
                assert_eq!(data_type, "character varying");
                assert_eq!(is_nullable, "NO"); // Explicitly NOT NULL
            },
            "age" => {
                assert_eq!(data_type, "integer");
                assert_eq!(is_nullable, "YES");
            },
            "salary" => {
                assert_eq!(data_type, "numeric");
                assert_eq!(is_nullable, "YES");
            },