trait_enum 0.5.0

Easy enum wrapper that implements all traits that the wrapped objects implement
Documentation 
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

use crate::*;

pub trait CommonTrait {
    fn test(&self) -> u32;
    fn update(&mut self) -> u32;
}

pub struct InnerOne(u32);
impl CommonTrait for InnerOne {
    fn test(&self) -> u32 {
        self.0
    }

    fn update(&mut self) -> u32 {
        self.0 += 1;
        self.0
    }
}

pub struct InnerTwo(u32);
impl CommonTrait for InnerTwo {
    fn test(&self) -> u32 {
        self.0
    }

    fn update(&mut self) -> u32 {
        self.0 += 1;
        self.0
    }
}

trait_enum! {
    pub enum Combined: CommonTrait {
        InnerOne,
        InnerTwo
    }
}

#[test]
fn test_enum() {
    #[cfg(not(feature = "std"))]
    pub use core::ops::{Deref, DerefMut};
    #[cfg(feature = "std")]
    pub use std::ops::{Deref, DerefMut};

    let mut combined = Combined::InnerOne(InnerOne(1));
    {
        let deref: &dyn CommonTrait = combined.deref();
        assert_eq!(deref.test(), 1);
    }
    {
        let deref_mut: &mut dyn CommonTrait = combined.deref_mut();
        assert_eq!(deref_mut.update(), 2);
        assert_eq!(deref_mut.test(), 2);
    }

    let mut combined = Combined::InnerTwo(InnerTwo(2));
    {
        let deref: &dyn CommonTrait = combined.deref();
        assert_eq!(deref.test(), 2);
    }
    {
        let deref_mut: &mut dyn CommonTrait = combined.deref_mut();
        assert_eq!(deref_mut.update(), 3);
        assert_eq!(deref_mut.test(), 3);
    }
}