toasty-driver-integration-suite 0.5.0

//! Test linking and unlinking has_many associations

use crate::prelude::*;

#[driver_test(id(ID))]
pub async fn remove_add_single_relation_option_belongs_to(test: &mut Test) -> Result<()> {
    #[derive(Debug, toasty::Model)]
    struct User {
        #[key]
        #[auto]
        id: ID,

        #[has_many]
        todos: toasty::HasMany<Todo>,
    }

    #[derive(Debug, toasty::Model)]
    struct Todo {
        #[key]
        #[auto]
        id: ID,

        #[index]
        user_id: Option<ID>,

        #[belongs_to(key = user_id, references = id)]
        user: toasty::BelongsTo<Option<User>>,
    }

    let mut db = test.setup_db(models!(User, Todo)).await;

    // Create a user with some todos
    let user = User::create()
        .todo(Todo::create())
        .todo(Todo::create())
        .exec(&mut db)
        .await?;

    let todos: Vec<_> = user.todos().exec(&mut db).await?;
    assert_eq!(2, todos.len());

    // Remove a todo from the list.
    user.todos().remove(&mut db, &todos[0]).await?;

    let todos_reloaded: Vec<_> = user.todos().exec(&mut db).await?;
    assert_eq!(1, todos_reloaded.len());
    assert_eq!(todos[1].id, todos_reloaded[0].id);

    // Can't find the TODO via a scoped query either
    assert_err!(user.todos().get_by_id(&mut db, &todos[0].id).await);

    // The todo is not deleted, but user is now None
    let todo = Todo::get_by_id(&mut db, todos[0].id).await?;
    assert_none!(todo.user_id);

    // Create a second user w/ a TODO. We will ensure that unlinking *only*
    // unlinks records currently associated with the base model.
    let u2 = User::create().todo(Todo::create()).exec(&mut db).await?;
    let u2_todos = u2.todos().exec(&mut db).await?;

    // Try unlinking u2's todo via user. This should fail
    assert_err!(user.todos().remove(&mut db, &u2_todos[0]).await);

    // Reload u2's todo
    let u2_todo = Todo::get_by_id(&mut db, u2_todos[0].id).await?;
    assert_eq!(*u2_todo.user_id.as_ref().unwrap(), u2.id);

    // Link the TODO back up
    user.todos().insert(&mut db, &todos[0]).await?;

    // The TODO is in the association again
    let todos_reloaded: Vec<_> = user.todos().exec(&mut db).await?;
    assert!(todos_reloaded.iter().any(|t| t.id == todos[0].id));
    assert_ok!(user.todos().get_by_id(&mut db, todos[0].id).await);
    Ok(())
}

#[driver_test(id(ID), scenario(crate::scenarios::has_many_belongs_to))]
pub async fn add_remove_single_relation_required_belongs_to(test: &mut Test) -> Result<()> {
    let mut db = setup(test).await;

    // Create a user with no todos
    let user = User::create().name("User 1").exec(&mut db).await?;

    // Create some TODOs
    let t1 = user.todos().create().title("todo 1").exec(&mut db).await?;
    let t2 = user.todos().create().title("todo 2").exec(&mut db).await?;
    let t3 = user.todos().create().title("todo 3").exec(&mut db).await?;

    let ids = vec![t1.id, t2.id, t3.id];

    let todos_reloaded: Vec<_> = user.todos().exec(&mut db).await?;
    assert_eq!(todos_reloaded.len(), 3);

    for id in ids {
        assert!(todos_reloaded.iter().any(|t| t.id == id));
    }

    // Unlinking a todo deletes it
    user.todos().remove(&mut db, &todos_reloaded[0]).await?;

    // The TODO no longer exists
    assert_err!(Todo::get_by_id(&mut db, todos_reloaded[0].id).await);

    // Rest of the todos exist
    let todos_reloaded: Vec<_> = user.todos().exec(&mut db).await?;
    assert_eq!(todos_reloaded.len(), 2);
    Ok(())
}

#[driver_test(id(ID), scenario(crate::scenarios::has_many_belongs_to))]
pub async fn reassign_relation_required_belongs_to(test: &mut Test) -> Result<()> {
    let mut db = setup(test).await;

    // Create users with no todos
    let u1 = User::create().name("User 1").exec(&mut db).await?;
    let u2 = User::create().name("User 2").exec(&mut db).await?;

    // Create a TODO associated with user 1
    let t1 = u1.todos().create().title("a todo").exec(&mut db).await?;

    // Associate the todo with user 2
    u2.todos().insert(&mut db, &t1).await?;

    // The TODO is no longer associated with user 1
    assert!(u1.todos().exec(&mut db).await?.is_empty());

    // The TODO is assiated with user 2
    let todos = u2.todos().exec(&mut db).await?;
    assert_eq!(1, todos.len());
    assert_eq!(t1.id, todos[0].id);
    Ok(())
}

#[driver_test(id(ID))]
pub async fn add_remove_multiple_relation_option_belongs_to(test: &mut Test) -> Result<()> {
    #[derive(Debug, toasty::Model)]
    struct User {
        #[key]
        #[auto]
        id: ID,

        #[has_many]
        todos: toasty::HasMany<Todo>,
    }

    #[derive(Debug, toasty::Model)]
    struct Todo {
        #[key]
        #[auto]
        id: ID,

        #[index]
        user_id: Option<ID>,

        #[belongs_to(key = user_id, references = id)]
        user: toasty::BelongsTo<Option<User>>,
    }

    let mut db = test.setup_db(models!(User, Todo)).await;

    // Create a user with no todos
    let user = User::create().exec(&mut db).await?;

    // Create some TODOs
    let t1 = Todo::create().exec(&mut db).await?;
    let t2 = Todo::create().exec(&mut db).await?;
    let t3 = Todo::create().exec(&mut db).await?;

    let ids = vec![t1.id, t2.id, t3.id];

    // Associate the todos with the user
    user.todos().insert(&mut db, &t1).await?;
    user.todos().insert(&mut db, &t2).await?;
    user.todos().insert(&mut db, &t3).await?;

    let todos_reloaded: Vec<_> = user.todos().exec(&mut db).await?;
    assert_eq!(todos_reloaded.len(), 3);

    for id in ids {
        assert!(todos_reloaded.iter().any(|t| t.id == id));
    }
    Ok(())
}