variant_counter 0.4.0

Rust's Enum variant counter
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
68

#![cfg(feature = "full")]
use variant_counter::*;

#[derive(VariantCount)]
enum Opt<T> {
    Some(T),
    None,
}

#[derive(VariantCount)]
enum Either<A, B> {
    A(A),
    B(B),
}

#[derive(VariantCount)]
enum Text<'a, T: Sized> {
    Str(&'a str),
    String(String),
    RichText(T),
}

#[test]
fn test_opt() {
    assert_eq!(Opt::<usize>::variant_count(), 2);

    let mut counter = Opt::<usize>::counter();
    let opt_some = Opt::<usize>::Some(1);
    let opt_none = Opt::<usize>::None;

    assert_eq!(counter.check_some(), 0);
    assert_eq!(counter.check_none(), 0);

    counter.record(&opt_some);
    counter.record(&opt_none);

    assert_eq!(counter.check_some(), 1);
    assert_eq!(counter.check_none(), 1);
}

#[test]
fn test_either() {
    type E = Either<bool, usize>;
    assert_eq!(E::variant_count(), 2);

    let mut counter = E::counter();
    assert_eq!(counter.check_a(), 0);
    assert_eq!(counter.check_b(), 0);

    counter.record(&E::A(true));
    counter.record(&E::B(1));

    assert_eq!(counter.check_a(), 1);
    assert_eq!(counter.check_b(), 1);
}

#[test]
fn test_text() {
    type T = Text<'static, Vec<u8>>;
    assert_eq!(T::variant_count(), 3);

    let mut counter = T::counter();
    counter.record(&T::Str("rust"));
    counter.record(&T::String(String::from("hello world")));
    counter.record(&T::RichText(Vec::from("<p>Rust</p>")));

    assert_eq!(counter.check_str(), 1);
}