third_party/rust/arbitrary/examples/derive_enum.rs


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

//! A simple example of deriving the `Arbitrary` trait for an `enum`.
//!
//! Note that this requires enabling the "derive" cargo feature.

// Various enums/fields that we are deriving `Arbitrary` for aren't actually
// used except to show off the derive.
#![allow(dead_code)]

use arbitrary::{Arbitrary, Unstructured};

#[derive(Arbitrary, Debug)]
enum MyEnum {
    UnitVariant,
    TupleVariant(bool, u32),
    StructVariant { x: i8, y: (u8, i32) },
}

fn main() {
    let raw = b"This is some raw, unstructured data!";

    let mut unstructured = Unstructured::new(raw);

    let instance = MyEnum::arbitrary(&mut unstructured)
        .expect("`unstructured` has enough underlying data to create all variants of `MyEnum`");

    println!("Here is an arbitrary enum: {:?}", instance);
}