ormer 0.2.13

#![cfg(feature = "sqlite")]

pub mod _test_common;

use ormer::PageCursor;

define_test_user_with_score!(CursorUser, "cursor_pagination_users_1");

#[tokio::test]
async fn sqlite_cursor_pagination_is_stable_across_inserts()
-> Result<(), Box<dyn std::error::Error>> {
    let db = _test_common::create_db_connection(&_test_common::sqlite_config()).await?;

    let _ = db.drop_table::<CursorUser>().execute().await;
    db.create_table::<CursorUser>().execute().await?;

    for user in [
        CursorUser {
            id: 0,
            name: "Alice".to_string(),
            age: 20,
            score: 100,
        },
        CursorUser {
            id: 0,
            name: "Bob".to_string(),
            age: 21,
            score: 90,
        },
        CursorUser {
            id: 0,
            name: "Carol".to_string(),
            age: 22,
            score: 90,
        },
        CursorUser {
            id: 0,
            name: "Dave".to_string(),
            age: 23,
            score: 80,
        },
        CursorUser {
            id: 0,
            name: "Eve".to_string(),
            age: 24,
            score: 70,
        },
    ] {
        db.insert(&user).execute().await?;
    }

    let page1 = db
        .select::<CursorUser>()
        .order_by_desc(|u| u.score)
        .order_by_desc(|u| u.id)
        .cursor_by(|u| (u.score, u.id))
        .limit(2)
        .fetch_page()
        .await?;

    assert_eq!(
        page1
            .items
            .iter()
            .map(|u| u.name.as_str())
            .collect::<Vec<_>>(),
        vec!["Alice", "Carol"]
    );

    let cursor = page1
        .next_cursor()
        .expect("first page should yield a cursor");

    db.insert(&CursorUser {
        id: 0,
        name: "Frank".to_string(),
        age: 25,
        score: 95,
    })
    .execute()
    .await?;

    let page2 = db
        .select::<CursorUser>()
        .order_by_desc(|u| u.score)
        .order_by_desc(|u| u.id)
        .cursor_by(|u| (u.score, u.id))
        .after(PageCursor::from(&cursor))
        .limit(2)
        .fetch_page()
        .await?;

    assert_eq!(
        page2
            .items
            .iter()
            .map(|u| u.name.as_str())
            .collect::<Vec<_>>(),
        vec!["Bob", "Dave"]
    );
    assert!(page2.next_cursor().is_some());

    let _ = db.drop_table::<CursorUser>().execute().await;

    Ok(())
}

#[tokio::test]
async fn sqlite_cursor_pagination_before_returns_closest_previous_rows()
-> Result<(), Box<dyn std::error::Error>> {
    let db = _test_common::create_db_connection(&_test_common::sqlite_config()).await?;

    let _ = db.drop_table::<CursorUser>().execute().await;
    db.create_table::<CursorUser>().execute().await?;

    for user in [
        CursorUser {
            id: 0,
            name: "Alice".to_string(),
            age: 20,
            score: 100,
        },
        CursorUser {
            id: 0,
            name: "Bob".to_string(),
            age: 21,
            score: 90,
        },
        CursorUser {
            id: 0,
            name: "Carol".to_string(),
            age: 22,
            score: 90,
        },
        CursorUser {
            id: 0,
            name: "Dave".to_string(),
            age: 23,
            score: 80,
        },
        CursorUser {
            id: 0,
            name: "Eve".to_string(),
            age: 24,
            score: 70,
        },
    ] {
        db.insert(&user).execute().await?;
    }

    // score DESC, id DESC 排序：Alice(100), Carol(90,id3), Bob(90,id2), Dave(80), Eve(70)
    let later_page = db
        .select::<CursorUser>()
        .order_by_desc(|u| u.score)
        .order_by_desc(|u| u.id)
        .cursor_by(|u| (u.score, u.id))
        .filter(|u| u.name.eq("Dave"))
        .fetch_page()
        .await?;
    let dave_cursor = later_page
        .next_cursor()
        .expect("single row page should yield a cursor");

    // Dave 之前有 4 行，取最近的 2 行（Frank 场景外的 Carol 与 Bob），
    // 旧行为会取回离 Dave 最远的 Alice 与 Carol
    let prev_page = db
        .select::<CursorUser>()
        .order_by_desc(|u| u.score)
        .order_by_desc(|u| u.id)
        .cursor_by(|u| (u.score, u.id))
        .before(PageCursor::from(&dave_cursor))
        .limit(2)
        .fetch_page()
        .await?;

    assert_eq!(
        prev_page
            .items
            .iter()
            .map(|u| u.name.as_str())
            .collect::<Vec<_>>(),
        vec!["Carol", "Bob"]
    );

    // 继续向前翻页应取到再往前最近的行（Alice）
    let first_page = db
        .select::<CursorUser>()
        .order_by_desc(|u| u.score)
        .order_by_desc(|u| u.id)
        .cursor_by(|u| (u.score, u.id))
        .before(prev_page.next_cursor().unwrap())
        .limit(2)
        .fetch_page()
        .await?;
    assert_eq!(
        first_page
            .items
            .iter()
            .map(|u| u.name.as_str())
            .collect::<Vec<_>>(),
        vec!["Alice"]
    );

    let _ = db.drop_table::<CursorUser>().execute().await;

    Ok(())
}