summaryrefslogtreecommitdiff
path: root/tests/random.rs
blob: 4d2ad42cd9b2f32c689f38530586725af18bafe1 (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
use libtest_mimic::{Arguments, Failed, Trial};

use mt_net::{generate_random::GenerateRandomVariant, rand, ToCltPkt, ToSrvPkt};
use mt_ser::{DefCfg, MtDeserialize, MtSerialize};
use std::{error::Error, fmt::Debug};

fn test_reserialize<T>(type_name: &'static str) -> impl Iterator<Item = Trial>
where
    T: MtSerialize + MtDeserialize + GenerateRandomVariant + PartialEq + Debug,
{
    (0..T::num_variants()).map(move |i| {
        Trial::test(format!("{type_name}::{}", T::variant_name(i)), move || {
            let mut rng = rand::thread_rng();

            for _ in 0..100 {
                let input = T::generate_random_variant(&mut rng, i);

                let mut writer = Vec::new();
                input
                    .mt_serialize::<DefCfg>(&mut writer)
                    .map_err(|e| format!("serialize error: {e}\ninput: {input:?}"))?;

                let mut reader = std::io::Cursor::new(writer);
                let output = T::mt_deserialize::<DefCfg>(&mut reader)
                    .map_err(|e| format!("deserialize error: {e}\ninput: {input:?}"))?;

                if input != output {
                    return Err(format!(
                        "output did not match input\n\
						input: {input:?}\n\
						output: {output:?}",
                    )
                    .into());
                }
            }

            Ok(())
        })
        .with_kind("random")
    })
}

fn main() -> Result<(), Box<dyn Error>> {
    let args = Arguments::from_args();
    let tests = test_reserialize::<ToSrvPkt>("ToSrvPkt")
        .chain(test_reserialize::<ToCltPkt>("ToCltPkt"))
        .collect();
    libtest_mimic::run(&args, tests).exit();
}