Page MenuHomePhabricator
Files F3492311

D4252.id13480.diff
No OneTemporary
Actions

D4252.id13480.diff
View Options

diff --git a/services/commtest/tests/backup_test.rs b/services/commtest/tests/backup_test.rs
--- a/services/commtest/tests/backup_test.rs
+++ b/services/commtest/tests/backup_test.rs
@@ -1,10 +1,126 @@
+#[path = "./backup/add_attachments.rs"]
+mod add_attachments;
+#[path = "./backup/backup_utils.rs"]
+mod backup_utils;
+#[path = "./backup/create_new_backup.rs"]
+mod create_new_backup;
+#[path = "./backup/pull_backup.rs"]
+mod pull_backup;
+#[path = "./backup/send_log.rs"]
+mod send_log;
#[path = "./lib/tools.rs"]
mod tools;
+use backup_utils::{BackupData, Item};
+use bytesize::ByteSize;
+use tools::get_grpc_chunk_size_limit;
use tools::Error;
+use backup_utils::BackupServiceClient;
+
#[tokio::test]
async fn backup_test() -> Result<(), Error> {
- assert!(false, "not implemented");
+ let mut client =
+ BackupServiceClient::connect("http://localhost:50052").await?;
+
+ let mut backup_data = BackupData {
+ user_id: "user0000".to_string(),
+ device_id: "device0000".to_string(),
+ backup_item: Item::new(
+ String::new(),
+ vec![ByteSize::mib(1).as_u64() as usize; 6],
+ vec![
+ "holder1".to_string(),
+ "holder2".to_string(),
+ "holder3".to_string(),
+ ],
+ ),
+ log_items: vec![
+ Item::new(String::new(), vec![100], vec!["holder1".to_string()]),
+ Item::new(
+ String::new(),
+ vec![ByteSize::kb(400).as_u64() as usize],
+ vec!["holder2".to_string(), "holder3".to_string()],
+ ),
+ Item::new(
+ String::new(),
+ vec![get_grpc_chunk_size_limit(), get_grpc_chunk_size_limit()],
+ vec![
+ "holder1".to_string(),
+ "holder2".to_string(),
+ "holder3".to_string(),
+ ],
+ ),
+ ],
+ };
+ backup_data.backup_item.id =
+ create_new_backup::run(&mut client, &backup_data).await?;
+ println!("backup id in main: {}", backup_data.backup_item.id);
+
+ add_attachments::run(&mut client, &backup_data, None).await?;
+
+ for log_index in 0..backup_data.log_items.len() {
+ backup_data.log_items[log_index].id =
+ send_log::run(&mut client, &backup_data, log_index).await?;
+ add_attachments::run(&mut client, &backup_data, Some(log_index)).await?;
+ }
+
+ let result = pull_backup::run(&mut client, &backup_data).await?;
+
+ // check backup size
+ let expected: usize = backup_data.backup_item.chunks_sizes.iter().sum();
+ let from_result: usize = result.backup_item.chunks_sizes.iter().sum();
+ assert!(
+ from_result == expected,
+ "backup sizes do not match, expected {}, got {}",
+ expected,
+ from_result
+ );
+
+ // check backup attachments
+ let expected: usize = backup_data.backup_item.attachments_holders.len();
+ let from_result: usize = result.backup_item.attachments_holders.len();
+ assert!(
+ from_result == expected,
+ "backup: number of attachments holders do not match, expected {}, got {}",
+ expected,
+ from_result
+ );
+
+ // check number of logs
+ let expected: usize = backup_data.log_items.len();
+ let from_result: usize = result.log_items.len();
+ assert!(
+ expected == from_result,
+ "number of logs do not match, expected {}, got {}",
+ expected,
+ from_result
+ );
+
+ // check log sizes
+ for i in 0..backup_data.log_items.len() {
+ let expected: usize = backup_data.log_items[i].chunks_sizes.iter().sum();
+ let from_result: usize = result.log_items[i].chunks_sizes.iter().sum();
+ assert!(
+ from_result == expected,
+ "log number {} sizes do not match, expected {}, got {}",
+ i,
+ expected,
+ from_result
+ );
+ }
+ // check logs attachments
+ for i in 0..backup_data.log_items.len() {
+ let expected: usize = backup_data.log_items[i].attachments_holders.len();
+ let from_result: usize = result.log_items[i].attachments_holders.len();
+ assert!(
+ from_result == expected,
+ "log {}: number of attachments holders do not match, expected {}, got {}",
+ i,
+ expected,
+ from_result
+ );
+ }
+
Ok(())
}

File Metadata

Mime Type
text/plain
Expires
Thu, Dec 19, 11:02 PM (21 h, 11 m)
Storage Engine
blob
Storage Format
Raw Data
Storage Handle
2679137
Default Alt Text
D4252.id13480.diff (3 KB)

Event Timeline