use crate::edit_test::mock::mock_test_data;
use crate::util::{check_doc_state_json, redis_stream, run_test_server};
use collab_entity::CollabType;
use tonic_proto::history::SnapshotRequest;

#[tokio::test]
async fn apply_update_stream_updates_test() {
  let redis_stream = redis_stream().await;
  let workspace_id = uuid::Uuid::new_v4().to_string();
  let object_id = uuid::Uuid::new_v4().to_string();
  let mock = mock_test_data(&workspace_id, &object_id, 30).await;
  let control_stream_key = uuid::Uuid::new_v4().to_string();
  let client = run_test_server(control_stream_key).await;

  let control_stream_key = client.config.stream_settings.control_key.clone();
  let mut control_group = redis_stream
    .collab_control_stream(&control_stream_key, "appflowy_cloud")
    .await
    .unwrap();

  // apply open event
  control_group
    .insert_message(mock.open_event.clone())
    .await
    .unwrap();

  let mut update_group = redis_stream
    .collab_update_stream(&workspace_id, &object_id, "appflowy_cloud")
    .await
    .unwrap();

  // apply updates
  for update_event in &mock.update_events {
    update_group
      .insert_message(update_event.clone())
      .await
      .unwrap();
  }

  let request = SnapshotRequest {
    workspace_id: workspace_id.to_string(),
    object_id: object_id.to_string(),
    collab_type: CollabType::Unknown.value(),
  };

  check_doc_state_json(&object_id, 60, mock.expected_json.clone(), move || {
    let mut cloned_client = client.clone();
    let cloned_request = request.clone();
    Box::pin(async move {
      cloned_client
        .get_in_memory_history(cloned_request)
        .await
        .map(|r| r.into_inner())
    })
  })
  .await
  .unwrap();
}

#[tokio::test]
async fn apply_missing_updates_test() {
  let redis_stream = redis_stream().await;
  let workspace_id = uuid::Uuid::new_v4().to_string();
  let object_id = uuid::Uuid::new_v4().to_string();
  let mock = mock_test_data(&workspace_id, &object_id, 30).await;
  let client = run_test_server(uuid::Uuid::new_v4().to_string()).await;

  let control_stream_key = client.config.stream_settings.control_key.clone();
  let mut control_group = redis_stream
    .collab_control_stream(&control_stream_key, "appflowy_cloud")
    .await
    .unwrap();

  // apply open event
  control_group
    .insert_message(mock.open_event.clone())
    .await
    .unwrap();

  let mut update_group = redis_stream
    .collab_update_stream(&workspace_id, &object_id, "appflowy_cloud")
    .await
    .unwrap();

  let mut missing_updates = vec![];
  // apply updates
  for (index, update_event) in mock.update_events.iter().enumerate() {
    if index % 2 == 0 {
      missing_updates.push(update_event.clone());
    } else {
      update_group
        .insert_message(update_event.clone())
        .await
        .unwrap();
    }
  }

  for update_event in missing_updates {
    update_group.insert_message(update_event).await.unwrap();
  }

  let request = SnapshotRequest {
    workspace_id: workspace_id.to_string(),
    object_id: object_id.to_string(),
    collab_type: CollabType::Unknown.value(),
  };

  check_doc_state_json(&object_id, 30, mock.expected_json.clone(), move || {
    let mut cloned_client = client.clone();
    let cloned_request = request.clone();
    Box::pin(async move {
      cloned_client
        .get_in_memory_history(cloned_request)
        .await
        .map(|r| r.into_inner())
    })
  })
  .await
  .unwrap();
}