AppFlowy/shared-lib/lib-ot/tests/node/operation_delete_test.rs

use crate::node::script::NodeScript::*;
use crate::node::script::NodeTest;

use lib_ot::core::{Changeset, NodeData, NodeDataBuilder};

#[test]
fn operation_delete_nested_node_test() {
    let mut test = NodeTest::new();
    let image_a = NodeData::new("image_a");
    let image_b = NodeData::new("image_b");

    let video_a = NodeData::new("video_a");
    let video_b = NodeData::new("video_b");

    let image_1 = NodeDataBuilder::new("image_1")
        .add_node_data(image_a.clone())
        .add_node_data(image_b.clone())
        .build();
    let video_1 = NodeDataBuilder::new("video_1")
        .add_node_data(video_a.clone())
        .add_node_data(video_b)
        .build();

    let text_node_1 = NodeDataBuilder::new("text_1")
        .add_node_data(image_1)
        .add_node_data(video_1.clone())
        .build();

    let image_2 = NodeDataBuilder::new("image_2")
        .add_node_data(image_a)
        .add_node_data(image_b.clone())
        .build();
    let text_node_2 = NodeDataBuilder::new("text_2").add_node_data(image_2).build();

    let scripts = vec![
        InsertNode {
            path: 0.into(),
            node_data: text_node_1,
            rev_id: 1,
        },
        InsertNode {
            path: 1.into(),
            node_data: text_node_2,
            rev_id: 2,
        },
        // 0:text_1
        //      0:image_1
        //             0:image_a
        //             1:image_b
        //      1:video_1
        //             0:video_a
        //             1:video_b
        // 1:text_2
        //      0:image_2
        //             0:image_a
        //             1:image_b
        DeleteNode {
            path: vec![0, 0, 0].into(),
            rev_id: 3,
        },
        AssertNode {
            path: vec![0, 0, 0].into(),
            expected: Some(image_b),
        },
        AssertNode {
            path: vec![0, 1].into(),
            expected: Some(video_1),
        },
        DeleteNode {
            path: vec![0, 1, 1].into(),
            rev_id: 4,
        },
        AssertNode {
            path: vec![0, 1, 0].into(),
            expected: Some(video_a),
        },
    ];
    test.run_scripts(scripts);
}

#[test]
fn operation_delete_node_with_revision_conflict_test() {
    let mut test = NodeTest::new();
    let text_1 = NodeDataBuilder::new("text_1").build();
    let text_2 = NodeDataBuilder::new("text_2").build();
    let text_3 = NodeDataBuilder::new("text_3").build();

    let scripts = vec![
        InsertNode {
            path: 0.into(),
            node_data: text_1.clone(),
            rev_id: 1,
        },
        InsertNode {
            path: 1.into(),
            node_data: text_2,
            rev_id: 2,
        },
        // The node's in the tree will be:
        // 0: text_1
        // 2: text_2
        //
        // The insert action is happened concurrently with the delete action, because they
        // share the same rev_id. aka, 3. The delete action is want to delete the node at index 1,
        // but it was moved to index 2.
        InsertNode {
            path: 1.into(),
            node_data: text_3.clone(),
            rev_id: 3,
        },
        // 0: text_1
        // 1: text_3
        // 2: text_2
        //
        // The path of the delete action will be transformed to a new path that point to the text_2.
        // 1 -> 2
        DeleteNode {
            path: 1.into(),
            rev_id: 3,
        },
        // After perform the delete action, the tree will be:
        // 0: text_1
        // 1: text_3
        AssertNumberOfChildrenAtPath {
            path: None,
            expected: 2,
        },
        AssertNode {
            path: 0.into(),
            expected: Some(text_1),
        },
        AssertNode {
            path: 1.into(),
            expected: Some(text_3),
        },
        AssertNode {
            path: 2.into(),
            expected: None,
        },
    ];
    test.run_scripts(scripts);
}

#[test]
fn operation_update_node_after_delete_test() {
    let mut test = NodeTest::new();
    let text_1 = NodeDataBuilder::new("text_1").build();
    let text_2 = NodeDataBuilder::new("text_2").build();
    let scripts = vec![
        InsertNode {
            path: 0.into(),
            node_data: text_1,
            rev_id: 1,
        },
        InsertNode {
            path: 1.into(),
            node_data: text_2,
            rev_id: 2,
        },
        DeleteNode {
            path: 0.into(),
            rev_id: 3,
        },
        // The node at path 1 is not exist. The following UpdateBody script will do nothing
        AssertNode {
            path: 1.into(),
            expected: None,
        },
        UpdateBody {
            path: 1.into(),
            changeset: Changeset::Delta {
                delta: Default::default(),
                inverted: Default::default(),
            },
        },
    ];
    test.run_scripts(scripts);
}