AppFlowy/frontend/rust-lib/flowy-document/tests/new_document/test.rs

use crate::new_document::script::DocumentEditorTest;
use crate::new_document::script::EditScript::*;

use lib_ot::text_delta::DeltaTextOperationBuilder;

#[tokio::test]
async fn document_initialize_test() {
  let scripts = vec![AssertContent {
    expected: r#"{"document":{"type":"editor","children":[{"type":"text"}]}}"#,
  }];
  DocumentEditorTest::new().await.run_scripts(scripts).await;
}

#[tokio::test]
async fn document_insert_text_test() {
  let delta = DeltaTextOperationBuilder::new()
    .insert("Hello world")
    .build();
  let expected = r#"{
  "document": {
    "type": "editor",
    "children": [
      {
        "type": "text",
        "delta": [
          {
            "insert": "Hello world"
          }
        ]
      },
      {
        "type": "text"
      }
    ]
  }
}"#;
  let scripts = vec![
    InsertText {
      path: vec![0, 0].into(),
      delta,
    },
    AssertPrettyContent { expected },
  ];
  DocumentEditorTest::new().await.run_scripts(scripts).await;
}

#[tokio::test]
async fn document_update_text_test() {
  let test = DocumentEditorTest::new().await;
  let hello_world = "Hello world".to_string();
  let scripts = vec![
    UpdateText {
      path: vec![0, 0].into(),
      delta: DeltaTextOperationBuilder::new()
        .insert(&hello_world)
        .build(),
    },
    AssertPrettyContent {
      expected: r#"{
  "document": {
    "type": "editor",
    "children": [
      {
        "type": "text",
        "delta": [
          {
            "insert": "Hello world"
          }
        ]
      }
    ]
  }
}"#,
    },
  ];

  test.run_scripts(scripts).await;

  let scripts = vec![
    UpdateText {
      path: vec![0, 0].into(),
      delta: DeltaTextOperationBuilder::new()
        .retain(hello_world.len())
        .insert(", AppFlowy")
        .build(),
    },
    AssertPrettyContent {
      expected: r#"{
  "document": {
    "type": "editor",
    "children": [
      {
        "type": "text",
        "delta": [
          {
            "insert": "Hello world, AppFlowy"
          }
        ]
      }
    ]
  }
}"#,
    },
  ];
  test.run_scripts(scripts).await;
}

#[tokio::test]
async fn document_delete_text_test() {
  let expected = r#"{
  "document": {
    "type": "editor",
    "children": [
      {
        "type": "text",
        "delta": [
          {
            "insert": "Hello"
          }
        ]
      }
    ]
  }
}"#;
  let hello_world = "Hello world".to_string();
  let scripts = vec![
    UpdateText {
      path: vec![0, 0].into(),
      delta: DeltaTextOperationBuilder::new()
        .insert(&hello_world)
        .build(),
    },
    UpdateText {
      path: vec![0, 0].into(),
      delta: DeltaTextOperationBuilder::new().retain(5).delete(6).build(),
    },
    AssertPrettyContent { expected },
  ];

  DocumentEditorTest::new().await.run_scripts(scripts).await;
}

#[tokio::test]
async fn document_delete_node_test() {
  let scripts = vec![
    UpdateText {
      path: vec![0, 0].into(),
      delta: DeltaTextOperationBuilder::new()
        .insert("Hello world")
        .build(),
    },
    AssertContent {
      expected: r#"{"document":{"type":"editor","children":[{"type":"text","delta":[{"insert":"Hello world"}]}]}}"#,
    },
    Delete {
      path: vec![0, 0].into(),
    },
    AssertContent {
      expected: r#"{"document":{"type":"editor"}}"#,
    },
  ];

  DocumentEditorTest::new().await.run_scripts(scripts).await;
}