Function setup_test_db

tests/implicit_cast_test.rs:7–57

()

use sqlparser::dialect::PostgreSqlDialect;

fn setup_test_db() -> Connection {
    let conn = Connection::open_in_memory().unwrap();
    
    // Initialize metadata table
    TypeMetadata::init(&conn).unwrap();
    
    // Register decimal functions
    pgsqlite::functions::register_all_functions(&conn).unwrap();
    
    // Create test tables
    conn.execute(
        "CREATE TABLE products (
            id INTEGER PRIMARY KEY,
            name TEXT,
            price TEXT,
            quantity INTEGER,
            category_id INTEGER
        )",
        [],
    ).unwrap();
    
    // Insert metadata for NUMERIC columns
    conn.execute(
        "INSERT INTO __pgsqlite_schema (table_name, column_name, pg_type, sqlite_type) VALUES
         ('products', 'id', 'INT4', 'INTEGER'),
         ('products', 'price', 'NUMERIC', 'DECIMAL'),
         ('products', 'quantity', 'INT4', 'INTEGER'),
         ('products', 'category_id', 'INT4', 'INTEGER')",
        [],
    ).unwrap();
    
    conn.execute(
        "CREATE TABLE orders (
            id INTEGER PRIMARY KEY,
            product_id INTEGER,
            amount TEXT,
            discount_pct TEXT
        )",
        [],
    ).unwrap();
    
    conn.execute(
        "INSERT INTO __pgsqlite_schema (table_name, column_name, pg_type, sqlite_type) VALUES
         ('orders', 'product_id', 'INT4', 'INTEGER'),
         ('orders', 'amount', 'NUMERIC', 'DECIMAL'),
         ('orders', 'discount_pct', 'NUMERIC', 'DECIMAL')",
        [],
    ).unwrap();
    
    conn
}

fn rewrite_query(conn: &Connection, sql: &str) -> Result<String, String> {
    let dialect = PostgreSqlDialect {};