tests/random.rs

   1 use libtest_mimic::{Arguments, Failed, Trial};
   2
   3 use mt_net::{generate_random::GenerateRandomVariant, rand, ToCltPkt, ToSrvPkt};
   4 use mt_ser::{DefCfg, MtDeserialize, MtSerialize};
   5 use std::{error::Error, fmt::Debug};
   6
   7 fn test_reserialize<T>(type_name: &'static str) -> impl Iterator<Item = Trial>
   8 where
   9     T: MtSerialize + MtDeserialize + GenerateRandomVariant + PartialEq + Debug,
  10 {
  11     (0..T::num_variants()).map(move |i| {
  12         Trial::test(format!("{type_name}::{}", T::variant_name(i)), move || {
  13             let mut rng = rand::thread_rng();
  14
  15             for _ in 0..100 {
  16                 let input = T::generate_random_variant(&mut rng, i);
  17
  18                 let mut writer = Vec::new();
  19                 input
  20                     .mt_serialize::<DefCfg>(&mut writer)
  21                     .map_err(|e| format!("serialize error: {e}\ninput: {input:?}"))?;
  22
  23                 let mut reader = std::io::Cursor::new(writer);
  24                 let output = T::mt_deserialize::<DefCfg>(&mut reader)
  25                     .map_err(|e| format!("deserialize error: {e}\ninput: {input:?}"))?;
  26
  27                 if input != output {
  28                     return Err(format!(
  29                         "output did not match input\n\
  30                                                 input: {input:?}\n\
  31                                                 output: {output:?}",
  32                     )
  33                     .into());
  34                 }
  35             }
  36
  37             Ok(())
  38         })
  39         .with_kind("random")
  40     })
  41 }
  42
  43 fn main() -> Result<(), Box<dyn Error>> {
  44     let args = Arguments::from_args();
  45     let tests = test_reserialize::<ToSrvPkt>("ToSrvPkt")
  46         .chain(test_reserialize::<ToCltPkt>("ToCltPkt"))
  47         .collect();
  48     libtest_mimic::run(&args, tests).exit();
  49 }