dioxus-mirror/packages/generational-box/tests/errors.rs

use std::cell::{Ref, RefMut};

use generational_box::{
    AlreadyBorrowedError, AlreadyBorrowedMutError, AnyStorage, BorrowError, BorrowMutError,
    GenerationalBox, GenerationalRef, GenerationalRefMut, Owner, UnsyncStorage, ValueDroppedError,
};

#[track_caller]
fn read_at_location(
    value: GenerationalBox<i32>,
) -> (
    GenerationalRef<Ref<'static, i32>>,
    &'static std::panic::Location<'static>,
) {
    let location = std::panic::Location::caller();
    let read = value.read();
    (read, location)
}

#[track_caller]
fn write_at_location(
    value: GenerationalBox<i32>,
) -> (
    GenerationalRefMut<RefMut<'static, i32>>,
    &'static std::panic::Location<'static>,
) {
    let location = std::panic::Location::caller();
    let write = value.write();
    (write, location)
}

#[track_caller]
fn create_at_location(
    owner: &Owner,
) -> (GenerationalBox<i32>, &'static std::panic::Location<'static>) {
    let location = std::panic::Location::caller();
    let value = owner.insert(1);
    (value, location)
}

#[test]
fn read_while_writing_error() {
    let owner = UnsyncStorage::owner();
    let value = owner.insert(1);

    let (write, location) = write_at_location(value);

    assert_eq!(
        value.try_read().err(),
        Some(BorrowError::AlreadyBorrowedMut(
            AlreadyBorrowedMutError::new(location)
        ))
    );
    drop(write);
}

#[test]
fn read_after_dropped_error() {
    let owner = UnsyncStorage::owner();
    let (value, location) = create_at_location(&owner);
    drop(owner);
    assert_eq!(
        value.try_read().err(),
        Some(BorrowError::Dropped(ValueDroppedError::new(location)))
    );
}

#[test]
fn write_while_writing_error() {
    let owner = UnsyncStorage::owner();
    let value = owner.insert(1);

    let (write, location) = write_at_location(value);

    assert_eq!(
        value.try_write().err(),
        Some(BorrowMutError::AlreadyBorrowedMut(
            AlreadyBorrowedMutError::new(location)
        ))
    );
    drop(write);
}

#[test]
fn write_while_reading_error() {
    let owner = UnsyncStorage::owner();
    let value = owner.insert(1);

    let (read, location) = read_at_location(value);

    assert_eq!(
        value.try_write().err(),
        Some(BorrowMutError::AlreadyBorrowed(AlreadyBorrowedError::new(
            vec![location]
        )))
    );

    drop(read);
}