AppFlowy/frontend/rust-lib/flowy-revision/tests/revision_test/local_revision_test.rs

use crate::revision_test::script::{RevisionScript::*, RevisionTest};
use flowy_revision::REVISION_WRITE_INTERVAL_IN_MILLIS;

#[tokio::test]
async fn revision_sync_test() {
    let test = RevisionTest::new().await;
    let (base_rev_id, rev_id) = test.next_rev_id_pair();

    test.run_script(AddLocalRevision {
        content: "123".to_string(),
        base_rev_id,
        rev_id,
    })
    .await;

    test.run_script(AssertNextSyncRevisionId { rev_id: Some(rev_id) }).await;
    test.run_script(AckRevision { rev_id }).await;
    test.run_script(AssertNextSyncRevisionId { rev_id: None }).await;
}

#[tokio::test]
async fn revision_sync_multiple_revisions() {
    let test = RevisionTest::new_with_configuration(2).await;
    let (base_rev_id, rev_id_1) = test.next_rev_id_pair();

    test.run_script(AddLocalRevision {
        content: "123".to_string(),
        base_rev_id,
        rev_id: rev_id_1,
    })
    .await;

    let (base_rev_id, rev_id_2) = test.next_rev_id_pair();
    test.run_script(AddLocalRevision {
        content: "456".to_string(),
        base_rev_id,
        rev_id: rev_id_2,
    })
    .await;

    test.run_scripts(vec![
        AssertNextSyncRevisionId { rev_id: Some(rev_id_1) },
        AckRevision { rev_id: rev_id_1 },
        AssertNextSyncRevisionId { rev_id: Some(rev_id_2) },
        AckRevision { rev_id: rev_id_2 },
        AssertNextSyncRevisionId { rev_id: None },
    ])
    .await;
}

#[tokio::test]
async fn revision_compress_three_revisions_test() {
    let test = RevisionTest::new_with_configuration(2).await;
    let (base_rev_id, rev_id_1) = test.next_rev_id_pair();

    test.run_script(AddLocalRevision {
        content: "1".to_string(),
        base_rev_id,
        rev_id: rev_id_1,
    })
    .await;

    let (base_rev_id, rev_id_2) = test.next_rev_id_pair();
    test.run_script(AddLocalRevision {
        content: "2".to_string(),
        base_rev_id,
        rev_id: rev_id_2,
    })
    .await;

    let (base_rev_id, rev_id_3) = test.next_rev_id_pair();
    test.run_script(AddLocalRevision {
        content: "3".to_string(),
        base_rev_id,
        rev_id: rev_id_3,
    })
    .await;

    let (base_rev_id, rev_id_4) = test.next_rev_id_pair();
    test.run_script(AddLocalRevision {
        content: "4".to_string(),
        base_rev_id,
        rev_id: rev_id_4,
    })
    .await;

    // rev_id_2,rev_id_3,rev_id4 will be merged into rev_id_1
    test.run_scripts(vec![
        Wait {
            milliseconds: REVISION_WRITE_INTERVAL_IN_MILLIS,
        },
        AssertNumberOfSyncRevisions { num: 1 },
        AssertNextSyncRevisionId { rev_id: Some(rev_id_1) },
        AssertNextSyncRevisionContent {
            expected: "1234".to_string(),
        },
        AckRevision { rev_id: rev_id_1 },
        AssertNextSyncRevisionId { rev_id: None },
    ])
    .await;
}

#[tokio::test]
async fn revision_compress_three_revisions_test2() {
    let test = RevisionTest::new_with_configuration(2).await;
    let (base_rev_id, rev_id_1) = test.next_rev_id_pair();

    test.run_script(AddLocalRevision {
        content: "1".to_string(),
        base_rev_id,
        rev_id: rev_id_1,
    })
    .await;

    let (base_rev_id, rev_id_2) = test.next_rev_id_pair();
    test.run_script(AddLocalRevision {
        content: "2".to_string(),
        base_rev_id,
        rev_id: rev_id_2,
    })
    .await;

    let (base_rev_id, rev_id_3) = test.next_rev_id_pair();
    test.run_script(AddLocalRevision {
        content: "3".to_string(),
        base_rev_id,
        rev_id: rev_id_3,
    })
    .await;

    let (base_rev_id, rev_id_4) = test.next_rev_id_pair();
    test.run_script(AddLocalRevision {
        content: "4".to_string(),
        base_rev_id,
        rev_id: rev_id_4,
    })
    .await;

    let (base_rev_id, rev_id_a) = test.next_rev_id_pair();
    test.run_script(AddLocalRevision {
        content: "a".to_string(),
        base_rev_id,
        rev_id: rev_id_a,
    })
    .await;

    let (base_rev_id, rev_id_b) = test.next_rev_id_pair();
    test.run_script(AddLocalRevision {
        content: "b".to_string(),
        base_rev_id,
        rev_id: rev_id_b,
    })
    .await;

    let (base_rev_id, rev_id_c) = test.next_rev_id_pair();
    test.run_script(AddLocalRevision {
        content: "c".to_string(),
        base_rev_id,
        rev_id: rev_id_c,
    })
    .await;

    let (base_rev_id, rev_id_d) = test.next_rev_id_pair();
    test.run_script(AddLocalRevision {
        content: "d".to_string(),
        base_rev_id,
        rev_id: rev_id_d,
    })
    .await;

    test.run_scripts(vec![
        // Wait {
        //     milliseconds: REVISION_WRITE_INTERVAL_IN_MILLIS,
        // },
        AssertNumberOfSyncRevisions { num: 2 },
        AssertNextSyncRevisionId { rev_id: Some(rev_id_1) },
        AssertNextSyncRevisionContent {
            expected: "1234".to_string(),
        },
        AckRevision { rev_id: rev_id_1 },
        AssertNextSyncRevisionId { rev_id: Some(rev_id_a) },
        AssertNextSyncRevisionContent {
            expected: "abcd".to_string(),
        },
        AckRevision { rev_id: rev_id_a },
        AssertNextSyncRevisionId { rev_id: None },
    ])
    .await;
}

#[tokio::test]
async fn revision_merge_per_5_revision_test() {
    let test = RevisionTest::new_with_configuration(4).await;
    for i in 0..20 {
        let content = format!("{}", i);
        let (base_rev_id, rev_id) = test.next_rev_id_pair();
        test.run_script(AddLocalRevision {
            content,
            base_rev_id,
            rev_id,
        })
        .await;
    }

    test.run_scripts(vec![AssertNumberOfSyncRevisions { num: 5 }]).await;
}