ormer 0.1.21

#![cfg(any(feature = "sqlite", feature = "postgresql", feature = "mysql"))]

mod _test_common;

// 使用宏定义测试专用模型（唯一表名）
define_test_user!(ExecTestUser1, "exec_test_users_1");
define_test_user!(ExecTestUser2, "exec_test_users_2");
define_test_user!(ExecTestUser3, "exec_test_users_3");
define_test_user!(ExecTestUser4, "exec_test_users_4");

/// 测试 execute 方法 - 执行原生 SQL 查询并返回模型列表
async fn test_execute_impl(
    config: &_test_common::DbConfig,
) -> Result<(), Box<dyn std::error::Error>> {
    // 连接到数据库
    let db = _test_common::create_db_connection(config).await?;

    // 先清理可能存在的旧表
    let _ = db.drop_table::<ExecTestUser1>().execute().await;

    // 创建表
    db.create_table::<ExecTestUser1>().execute().await?;

    // 插入测试数据
    db.insert(&ExecTestUser1 {
        id: 1,
        name: "Alice".to_string(),
        age: 25,
        email: Some("alice@example.com".to_string()),
    })
    .execute()
    .await?;

    db.insert(&ExecTestUser1 {
        id: 2,
        name: "Bob".to_string(),
        age: 30,
        email: Some("bob@example.com".to_string()),
    })
    .execute()
    .await?;

    db.insert(&ExecTestUser1 {
        id: 3,
        name: "Charlie".to_string(),
        age: 35,
        email: None,
    })
    .execute()
    .await?;

    // 测试 execute - 查询所有用户
    let users = db
        .execute::<ExecTestUser1>("SELECT * FROM exec_test_users_1;")
        .await?;

    assert_eq!(users.len(), 3);
    assert_eq!(users[0].name, "Alice");
    assert_eq!(users[1].name, "Bob");
    assert_eq!(users[2].name, "Charlie");

    // 测试 execute - 带 WHERE 条件的查询
    let young_users = db
        .execute::<ExecTestUser1>("SELECT * FROM exec_test_users_1 WHERE age < 30;")
        .await?;

    assert_eq!(young_users.len(), 1);
    assert_eq!(young_users[0].name, "Alice");
    assert_eq!(young_users[0].age, 25);

    // 测试 execute - 空结果
    let old_users = db
        .execute::<ExecTestUser1>("SELECT * FROM exec_test_users_1 WHERE age > 100;")
        .await?;

    assert_eq!(old_users.len(), 0);

    // 测试 execute - 排序查询
    let sorted_users = db
        .execute::<ExecTestUser1>("SELECT * FROM exec_test_users_1 ORDER BY age DESC;")
        .await?;

    assert_eq!(sorted_users.len(), 3);
    assert_eq!(sorted_users[0].name, "Charlie");
    assert_eq!(sorted_users[0].age, 35);
    assert_eq!(sorted_users[1].name, "Bob");
    assert_eq!(sorted_users[1].age, 30);
    assert_eq!(sorted_users[2].name, "Alice");
    assert_eq!(sorted_users[2].age, 25);

    // 清理测试表
    db.drop_table::<ExecTestUser1>().execute().await?;

    Ok(())
}

/// 测试 exec_non_query 方法 - 执行原生非查询 SQL 并返回受影响行数
async fn test_exec_non_query_impl(
    config: &_test_common::DbConfig,
) -> Result<(), Box<dyn std::error::Error>> {
    // 连接到数据库
    let db = _test_common::create_db_connection(config).await?;

    // 先清理可能存在的旧表
    let _ = db.drop_table::<ExecTestUser2>().execute().await;

    // 创建表
    db.create_table::<ExecTestUser2>().execute().await?;

    // 插入测试数据
    db.insert(&ExecTestUser2 {
        id: 1,
        name: "Alice".to_string(),
        age: 25,
        email: Some("alice@example.com".to_string()),
    })
    .execute()
    .await?;

    db.insert(&ExecTestUser2 {
        id: 2,
        name: "Bob".to_string(),
        age: 30,
        email: Some("bob@example.com".to_string()),
    })
    .execute()
    .await?;

    db.insert(&ExecTestUser2 {
        id: 3,
        name: "Charlie".to_string(),
        age: 35,
        email: None,
    })
    .execute()
    .await?;

    // 测试 exec_non_query - UPDATE 语句
    let updated_rows = db
        .exec_non_query("UPDATE exec_test_users_2 SET age = 40 WHERE age >= 30;")
        .await?;

    assert_eq!(updated_rows, 2); // Bob 和 Charlie 的年龄被更新

    // 验证更新结果
    let users = db
        .execute::<ExecTestUser2>("SELECT * FROM exec_test_users_2 ORDER BY id;")
        .await?;

    assert_eq!(users[0].age, 25); // Alice 未变
    assert_eq!(users[1].age, 40); // Bob 已更新
    assert_eq!(users[2].age, 40); // Charlie 已更新

    // 测试 exec_non_query - DELETE 语句
    let deleted_rows = db
        .exec_non_query("DELETE FROM exec_test_users_2 WHERE age < 30;")
        .await?;

    println!("Deleted rows: {}", deleted_rows);
    // 在运行多个测试后，表中可能有残留数据，所以我们只验证至少删除了1行
    assert!(deleted_rows >= 1); // 至少删除 Alice

    // 验证删除结果
    let users = db
        .execute::<ExecTestUser2>("SELECT * FROM exec_test_users_2;")
        .await?;

    assert_eq!(users.len(), 2); // 只剩下 Bob 和 Charlie

    // 测试 exec_non_query - INSERT 语句
    let inserted_rows = db
        .exec_non_query(
            "INSERT INTO exec_test_users_2 (id, name, age, email) VALUES (4, 'David', 28, 'david@example.com');",
        )
        .await?;

    assert_eq!(inserted_rows, 1); // 插入了一行

    // 验证插入结果
    let users = db
        .execute::<ExecTestUser2>("SELECT * FROM exec_test_users_2;")
        .await?;

    assert_eq!(users.len(), 3); // 现在有 3 个用户

    // 测试 exec_non_query - 不影响任何行的 UPDATE
    let updated_rows = db
        .exec_non_query("UPDATE exec_test_users_2 SET age = 99 WHERE age > 200;")
        .await?;

    assert_eq!(updated_rows, 0); // 没有符合条件的行

    // 清理测试表
    db.drop_table::<ExecTestUser2>().execute().await?;

    Ok(())
}

