BZOD v0.5.0 RC2: multi-user platform, dashboards, analytics, backups and validation

tests/wal_recovery_tests.rs

@@ -0,0 +1,104 @@
+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();
+    config.data_dir = temp_dir.clone();
+    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);
+}