georm/tests/simple_struct.rs

use georm::Georm;
use rand::seq::SliceRandom;

use models::Author;
mod models;

#[sqlx::test(fixtures("simple_struct"))]
async fn find_all_query_works(pool: sqlx::PgPool) -> sqlx::Result<()> {
    let result = Author::find_all(&pool).await?;
    assert_eq!(3, result.len());
    Ok(())
}

#[sqlx::test]
async fn find_all_returns_empty_vec_on_empty_table(pool: sqlx::PgPool) -> sqlx::Result<()> {
    let result = Author::find_all(&pool).await?;
    assert_eq!(0, result.len());
    Ok(())
}

#[sqlx::test(fixtures("simple_struct"))]
async fn find_query_works(pool: sqlx::PgPool) -> sqlx::Result<()> {
    let id = 1;
    let res = Author::find(&pool, &id).await?;
    assert!(res.is_some());
    let res = res.unwrap();
    assert_eq!(String::from("J.R.R. Tolkien"), res.name);
    assert_eq!(1, res.id);
    Ok(())
}

#[sqlx::test]
async fn find_returns_none_if_not_found(pool: sqlx::PgPool) -> sqlx::Result<()> {
    let res = Author::find(&pool, &420).await?;
    assert!(res.is_none());
    Ok(())
}

#[sqlx::test]
async fn create_works(pool: sqlx::PgPool) -> sqlx::Result<()> {
    let author = Author {
        id: 1,
        name: "J.R.R. Tolkien".into(),
        ..Default::default()
    };
    author.create(&pool).await?;
    let all_authors = Author::find_all(&pool).await?;
    assert_eq!(1, all_authors.len());
    assert_eq!(vec![author], all_authors);
    Ok(())
}

#[sqlx::test(fixtures("simple_struct"))]
async fn create_fails_if_already_exists(pool: sqlx::PgPool) -> sqlx::Result<()> {
    let author = Author {
        id: 2,
        name: "Miura Kentaro".into(),
        ..Default::default()
    };
    let result = author.create(&pool).await;
    assert!(result.is_err());
    let error = result.err().unwrap();
    assert_eq!("error returned from database: duplicate key value violates unique constraint \"authors_pkey\"", error.to_string());
    Ok(())
}

#[sqlx::test(fixtures("simple_struct"))]
async fn update_works(pool: sqlx::PgPool) -> sqlx::Result<()> {
    let expected_initial = Author {
        name: "J.R.R. Tolkien".into(),
        id: 1,
        biography_id: Some(2),
    };
    let expected_final = Author {
        name: "Jolkien Rolkien Rolkien Tolkien".into(),
        id: 1,
        biography_id: Some(2),
    };
    let tolkien = Author::find(&pool, &1).await?;
    assert!(tolkien.is_some());
    let mut tolkien = tolkien.unwrap();
    assert_eq!(expected_initial, tolkien);
    tolkien.name = expected_final.name.clone();
    let updated = tolkien.update(&pool).await?;
    assert_eq!(expected_final, updated);
    Ok(())
}

#[sqlx::test]
async fn update_fails_if_not_already_exists(pool: sqlx::PgPool) -> sqlx::Result<()> {
    let author = Author {
        id: 2,
        name: "Miura Kentaro".into(),
        ..Default::default()
    };
    let result = author.update(&pool).await;
    assert!(result.is_err());
    let error = result.err().unwrap();
    assert_eq!(
        "no rows returned by a query that expected to return at least one row",
        error.to_string()
    );
    Ok(())
}

#[sqlx::test]
async fn should_create_if_does_not_exist(pool: sqlx::PgPool) -> sqlx::Result<()> {
    let all_authors = Author::find_all(&pool).await?;
    assert_eq!(0, all_authors.len());
    let author = Author {
        id: 4,
        name: "Miura Kentaro".into(),
        ..Default::default()
    };
    author.create_or_update(&pool).await?;
    let all_authors = Author::find_all(&pool).await?;
    assert_eq!(1, all_authors.len());
    Ok(())
}

#[sqlx::test(fixtures("simple_struct"))]
async fn should_update_if_exist(pool: sqlx::PgPool) -> sqlx::Result<()> {
    let all_authors = Author::find_all(&pool).await?;
    assert_eq!(3, all_authors.len());
    let author = Author {
        id: 2,
        name: "Miura Kentaro".into(),
        ..Default::default()
    };
    author.create_or_update(&pool).await?;
    let mut all_authors = Author::find_all(&pool).await?;
    all_authors.sort();
    assert_eq!(3, all_authors.len());
    assert_eq!(author, all_authors[1]);
    Ok(())
}

#[sqlx::test(fixtures("simple_struct"))]
async fn delete_by_id_should_delete_only_one_entry(pool: sqlx::PgPool) -> sqlx::Result<()> {
    let id = 2;
    let all_authors = Author::find_all(&pool).await?;
    assert_eq!(3, all_authors.len());
    assert!(all_authors.iter().any(|author| author.get_id() == &id));
    let result = Author::delete_by_id(&pool, &id).await?;
    assert_eq!(1, result);
    let all_authors = Author::find_all(&pool).await?;
    assert_eq!(2, all_authors.len());
    assert!(all_authors.iter().all(|author| author.get_id() != &id));
    Ok(())
}

#[sqlx::test(fixtures("simple_struct"))]
async fn delete_should_delete_current_entity_from_db(pool: sqlx::PgPool) -> sqlx::Result<()> {
    let mut all_authors = Author::find_all(&pool).await?;
    assert_eq!(3, all_authors.len());
    all_authors.shuffle(&mut rand::rng());
    let author = all_authors.first().unwrap();
    let result = author.delete(&pool).await?;
    assert_eq!(1, result);
    let all_authors = Author::find_all(&pool).await?;
    assert_eq!(2, all_authors.len());
    assert!(all_authors.iter().all(|a| a.get_id() != author.get_id()));
    Ok(())
}