nx9-url-shortener/tests/slug_namespace_tests.rs

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);
}