ref_wrapper 0.1.4

mod common;

pub use common::*;
use ref_wrapper::RefWrapMut;
use std::cell::RefCell;
use std::ops::{Deref, DerefMut};
use std::sync::OnceLock;

#[test]
fn deref() {
    let src = RefCell::new(samples());
    let target = RefWrapMut::new(src.borrow_mut(), |x| Box::new(VecEdit::new(x)));

    let result = target.deref();

    assert_eq!(*result.base(), samples());
    assert!(src.try_borrow().is_err());
    assert!(src.try_borrow_mut().is_err());
}

#[test]
fn deref_mut() {
    let src = RefCell::new(samples());
    let mut target = RefWrapMut::new(src.borrow_mut(), |x| Box::new(VecEdit::new(x)));

    let result = target.deref_mut();

    result.add(1);
    let expected_base = Vec::from_iter(samples().iter().map(|x| x + 1));
    assert_eq!(*target.base(), expected_base);
    assert!(src.try_borrow().is_err());
    assert!(src.try_borrow_mut().is_err());
}

#[test]
fn test_drop_count() {
    static DROP_TRACER: OnceLock<DropTracer> = OnceLock::new();
    let drop_tracer = DROP_TRACER.get_or_init(DropTracer::new);
    let item = drop_tracer.new_item();

    {
        assert_eq!(drop_tracer.count(), 1);
        let src = RefCell::new(item);
        RefWrapMut::new(src.borrow_mut(), Box::new);
        assert_eq!(drop_tracer.count(), 1);
    }

    assert_eq!(drop_tracer.count(), 0);
}

pub struct VecEdit<'a> {
    base: &'a mut Vec<i32>,
}

impl<'a> VecEdit<'a> {
    pub fn new(base: &'a mut Vec<i32>) -> Self {
        Self { base }
    }

    pub fn base(&'a self) -> &'a Vec<i32> {
        self.base
    }

    pub fn add(&mut self, value: i32) {
        self.base.iter_mut().for_each(|x| *x += value);
    }
}