test_util.rs

use diesel::prelude::*;

use crate::models::{Crate, NewTeam, Team};
use crates_io_test_utils::github::next_gh_id;

pub mod faker {
    use super::*;
    use anyhow::anyhow;
    use crates_io_test_utils::builders::{CrateBuilder, UserBuilder};
    use diesel_async::AsyncPgConnection;

    pub async fn crate_and_version(
        conn: &mut AsyncPgConnection,
        name: &str,
        description: &str,
        user_id: i32,
        downloads: i32,
    ) -> anyhow::Result<Crate> {
        CrateBuilder::new(name, user_id)
            .description(description)
            .downloads(downloads)
            .version("1.0.0")
            .build(conn)
            .await
            .map_err(|err| anyhow!(err.to_string()))
    }

    pub async fn team(conn: &mut AsyncPgConnection, org: &str, team: &str) -> anyhow::Result<Team> {
        let login = format!("github:{org}:{team}");
        let team = NewTeam::builder()
            .login(&login)
            .org_id(next_gh_id())
            .github_id(next_gh_id())
            .name(team)
            .build();

        Ok(team.create_or_update(conn).await?)
    }

    pub async fn user(conn: &mut AsyncPgConnection, login: &str) -> QueryResult<i32> {
        UserBuilder::new()
            .with_username(login)
            .new_user()
            .insert(conn)
            .await
    }
}