summaryrefslogtreecommitdiffstats
path: root/vendor/derive_setters/tests/ref.rs
blob: b8efc01a4444fe63468e8064a1cba433a14ba962 (plain)
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
use derive_setters::*;

#[derive(Default, Setters, Debug, PartialEq, Eq)]
#[setters(borrow_self)]
struct BasicRefStruct {
    #[setters(rename = "test")]
    a: u32,
    b: u32,
    #[setters(borrow_self = "false")]
    c: u32,
    #[setters(borrow_self = false)]
    d: u32,
}

#[test]
fn basic_ref_struct() {
    let mut a = BasicRefStruct::default().c(34).d(4);
    a.test(1);
    a.b(3);

    assert_eq!(a.a, 1);
    assert_eq!(a.b, 3);
    assert_eq!(a.c, 34);
    assert_eq!(a.d, 4);
}

#[derive(Default, Setters, Debug, PartialEq, Eq)]
struct FieldRefStruct {
    a: u32,
    #[setters(borrow_self)]
    b: u32,
}

#[test]
fn field_ref_struct() {
    let mut a = FieldRefStruct::default().a(10);
    a.b(20);

    assert_eq!(a.a, 10);
    assert_eq!(a.b, 20);
}

#[derive(Default, Setters, Debug, PartialEq, Eq)]
#[setters(borrow_self)]
#[setters(generate_delegates(ty = "BasicRefDelegateField", field = "x"))]
#[setters(generate_delegates(ty = "BasicRefDelegateMethod", method = "get_x"))]
struct InnerRefDelegateStruct {
    #[setters(rename = "test")]
    a: u32,
    b: u32,
    c: u32,
}

#[derive(Default, Debug, PartialEq, Eq)]
struct BasicRefDelegateField {
    x: InnerRefDelegateStruct,
}

#[derive(Default, Debug, PartialEq, Eq)]
struct BasicRefDelegateMethod {
    x: Option<InnerRefDelegateStruct>,
}
impl BasicRefDelegateMethod {
    fn get_x(&mut self) -> &mut InnerRefDelegateStruct {
        if self.x.is_none() {
            self.x = Some(InnerRefDelegateStruct::default());
        }
        self.x.as_mut().unwrap()
    }
}

#[test]
fn basic_ref_delegate_field() {
    let mut a = BasicRefDelegateField::default();
    a.test(1);
    a.b(3);
    a.c(34);

    assert_eq!(a.x, InnerRefDelegateStruct{ a: 1, b: 3, c: 34 });
}

#[test]
fn basic_ref_delegate_method() {
    let mut a = BasicRefDelegateMethod::default();
    a.test(1);
    a.b(3);
    a.c(34);

    assert_eq!(a.x, Some(InnerRefDelegateStruct{ a: 1, b: 3, c: 34 }));
}