104 lines
2.7 KiB
Rust
104 lines
2.7 KiB
Rust
use bzod::config::Config;
|
|
use bzod::db::Db;
|
|
use std::fs;
|
|
use std::path::PathBuf;
|
|
|
|
fn create_temp_config(temp_dir: PathBuf) -> Config {
|
|
let mut config = Config::load_with_cli(Some(&temp_dir)).unwrap();
|
|
config.backup_dir = temp_dir.clone();
|
|
config.base_url = Some("http://bzo.in".to_string());
|
|
config
|
|
}
|
|
|
|
#[tokio::test]
|
|
async fn test_wal_recovery_after_backup() {
|
|
let temp_dir = std::env::temp_dir().join(format!("bzod_test_wal_rec_{}", uuid::Uuid::new_v4()));
|
|
fs::create_dir_all(&temp_dir).unwrap();
|
|
let config = create_temp_config(temp_dir.clone());
|
|
|
|
let db = Db::init(&config).expect("Failed to init Db");
|
|
|
|
let _ = bzod::cli::create_user::run(
|
|
Some("testuser".to_string()),
|
|
Some("password123".to_string()),
|
|
None,
|
|
config.clone(),
|
|
)
|
|
.await
|
|
.unwrap();
|
|
|
|
let user_id = {
|
|
let conn = db.users.lock().unwrap();
|
|
bzod::db::users::get_user_by_username(&conn, "testuser")
|
|
.unwrap()
|
|
.unwrap()
|
|
.id
|
|
};
|
|
|
|
// 1. Write some content
|
|
{
|
|
let conn = bzod::jobs::open_user_content_conn(&db, user_id).unwrap();
|
|
bzod::db::content::create_url_extended(
|
|
&conn,
|
|
"!wal-slug",
|
|
"https://google.com/wal",
|
|
None,
|
|
None,
|
|
&vec![],
|
|
None,
|
|
None,
|
|
None,
|
|
)
|
|
.unwrap();
|
|
}
|
|
|
|
// 2. Perform backup (this checkpoints and flushes WAL to DB files)
|
|
let backup_file = temp_dir.join("testuser-backup.tar.zst");
|
|
bzod::cli::backup_user::run(
|
|
"testuser".to_string(),
|
|
Some(backup_file.to_string_lossy().to_string()),
|
|
None,
|
|
config.clone(),
|
|
)
|
|
.await
|
|
.unwrap();
|
|
|
|
// 3. Restore user
|
|
bzod::cli::restore_user::run(
|
|
backup_file.to_string_lossy().to_string(),
|
|
None,
|
|
config.clone(),
|
|
)
|
|
.await
|
|
.unwrap();
|
|
|
|
// 4. Verify that restored databases are consistent and can successfully perform WAL writes
|
|
{
|
|
let conn = bzod::jobs::open_user_content_conn(&db, user_id).unwrap();
|
|
|
|
// SQLite integrity check
|
|
bzod::db::sqlite::integrity_check(&conn, "content").unwrap();
|
|
|
|
// Write a new url after restore to verify WAL write works
|
|
bzod::db::content::create_url_extended(
|
|
&conn,
|
|
"!new-wal-slug",
|
|
"https://google.com/new-wal",
|
|
None,
|
|
None,
|
|
&vec![],
|
|
None,
|
|
None,
|
|
None,
|
|
)
|
|
.unwrap();
|
|
|
|
let url = bzod::db::content::get_url_by_code(&conn, "!new-wal-slug")
|
|
.unwrap()
|
|
.unwrap();
|
|
assert_eq!(url.destination, "https://google.com/new-wal");
|
|
}
|
|
|
|
let _ = fs::remove_dir_all(&temp_dir);
|
|
}
|