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
69
70
71
72
73
74

use crate::models::user_info::UserInfo;
use crate::parsers::basic_parsers::*;
use oni_comb_parser_rs::prelude::*;

//  userinfo      = *( unreserved / pct-encoded / sub-delims / ":" )
pub fn user_info<'a>() -> Parser<'a, u8, UserInfo> {
  let p = || (unreserved().attempt() | pct_encoded().attempt() | sub_delims());
  (p().of_many0().collect().map(|e| e.to_vec()).map_res(String::from_utf8)
    + (elm(b':') * p().of_many1().collect().map(|e| e.to_vec()).map_res(String::from_utf8)).opt())
  .map(|(user_name, password)| UserInfo::new(user_name, password))
  .name("user_info")
}

#[cfg(test)]
pub mod gens {
  use prop_check_rs::gen::{Gen, Gens};

  use crate::parsers::basic_parsers::gens::{pct_encoded_gen, repeat_gen_of_string, sub_delims_gen, unreserved_gen};

  pub fn user_info_gen() -> Gen<String> {
    let gen = {
      repeat_gen_of_string(1, 5, {
        Gens::frequency([(1, unreserved_gen(1)), (1, pct_encoded_gen()), (1, sub_delims_gen(1))])
      })
    };
    Gens::one_bool().flat_map(move |b| {
      let g = gen.clone();
      if b {
        gen
          .clone()
          .flat_map(move |s1| g.clone().map(move |s2| format!("{}:{}", s1, s2)))
      } else {
        gen.clone().map(|s| format!("{}", s))
      }
    })
  }
}

#[cfg(test)]
mod tests {
  use std::env;

  use anyhow::Result;
  use prop_check_rs::prop;
  use prop_check_rs::prop::TestCases;
  use prop_check_rs::rng::RNG;

  use super::gens::*;
  use super::*;

  const TEST_COUNT: TestCases = 100;

  #[ctor::ctor]
  fn init_logger() {
    env::set_var("RUST_LOG", "debug");
    let _ = env_logger::builder().is_test(true).try_init();
  }

  #[test]
  fn test_user_info() -> Result<()> {
    let mut counter = 0;
    let prop = prop::for_all_gen(user_info_gen(), move |s| {
      counter += 1;
      log::debug!("{:>03}, user_info:string = {}", counter, s);
      let input = s.as_bytes();
      let result = (user_info() - end()).parse(input).to_result();
      let user_info = result.unwrap();
      log::debug!("{:>03}, user_info:object = {:?}", counter, user_info);
      assert_eq!(user_info.to_string(), s);
      true
    });
    prop::test_with_prop(prop, 5, TEST_COUNT, RNG::new())
  }
}