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_global_slug_uniqueness() { let temp_dir = std::env::temp_dir().join(format!("bzod_test_slug_uniq_{}", 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 urls_conn = db.global_urls.lock().unwrap(); let reserved_conn = db.reserved.lock().unwrap(); let pages_conn = db.global_landing_pages.lock().unwrap(); // Register a slug let now = chrono::Utc::now().to_rfc3339(); urls_conn .execute( "INSERT INTO global_urls (slug, owner_tenant_id, target_id, created_at, updated_at, status, retired_at) VALUES ('!myslug', 't-tenant1234', 'url1', ?1, ?2, 'active', NULL);", rusqlite::params![now, now], ) .unwrap(); // Verify it is not available let avail = bzod::db::slugs::is_slug_available(&reserved_conn, &urls_conn, &pages_conn, "!myslug") .unwrap(); assert!(!avail); let _ = fs::remove_dir_all(&temp_dir); } #[tokio::test] async fn test_reserved_slug_rejection() { let temp_dir = std::env::temp_dir().join(format!("bzod_test_reserved_slug_{}", 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 urls_conn = db.global_urls.lock().unwrap(); let reserved_conn = db.reserved.lock().unwrap(); let pages_conn = db.global_landing_pages.lock().unwrap(); let reserved = vec![ "admin", "login", "logout", "dashboard", "api", "docs", "assets", "static", "favicon.ico", "robots.txt", "health", "metrics", "install", "setup", "support", "help", "security", "abuse", "billing", "status", "legacy_admin", "administrator", "system", "root", "www", ]; for slug in reserved { let avail = bzod::db::slugs::is_slug_available(&reserved_conn, &urls_conn, &pages_conn, slug) .unwrap(); assert!(!avail, "Reserved slug '{}' should not be available", slug); } let _ = fs::remove_dir_all(&temp_dir); } #[tokio::test] async fn test_slug_release_on_user_deletion() { let temp_dir = std::env::temp_dir().join(format!("bzod_test_slug_release_{}", 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, tid) = { let conn = db.users.lock().unwrap(); let u = bzod::db::users::get_user_by_username(&conn, "testuser") .unwrap() .unwrap(); (u.id, u.tenant_id.unwrap()) }; // Register slug for this user { let urls_conn = db.global_urls.lock().unwrap(); let reserved_conn = db.reserved.lock().unwrap(); let pages_conn = db.global_landing_pages.lock().unwrap(); let now = chrono::Utc::now().to_rfc3339(); urls_conn .execute( "INSERT INTO global_urls (slug, owner_tenant_id, target_id, created_at, updated_at, status, retired_at) VALUES ('!user-slug', ?1, 'url_xyz', ?2, ?3, 'active', NULL);", rusqlite::params![tid.as_str(), now, now], ) .unwrap(); let avail = bzod::db::slugs::is_slug_available( &reserved_conn, &urls_conn, &pages_conn, "!user-slug", ) .unwrap(); assert!(!avail); } // Delete user bzod::cli::delete_user::run(user_id, false, None, config.clone()) .await .unwrap(); // Slug should now be released and available { let urls_conn = db.global_urls.lock().unwrap(); let reserved_conn = db.reserved.lock().unwrap(); let pages_conn = db.global_landing_pages.lock().unwrap(); let avail = bzod::db::slugs::is_slug_available( &reserved_conn, &urls_conn, &pages_conn, "!user-slug", ) .unwrap(); assert!(avail); // Verify history populated let system_conn = db.system.lock().unwrap(); let count: i64 = system_conn .query_row( "SELECT COUNT(*) FROM slug_history WHERE slug = ?1 AND action = 'deleted';", ["!user-slug"], |row| row.get(0), ) .unwrap(); assert!(count > 0); } let _ = fs::remove_dir_all(&temp_dir); }