116 lines
2.7 KiB
Rust
116 lines
2.7 KiB
Rust
use crate::client::utils::REGISTERED_USERS_MUTEX;
|
|
use crate::collab::workspace_id_from_client;
|
|
use crate::user_1_signed_in;
|
|
|
|
use collab_define::CollabType;
|
|
use shared_entity::error_code::ErrorCode;
|
|
use sqlx::types::Uuid;
|
|
use storage_entity::{DeleteCollabParams, InsertCollabParams, QueryCollabParams};
|
|
|
|
#[tokio::test]
|
|
async fn success_insert_collab_test() {
|
|
let _guard = REGISTERED_USERS_MUTEX.lock().await;
|
|
|
|
let mut c = user_1_signed_in().await;
|
|
let raw_data = "hello world".to_string().as_bytes().to_vec();
|
|
let workspace_id = workspace_id_from_client(&mut c).await;
|
|
let object_id = Uuid::new_v4().to_string();
|
|
c.create_collab(InsertCollabParams::new(
|
|
1,
|
|
&object_id,
|
|
CollabType::Document,
|
|
raw_data.clone(),
|
|
workspace_id,
|
|
))
|
|
.await
|
|
.unwrap();
|
|
|
|
let bytes = c
|
|
.get_collab(QueryCollabParams {
|
|
object_id,
|
|
collab_type: CollabType::Document,
|
|
})
|
|
.await
|
|
.unwrap();
|
|
|
|
assert_eq!(bytes, raw_data);
|
|
}
|
|
|
|
#[tokio::test]
|
|
async fn success_delete_collab_test() {
|
|
let _guard = REGISTERED_USERS_MUTEX.lock().await;
|
|
|
|
let mut c = user_1_signed_in().await;
|
|
|
|
let raw_data = "hello world".to_string().as_bytes().to_vec();
|
|
let workspace_id = workspace_id_from_client(&mut c).await;
|
|
let object_id = Uuid::new_v4().to_string();
|
|
c.create_collab(InsertCollabParams::new(
|
|
1,
|
|
object_id.clone(),
|
|
CollabType::Document,
|
|
raw_data.clone(),
|
|
workspace_id,
|
|
))
|
|
.await
|
|
.unwrap();
|
|
|
|
c.delete_collab(DeleteCollabParams {
|
|
object_id: object_id.clone(),
|
|
})
|
|
.await
|
|
.unwrap();
|
|
|
|
let error = c
|
|
.get_collab(QueryCollabParams {
|
|
object_id,
|
|
collab_type: CollabType::Document,
|
|
})
|
|
.await
|
|
.unwrap_err();
|
|
|
|
assert_eq!(error.code, ErrorCode::RecordNotFound);
|
|
}
|
|
|
|
#[tokio::test]
|
|
async fn fail_insert_collab_with_empty_payload_test() {
|
|
let _guard = REGISTERED_USERS_MUTEX.lock().await;
|
|
|
|
let mut c = user_1_signed_in().await;
|
|
let workspace_id = workspace_id_from_client(&mut c).await;
|
|
let error = c
|
|
.create_collab(InsertCollabParams::new(
|
|
1,
|
|
Uuid::new_v4().to_string(),
|
|
CollabType::Document,
|
|
vec![],
|
|
workspace_id,
|
|
))
|
|
.await
|
|
.unwrap_err();
|
|
|
|
assert_eq!(error.code, ErrorCode::StorageError);
|
|
}
|
|
|
|
#[tokio::test]
|
|
async fn fail_insert_collab_with_invalid_workspace_id_test() {
|
|
let _guard = REGISTERED_USERS_MUTEX.lock().await;
|
|
|
|
let mut c = user_1_signed_in().await;
|
|
|
|
let workspace_id = Uuid::new_v4().to_string();
|
|
let raw_data = "hello world".to_string().as_bytes().to_vec();
|
|
let error = c
|
|
.create_collab(InsertCollabParams::new(
|
|
1,
|
|
Uuid::new_v4().to_string(),
|
|
CollabType::Document,
|
|
raw_data.clone(),
|
|
workspace_id,
|
|
))
|
|
.await
|
|
.unwrap_err();
|
|
|
|
assert_eq!(error.code, ErrorCode::StorageError);
|
|
}
|