clawmates/crates/cm-db/tests/repos.rs

use std::str::FromStr;

use cm_db::repo::{agents, audit, credits, users, workspaces};
use cm_db::DbError;
use cm_domain::{
    AccessPolicy, Agent, AgentId, AgentScope, AgentStatus, HumanScope, Role, User, UserId,
    Workspace, WorkspaceId,
};

fn workspace() -> Workspace {
    Workspace {
        id: WorkspaceId::new(),
        name: "Acme".into(),
        plan: "team".into(),
    }
}

fn user_in(ws: &Workspace, role: Role) -> User {
    User {
        id: UserId::new(),
        workspace_id: ws.id,
        email: format!("{}@acme.test", UserId::new()),
        role,
        display_name: "Test User".into(),
        created_at: time::OffsetDateTime::UNIX_EPOCH,
    }
}

fn agent_in(ws: &Workspace, owner: &User) -> Agent {
    Agent {
        id: AgentId::new(),
        workspace_id: ws.id,
        name: "Scout".into(),
        job_title: "Research Analyst".into(),
        system_prompt: "You research things.".into(),
        avatar: "scout-1".into(),
        accent: "#f96565".into(),
        wallpaper: "dunes".into(),
        managed_by: owner.id,
        status: AgentStatus::Provisioning,
    }
}

#[tokio::test]
async fn workspace_round_trips() {
    let pool = cm_testkit::test_pool().await;
    let ws = workspace();
    workspaces::insert(&pool, &ws).await.unwrap();
    let found = workspaces::get(&pool, ws.id).await.unwrap();
    assert_eq!(found, ws);
}

#[tokio::test]
async fn missing_workspace_is_not_found() {
    let pool = cm_testkit::test_pool().await;
    let err = workspaces::get(&pool, WorkspaceId::new())
        .await
        .unwrap_err();
    assert!(matches!(err, DbError::NotFound));
}

#[tokio::test]
async fn user_round_trips_and_finds_by_email() {
    let pool = cm_testkit::test_pool().await;
    let ws = workspace();
    workspaces::insert(&pool, &ws).await.unwrap();
    let user = user_in(&ws, Role::Owner);
    users::insert(&pool, &user).await.unwrap();

    let by_id = users::get(&pool, user.id).await.unwrap();
    assert_eq!(by_id.email, user.email);
    assert_eq!(by_id.role, Role::Owner);

    let by_email = users::find_by_email(&pool, &user.email).await.unwrap();
    assert_eq!(by_email.id, user.id);
}

#[tokio::test]
async fn duplicate_email_is_a_conflict() {
    let pool = cm_testkit::test_pool().await;
    let ws = workspace();
    workspaces::insert(&pool, &ws).await.unwrap();
    let mut a = user_in(&ws, Role::Member);
    let mut b = user_in(&ws, Role::Member);
    b.email = a.email.clone();
    users::insert(&pool, &a).await.unwrap();
    let err = users::insert(&pool, &b).await.unwrap_err();
    assert!(matches!(err, DbError::Conflict(_)));
    // Silence unused warnings for fields we only compare implicitly.
    a.display_name.clear();
}

#[tokio::test]
async fn workspace_members_lists_in_join_order() {
    let pool = cm_testkit::test_pool().await;
    let ws = workspace();
    workspaces::insert(&pool, &ws).await.unwrap();
    let owner = user_in(&ws, Role::Owner);
    let member = user_in(&ws, Role::Member);
    users::insert(&pool, &owner).await.unwrap();
    users::insert(&pool, &member).await.unwrap();

    let members = users::list_by_workspace(&pool, ws.id).await.unwrap();
    assert_eq!(members.len(), 2);
    assert_eq!(members[0].id, owner.id);
    assert_eq!(members[1].id, member.id);
}

#[tokio::test]
async fn agent_insert_creates_default_access_policy() {
    let pool = cm_testkit::test_pool().await;
    let ws = workspace();
    workspaces::insert(&pool, &ws).await.unwrap();
    let owner = user_in(&ws, Role::Owner);
    users::insert(&pool, &owner).await.unwrap();
    let agent = agent_in(&ws, &owner);

    agents::insert(&pool, &agent, &AccessPolicy::default())
        .await
        .unwrap();

    let policy = agents::access_policy(&pool, agent.id).await.unwrap();
    assert_eq!(policy.humans, HumanScope::EntireTeam);
    assert_eq!(policy.agents, AgentScope::Any);
}