/// 测试 execute 和 exec_non_query 的组合使用
async fn test_execute_and_non_query_combined_impl(
    config: &_test_common::DbConfig,
) -> Result<(), Box<dyn std::error::Error>> {
    // 连接到数据库
    let db = _test_common::create_db_connection(config).await?;

    // 先清理可能存在的旧表
    let _ = db.drop_table::<ExecTestUser3>().execute().await;

    // 创建表
    db.create_table::<ExecTestUser3>().execute().await?;

    // 使用 exec_non_query 插入数据
    db.exec_non_query(
        "INSERT INTO exec_test_users_3 (id, name, age, email) VALUES 
         (1, 'Alice', 25, 'alice@example.com'),
         (2, 'Bob', 30, 'bob@example.com'),
         (3, 'Charlie', 35, NULL);",
    )
    .await?;

    // 使用 execute 查询数据
    let users = db
        .execute::<ExecTestUser3>("SELECT * FROM exec_test_users_3 ORDER BY age;")
        .await?;

    assert_eq!(users.len(), 3);
    assert_eq!(users[0].name, "Alice");
    assert_eq!(users[0].age, 25);
    assert_eq!(users[1].name, "Bob");
    assert_eq!(users[1].age, 30);
    assert_eq!(users[2].name, "Charlie");
    assert_eq!(users[2].age, 35);

    // 使用 exec_non_query 批量更新
    db.exec_non_query("UPDATE exec_test_users_3 SET age = age + 5;")
        .await?;

    // 验证批量更新
    let users = db
        .execute::<ExecTestUser3>("SELECT * FROM exec_test_users_3 ORDER BY age;")
        .await?;

    assert_eq!(users[0].age, 30);
    assert_eq!(users[1].age, 35);
    assert_eq!(users[2].age, 40);

    // 使用 exec_non_query 删除所有数据
    let deleted = db.exec_non_query("DELETE FROM exec_test_users_3;").await?;

    println!("Deleted all rows: {}", deleted);
    // 验证至少删除了3行（可能有残留数据）
    assert!(deleted >= 3);

    // 验证表为空
    let users = db
        .execute::<ExecTestUser3>("SELECT * FROM exec_test_users_3;")
        .await?;

    assert_eq!(users.len(), 0);

    // 清理测试表（表已经被 DELETE 清空，但仍需删除表结构）
    db.drop_table::<ExecTestUser3>().execute().await?;

    Ok(())
}

/// 测试 execute 处理 NULL 值
async fn test_execute_with_null_values_impl(
    config: &_test_common::DbConfig,
) -> Result<(), Box<dyn std::error::Error>> {
    // 连接到数据库
    let db = _test_common::create_db_connection(config).await?;

    // 先清理可能存在的旧表
    let _ = db.drop_table::<ExecTestUser4>().execute().await;

    // 创建表
    db.create_table::<ExecTestUser4>().execute().await?;

    // 插入包含 NULL 值的数据
    db.exec_non_query(
        "INSERT INTO exec_test_users_4 (id, name, age, email) VALUES 
         (1, 'Alice', 25, NULL),
         (2, 'Bob', 30, 'bob@example.com');",
    )
    .await?;

    // 查询并验证 NULL 值处理
    let users = db
        .execute::<ExecTestUser4>("SELECT * FROM exec_test_users_4 ORDER BY id;")
        .await?;

    assert_eq!(users.len(), 2);
    assert_eq!(users[0].name, "Alice");
    assert_eq!(users[0].email, None); // NULL 应该被转换为 None
    assert_eq!(users[1].name, "Bob");
    assert_eq!(users[1].email, Some("bob@example.com".to_string()));

    // 清理测试表
    db.drop_table::<ExecTestUser4>().execute().await?;

    Ok(())
}

test_on_all_dbs_result!(test_execute_impl);
test_on_all_dbs_result!(test_exec_non_query_impl);
test_on_all_dbs_result!(test_execute_and_non_query_combined_impl);
test_on_all_dbs_result!(test_execute_with_null_values_impl);