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

tests/quota_tests.rs

@@ -0,0 +1,141 @@
+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_quota_limit_enforcement() {
+    let temp_dir =
+        std::env::temp_dir().join(format!("bzod_test_quota_limits_{}", 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
+    };
+
+    // Set max_urls to 1
+    {
+        let conn = db.users.lock().unwrap();
+        conn.execute(
+            "UPDATE quotas SET max_urls = 1 WHERE user_id = ?1;",
+            [user_id],
+        )
+        .unwrap();
+    }
+
+    // Attempt to add URLs and manually increment count
+    {
+        let users_conn = db.users.lock().unwrap();
+        let quota = bzod::db::users::get_user_quotas(&users_conn, user_id)
+            .unwrap()
+            .unwrap();
+        assert_eq!(quota.max_urls, 1);
+        assert_eq!(quota.current_urls, 0);
+
+        // 1st Url: Success
+        bzod::db::users::increment_quota_counter(&users_conn, user_id, "urls").unwrap();
+        let quota1 = bzod::db::users::get_user_quotas(&users_conn, user_id)
+            .unwrap()
+            .unwrap();
+        assert_eq!(quota1.current_urls, 1);
+
+        // 2nd Url: Quota check fails
+        let quota_check = quota1.current_urls >= quota1.max_urls;
+        assert!(quota_check);
+    }
+
+    let _ = fs::remove_dir_all(&temp_dir);
+}
+
+#[tokio::test]
+async fn test_quota_reconcile_job() {
+    let temp_dir = std::env::temp_dir().join(format!(
+        "bzod_test_quota_reconcile_{}",
+        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
+    };
+
+    // Introduce drift manually: make current_urls = 10, when actually 1 URL exists
+    {
+        let user_content_conn = bzod::jobs::open_user_content_conn(&db, user_id).unwrap();
+        bzod::db::content::create_url_extended(
+            &user_content_conn,
+            "!my-reconcile-slug",
+            "https://example.com",
+            None,
+            None,
+            &vec![],
+            None,
+            None,
+            None,
+        )
+        .unwrap();
+
+        let users_conn = db.users.lock().unwrap();
+        users_conn
+            .execute(
+                "UPDATE quotas SET current_urls = 10 WHERE user_id = ?1;",
+                [user_id],
+            )
+            .unwrap();
+    }
+
+    // Run reconciliation
+    {
+        let users_conn = db.users.lock().unwrap();
+        let user_content_conn = bzod::jobs::open_user_content_conn(&db, user_id).unwrap();
+        bzod::db::users::reconcile_user_quotas(&users_conn, user_id, &user_content_conn).unwrap();
+
+        // Verify drift is repaired
+        let quota = bzod::db::users::get_user_quotas(&users_conn, user_id)
+            .unwrap()
+            .unwrap();
+        assert_eq!(quota.current_urls, 1);
+    }
+
+    let _ = fs::remove_dir_all(&temp_dir);
+}