#[tokio::test]
async fn agent_roster_excludes_deleted_and_round_trips_fields() {
    let pool = cm_testkit::test_pool().await;
    let ws = workspace();
    workspaces::insert(&pool, &ws).await.unwrap();
    let owner = user_in(&ws, Role::Owner);
    users::insert(&pool, &owner).await.unwrap();

    let keep = agent_in(&ws, &owner);
    let remove = agent_in(&ws, &owner);
    agents::insert(&pool, &keep, &AccessPolicy::default())
        .await
        .unwrap();
    agents::insert(&pool, &remove, &AccessPolicy::default())
        .await
        .unwrap();
    agents::soft_delete(&pool, remove.id).await.unwrap();

    let roster = agents::roster(&pool, ws.id).await.unwrap();
    assert_eq!(roster.len(), 1);
    assert_eq!(roster[0], keep);
}

#[tokio::test]
async fn agent_status_updates() {
    let pool = cm_testkit::test_pool().await;
    let ws = workspace();
    workspaces::insert(&pool, &ws).await.unwrap();
    let owner = user_in(&ws, Role::Owner);
    users::insert(&pool, &owner).await.unwrap();
    let agent = agent_in(&ws, &owner);
    agents::insert(&pool, &agent, &AccessPolicy::default())
        .await
        .unwrap();

    agents::set_status(&pool, agent.id, AgentStatus::Online)
        .await
        .unwrap();
    let roster = agents::roster(&pool, ws.id).await.unwrap();
    assert_eq!(roster[0].status, AgentStatus::Online);
}

#[tokio::test]
async fn specific_access_policy_round_trips() {
    let pool = cm_testkit::test_pool().await;
    let ws = workspace();
    workspaces::insert(&pool, &ws).await.unwrap();
    let owner = user_in(&ws, Role::Owner);
    users::insert(&pool, &owner).await.unwrap();
    let agent = agent_in(&ws, &owner);

    let policy = AccessPolicy {
        humans: HumanScope::Specific(vec![owner.id]),
        agents: AgentScope::Specific(vec![AgentId::from_str(&agent.id.to_string()).unwrap()]),
    };
    agents::insert(&pool, &agent, &policy).await.unwrap();
    let stored = agents::access_policy(&pool, agent.id).await.unwrap();
    assert_eq!(stored, policy);
}

#[tokio::test]
async fn credit_balance_sums_remaining_lots() {
    let pool = cm_testkit::test_pool().await;
    let ws = workspace();
    workspaces::insert(&pool, &ws).await.unwrap();

    assert_eq!(credits::balance(&pool, ws.id).await.unwrap(), 0);
    credits::add_lot(&pool, ws.id, 1000, "purchase")
        .await
        .unwrap();
    credits::add_lot(&pool, ws.id, 250, "promo").await.unwrap();
    assert_eq!(credits::balance(&pool, ws.id).await.unwrap(), 1250);
}

#[tokio::test]
async fn audit_log_appends_and_rejects_mutation() {
    let pool = cm_testkit::test_pool().await;
    let ws = workspace();
    workspaces::insert(&pool, &ws).await.unwrap();

    let entry_id = audit::append(
        &pool,
        ws.id,
        audit::Actor::System,
        "workspace.created",
        "workspace",
        &ws.id.to_string(),
        serde_json::json!({"plan": "team"}),
    )
    .await
    .unwrap();
    assert!(entry_id > 0);

    // Append-only is enforced by the database itself, not convention.
    let update = sqlx::query("UPDATE audit_log SET event_type = 'tampered' WHERE id = $1")
        .bind(entry_id)
        .execute(&pool)
        .await;
    assert!(update.is_err());
    let delete = sqlx::query("DELETE FROM audit_log WHERE id = $1")
        .bind(entry_id)
        .execute(&pool)
        .await;
    assert!(delete.is_err());
}