1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
|
#[macro_use]
extern crate rental;
pub struct Foo {
i: i32,
}
impl Foo {
fn try_borrow(&self) -> Result<&i32, ()> { Ok(&self.i) }
fn fail_borrow(&self) -> Result<&i32, ()> { Err(()) }
}
pub struct FooRef<'i> {
iref: &'i i32,
misc: i32,
}
impl<'i> ::std::ops::Deref for FooRef<'i> {
type Target = i32;
fn deref(&self) -> &i32 { self.iref }
}
rental! {
mod rentals {
use super::*;
#[rental]
pub struct SimpleRef {
foo: Box<Foo>,
fr: FooRef<'foo>,
}
}
}
#[test]
fn new() {
let foo = Foo { i: 5 };
let _ = rentals::SimpleRef::new(Box::new(foo), |foo| FooRef{ iref: &foo.i, misc: 12 });
let foo = Foo { i: 5 };
let sr: rental::RentalResult<rentals::SimpleRef, (), _> = rentals::SimpleRef::try_new(Box::new(foo), |foo| Ok(FooRef{ iref: foo.try_borrow()?, misc: 12 }));
assert!(sr.is_ok());
let foo = Foo { i: 5 };
let sr: rental::RentalResult<rentals::SimpleRef, (), _> = rentals::SimpleRef::try_new(Box::new(foo), |foo| Ok(FooRef{ iref: foo.fail_borrow()?, misc: 12 }));
assert!(sr.is_err());
}
#[test]
fn read() {
let foo = Foo { i: 5 };
let mut sr = rentals::SimpleRef::new(Box::new(foo), |foo| FooRef{ iref: &foo.i, misc: 12 });
{
let i: i32 = sr.rent(|iref| **iref);
assert_eq!(i, 5);
}
{
let iref: &i32 = sr.ref_rent(|fr| fr.iref);
assert_eq!(*iref, 5);
let iref: Option<&i32> = sr.maybe_ref_rent(|fr| Some(fr.iref));
assert_eq!(iref, Some(&5));
let iref: Result<&i32, ()> = sr.try_ref_rent(|fr| Ok(fr.iref));
assert_eq!(iref, Ok(&5));
}
{
assert_eq!(sr.head().i, 5);
assert_eq!(sr.rent_all(|borrows| borrows.foo.i), 5);
assert_eq!(sr.rent_all_mut(|borrows| borrows.foo.i), 5);
}
{
let iref: Option<&i32> = sr.maybe_ref_rent_all(|borrows| Some(borrows.fr.iref));
assert_eq!(iref, Some(&5));
let iref: Result<&i32, ()> = sr.try_ref_rent_all(|borrows| Ok(borrows.fr.iref));
assert_eq!(iref, Ok(&5));
}
{
let iref: &mut i32 = sr.ref_rent_all_mut(|borrows| &mut borrows.fr.misc);
*iref = 57;
assert_eq!(*iref, 57);
}
{
let iref: Option<&mut i32> = sr.maybe_ref_rent_all_mut(|borrows| Some(&mut borrows.fr.misc));
assert_eq!(iref, Some(&mut 57));
}
{
let iref: Result<&mut i32, ()> = sr.try_ref_rent_all_mut(|borrows| Ok(&mut borrows.fr.misc));
assert_eq!(iref, Ok(&mut 57));
}
}
|
