100 lines
2.3 KiB
Rust
100 lines
2.3 KiB
Rust
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);
|
|
}
|