random-constructible-derive 0.13.0

Provides a derive macro for the random-constructible crate which is used for creating random instances of enums with weighted probabilities
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

// ---------------- [ File: random-constructible-derive/src/extract_enum_variants.rs ]
crate::ix!();

pub fn extract_enum_variants(input: &syn::DeriveInput) 
    -> &syn::punctuated::Punctuated<syn::Variant, syn::token::Comma> 
{
    if let syn::Data::Enum(ref data_enum) = input.data {
        &data_enum.variants
    } else {
        panic!("Expected enum data");
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use syn::{parse_quote, DeriveInput};

    #[test]
    fn test_extract_enum_variants_with_valid_enum() {
        let input: DeriveInput = parse_quote! {
            enum TestEnum {
                Variant1,
                Variant2,
            }
        };

        let variants = extract_enum_variants(&input);
        assert_eq!(variants.len(), 2); // Two variants in the enum
        assert_eq!(variants[0].ident, "Variant1");
        assert_eq!(variants[1].ident, "Variant2");
    }

    #[test]
    #[should_panic(expected = "Expected enum data")]
    fn test_extract_enum_variants_with_non_enum() {
        let input: DeriveInput = parse_quote! {
            struct TestStruct {
                field: i32,
            }
        };

        extract_enum_variants(&input); // Should panic
    }
}