use super::*; use crate::ascii::digit1 as digit; use crate::binary::u16; use crate::binary::u8; use crate::binary::Endianness; use crate::error::ErrMode; use crate::error::ErrorKind; use crate::error::InputError; use crate::error::Needed; use crate::error::ParserError; use crate::stream::Stream; use crate::token::take; use crate::unpeek; use crate::IResult; use crate::PResult; use crate::Parser; use crate::Partial; #[cfg(feature = "alloc")] use crate::lib::std::vec::Vec; macro_rules! assert_parse( ($left: expr, $right: expr) => { let res: $crate::IResult<_, _, InputError<_>> = $left; assert_eq!(res, $right); }; ); #[test] fn eof_on_slices() { let not_over: &[u8] = &b"Hello, world!"[..]; let is_over: &[u8] = &b""[..]; let res_not_over = eof.parse_peek(not_over); assert_parse!( res_not_over, Err(ErrMode::Backtrack(error_position!( ¬_over, ErrorKind::Eof ))) ); let res_over = eof.parse_peek(is_over); assert_parse!(res_over, Ok((is_over, is_over))); } #[test] fn eof_on_strs() { let not_over: &str = "Hello, world!"; let is_over: &str = ""; let res_not_over = eof.parse_peek(not_over); assert_parse!( res_not_over, Err(ErrMode::Backtrack(error_position!( ¬_over, ErrorKind::Eof ))) ); let res_over = eof.parse_peek(is_over); assert_parse!(res_over, Ok((is_over, is_over))); } #[test] fn rest_on_slices() { let input: &[u8] = &b"Hello, world!"[..]; let empty: &[u8] = &b""[..]; assert_parse!(rest.parse_peek(input), Ok((empty, input))); } #[test] fn rest_on_strs() { let input: &str = "Hello, world!"; let empty: &str = ""; assert_parse!(rest.parse_peek(input), Ok((empty, input))); } #[test] fn rest_len_on_slices() { let input: &[u8] = &b"Hello, world!"[..]; assert_parse!(rest_len.parse_peek(input), Ok((input, input.len()))); } use crate::lib::std::convert::From; impl From for CustomError { fn from(_: u32) -> Self { CustomError } } impl ParserError for CustomError { fn from_error_kind(_: &I, _: ErrorKind) -> Self { CustomError } fn append(self, _: &I, _: ErrorKind) -> Self { CustomError } } struct CustomError; #[allow(dead_code)] fn custom_error(input: &[u8]) -> IResult<&[u8], &[u8], CustomError> { //fix_error!(input, CustomError<_>, alphanumeric) crate::ascii::alphanumeric1.parse_peek(input) } #[test] fn test_parser_flat_map() { let input: &[u8] = &[3, 100, 101, 102, 103, 104][..]; assert_parse!( u8.flat_map(take).parse_peek(input), Ok((&[103, 104][..], &[100, 101, 102][..])) ); } #[allow(dead_code)] fn test_closure_compiles_195(input: &[u8]) -> IResult<&[u8], ()> { u8.flat_map(|num| repeat(num as usize, u16(Endianness::Big))) .parse_peek(input) } #[test] fn test_parser_verify_map() { let input: &[u8] = &[50][..]; assert_parse!( u8.verify_map(|u| if u < 20 { Some(u) } else { None }) .parse_peek(input), Err(ErrMode::Backtrack(InputError::new( &[50][..], ErrorKind::Verify ))) ); assert_parse!( u8.verify_map(|u| if u > 20 { Some(u) } else { None }) .parse_peek(input), Ok((&[][..], 50)) ); } #[test] fn test_parser_map_parser() { let input: &[u8] = &[100, 101, 102, 103, 104][..]; assert_parse!( take(4usize).and_then(take(2usize)).parse_peek(input), Ok((&[104][..], &[100, 101][..])) ); } #[test] #[cfg(feature = "std")] fn test_parser_into() { use crate::error::InputError; use crate::token::take; let mut parser = take::<_, _, InputError<_>>(3u8).output_into(); let result: IResult<&[u8], Vec> = parser.parse_peek(&b"abcdefg"[..]); assert_eq!(result, Ok((&b"defg"[..], vec![97, 98, 99]))); } #[test] fn opt_test() { fn opt_abcd(i: Partial<&[u8]>) -> IResult, Option<&[u8]>> { opt("abcd").parse_peek(i) } let a = &b"abcdef"[..]; let b = &b"bcdefg"[..]; let c = &b"ab"[..]; assert_eq!( opt_abcd(Partial::new(a)), Ok((Partial::new(&b"ef"[..]), Some(&b"abcd"[..]))) ); assert_eq!( opt_abcd(Partial::new(b)), Ok((Partial::new(&b"bcdefg"[..]), None)) ); assert_eq!( opt_abcd(Partial::new(c)), Err(ErrMode::Incomplete(Needed::new(2))) ); } #[test] fn peek_test() { fn peek_tag(i: Partial<&[u8]>) -> IResult, &[u8]> { peek("abcd").parse_peek(i) } assert_eq!( peek_tag(Partial::new(&b"abcdef"[..])), Ok((Partial::new(&b"abcdef"[..]), &b"abcd"[..])) ); assert_eq!( peek_tag(Partial::new(&b"ab"[..])), Err(ErrMode::Incomplete(Needed::new(2))) ); assert_eq!( peek_tag(Partial::new(&b"xxx"[..])), Err(ErrMode::Backtrack(error_position!( &Partial::new(&b"xxx"[..]), ErrorKind::Tag ))) ); } #[test] fn not_test() { fn not_aaa(i: Partial<&[u8]>) -> IResult, ()> { not("aaa").parse_peek(i) } assert_eq!( not_aaa(Partial::new(&b"aaa"[..])), Err(ErrMode::Backtrack(error_position!( &Partial::new(&b"aaa"[..]), ErrorKind::Not ))) ); assert_eq!( not_aaa(Partial::new(&b"aa"[..])), Err(ErrMode::Incomplete(Needed::new(1))) ); assert_eq!( not_aaa(Partial::new(&b"abcd"[..])), Ok((Partial::new(&b"abcd"[..]), ())) ); } #[test] fn test_parser_verify() { use crate::token::take; fn test(i: Partial<&[u8]>) -> IResult, &[u8]> { take(5u8) .verify(|slice: &[u8]| slice[0] == b'a') .parse_peek(i) } assert_eq!( test(Partial::new(&b"bcd"[..])), Err(ErrMode::Incomplete(Needed::new(2))) ); assert_eq!( test(Partial::new(&b"bcdefg"[..])), Err(ErrMode::Backtrack(error_position!( &Partial::new(&b"bcdefg"[..]), ErrorKind::Verify ))) ); assert_eq!( test(Partial::new(&b"abcdefg"[..])), Ok((Partial::new(&b"fg"[..]), &b"abcde"[..])) ); } #[test] #[allow(unused)] fn test_parser_verify_ref() { use crate::token::take; let mut parser1 = take(3u8).verify(|s: &[u8]| s == &b"abc"[..]); assert_eq!( parser1.parse_peek(&b"abcd"[..]), Ok((&b"d"[..], &b"abc"[..])) ); assert_eq!( parser1.parse_peek(&b"defg"[..]), Err(ErrMode::Backtrack(InputError::new( &b"defg"[..], ErrorKind::Verify ))) ); fn parser2(i: &[u8]) -> IResult<&[u8], u32> { crate::binary::be_u32 .verify(|val: &u32| *val < 3) .parse_peek(i) } } #[test] #[cfg(feature = "alloc")] fn test_parser_verify_alloc() { use crate::token::take; let mut parser1 = take(3u8) .map(|s: &[u8]| s.to_vec()) .verify(|s: &[u8]| s == &b"abc"[..]); assert_eq!( parser1.parse_peek(&b"abcd"[..]), Ok((&b"d"[..], b"abc".to_vec())) ); assert_eq!( parser1.parse_peek(&b"defg"[..]), Err(ErrMode::Backtrack(InputError::new( &b"defg"[..], ErrorKind::Verify ))) ); } #[test] fn fail_test() { let a = "string"; let b = "another string"; assert_eq!( fail::<_, &str, _>.parse_peek(a), Err(ErrMode::Backtrack(InputError::new(a, ErrorKind::Fail))) ); assert_eq!( fail::<_, &str, _>.parse_peek(b), Err(ErrMode::Backtrack(InputError::new(b, ErrorKind::Fail))) ); } #[test] fn complete() { fn err_test(i: &[u8]) -> IResult<&[u8], &[u8]> { let (i, _) = "ijkl".parse_peek(i)?; "mnop".parse_peek(i) } let a = &b"ijklmn"[..]; let res_a = err_test(a); assert_eq!( res_a, Err(ErrMode::Backtrack(error_position!( &&b"mn"[..], ErrorKind::Tag ))) ); } #[test] fn separated_pair_test() { #[allow(clippy::type_complexity)] fn sep_pair_abc_def(i: Partial<&[u8]>) -> IResult, (&[u8], &[u8])> { separated_pair("abc", ",", "def").parse_peek(i) } assert_eq!( sep_pair_abc_def(Partial::new(&b"abc,defghijkl"[..])), Ok((Partial::new(&b"ghijkl"[..]), (&b"abc"[..], &b"def"[..]))) ); assert_eq!( sep_pair_abc_def(Partial::new(&b"ab"[..])), Err(ErrMode::Incomplete(Needed::new(1))) ); assert_eq!( sep_pair_abc_def(Partial::new(&b"abc,d"[..])), Err(ErrMode::Incomplete(Needed::new(2))) ); assert_eq!( sep_pair_abc_def(Partial::new(&b"xxx"[..])), Err(ErrMode::Backtrack(error_position!( &Partial::new(&b"xxx"[..]), ErrorKind::Tag ))) ); assert_eq!( sep_pair_abc_def(Partial::new(&b"xxx,def"[..])), Err(ErrMode::Backtrack(error_position!( &Partial::new(&b"xxx,def"[..]), ErrorKind::Tag ))) ); assert_eq!( sep_pair_abc_def(Partial::new(&b"abc,xxx"[..])), Err(ErrMode::Backtrack(error_position!( &Partial::new(&b"xxx"[..]), ErrorKind::Tag ))) ); } #[test] fn preceded_test() { fn preceded_abcd_efgh(i: Partial<&[u8]>) -> IResult, &[u8]> { preceded("abcd", "efgh").parse_peek(i) } assert_eq!( preceded_abcd_efgh(Partial::new(&b"abcdefghijkl"[..])), Ok((Partial::new(&b"ijkl"[..]), &b"efgh"[..])) ); assert_eq!( preceded_abcd_efgh(Partial::new(&b"ab"[..])), Err(ErrMode::Incomplete(Needed::new(2))) ); assert_eq!( preceded_abcd_efgh(Partial::new(&b"abcde"[..])), Err(ErrMode::Incomplete(Needed::new(3))) ); assert_eq!( preceded_abcd_efgh(Partial::new(&b"xxx"[..])), Err(ErrMode::Backtrack(error_position!( &Partial::new(&b"xxx"[..]), ErrorKind::Tag ))) ); assert_eq!( preceded_abcd_efgh(Partial::new(&b"xxxxdef"[..])), Err(ErrMode::Backtrack(error_position!( &Partial::new(&b"xxxxdef"[..]), ErrorKind::Tag ))) ); assert_eq!( preceded_abcd_efgh(Partial::new(&b"abcdxxx"[..])), Err(ErrMode::Backtrack(error_position!( &Partial::new(&b"xxx"[..]), ErrorKind::Tag ))) ); } #[test] fn terminated_test() { fn terminated_abcd_efgh(i: Partial<&[u8]>) -> IResult, &[u8]> { terminated("abcd", "efgh").parse_peek(i) } assert_eq!( terminated_abcd_efgh(Partial::new(&b"abcdefghijkl"[..])), Ok((Partial::new(&b"ijkl"[..]), &b"abcd"[..])) ); assert_eq!( terminated_abcd_efgh(Partial::new(&b"ab"[..])), Err(ErrMode::Incomplete(Needed::new(2))) ); assert_eq!( terminated_abcd_efgh(Partial::new(&b"abcde"[..])), Err(ErrMode::Incomplete(Needed::new(3))) ); assert_eq!( terminated_abcd_efgh(Partial::new(&b"xxx"[..])), Err(ErrMode::Backtrack(error_position!( &Partial::new(&b"xxx"[..]), ErrorKind::Tag ))) ); assert_eq!( terminated_abcd_efgh(Partial::new(&b"xxxxdef"[..])), Err(ErrMode::Backtrack(error_position!( &Partial::new(&b"xxxxdef"[..]), ErrorKind::Tag ))) ); assert_eq!( terminated_abcd_efgh(Partial::new(&b"abcdxxxx"[..])), Err(ErrMode::Backtrack(error_position!( &Partial::new(&b"xxxx"[..]), ErrorKind::Tag ))) ); } #[test] fn delimited_test() { fn delimited_abc_def_ghi(i: Partial<&[u8]>) -> IResult, &[u8]> { delimited("abc", "def", "ghi").parse_peek(i) } assert_eq!( delimited_abc_def_ghi(Partial::new(&b"abcdefghijkl"[..])), Ok((Partial::new(&b"jkl"[..]), &b"def"[..])) ); assert_eq!( delimited_abc_def_ghi(Partial::new(&b"ab"[..])), Err(ErrMode::Incomplete(Needed::new(1))) ); assert_eq!( delimited_abc_def_ghi(Partial::new(&b"abcde"[..])), Err(ErrMode::Incomplete(Needed::new(1))) ); assert_eq!( delimited_abc_def_ghi(Partial::new(&b"abcdefgh"[..])), Err(ErrMode::Incomplete(Needed::new(1))) ); assert_eq!( delimited_abc_def_ghi(Partial::new(&b"xxx"[..])), Err(ErrMode::Backtrack(error_position!( &Partial::new(&b"xxx"[..]), ErrorKind::Tag ))) ); assert_eq!( delimited_abc_def_ghi(Partial::new(&b"xxxdefghi"[..])), Err(ErrMode::Backtrack(error_position!( &Partial::new(&b"xxxdefghi"[..]), ErrorKind::Tag ),)) ); assert_eq!( delimited_abc_def_ghi(Partial::new(&b"abcxxxghi"[..])), Err(ErrMode::Backtrack(error_position!( &Partial::new(&b"xxxghi"[..]), ErrorKind::Tag ))) ); assert_eq!( delimited_abc_def_ghi(Partial::new(&b"abcdefxxx"[..])), Err(ErrMode::Backtrack(error_position!( &Partial::new(&b"xxx"[..]), ErrorKind::Tag ))) ); } #[cfg(feature = "alloc")] #[test] fn alt_test() { #[cfg(feature = "alloc")] use crate::{ error::ParserError, lib::std::{ fmt::Debug, string::{String, ToString}, }, }; #[cfg(feature = "alloc")] #[derive(Debug, Clone, Eq, PartialEq)] pub struct ErrorStr(String); #[cfg(feature = "alloc")] impl From for ErrorStr { fn from(i: u32) -> Self { ErrorStr(format!("custom error code: {}", i)) } } #[cfg(feature = "alloc")] impl<'a> From<&'a str> for ErrorStr { fn from(i: &'a str) -> Self { ErrorStr(format!("custom error message: {}", i)) } } #[cfg(feature = "alloc")] impl ParserError for ErrorStr { fn from_error_kind(input: &I, kind: ErrorKind) -> Self { ErrorStr(format!("custom error message: ({:?}, {:?})", input, kind)) } fn append(self, input: &I, kind: ErrorKind) -> Self { ErrorStr(format!( "custom error message: ({:?}, {:?}) - {:?}", input, kind, self )) } } fn work(input: &[u8]) -> IResult<&[u8], &[u8], ErrorStr> { Ok(input.peek_finish()) } #[allow(unused_variables)] fn dont_work(input: &[u8]) -> IResult<&[u8], &[u8], ErrorStr> { Err(ErrMode::Backtrack(ErrorStr("abcd".to_string()))) } fn work2(input: &[u8]) -> IResult<&[u8], &[u8], ErrorStr> { Ok((input, &b""[..])) } fn alt1(i: &[u8]) -> IResult<&[u8], &[u8], ErrorStr> { alt((unpeek(dont_work), unpeek(dont_work))).parse_peek(i) } fn alt2(i: &[u8]) -> IResult<&[u8], &[u8], ErrorStr> { alt((unpeek(dont_work), unpeek(work))).parse_peek(i) } fn alt3(i: &[u8]) -> IResult<&[u8], &[u8], ErrorStr> { alt(( unpeek(dont_work), unpeek(dont_work), unpeek(work2), unpeek(dont_work), )) .parse_peek(i) } //named!(alt1, alt!(dont_work | dont_work)); //named!(alt2, alt!(dont_work | work)); //named!(alt3, alt!(dont_work | dont_work | work2 | dont_work)); let a = &b"abcd"[..]; assert_eq!( alt1(a), Err(ErrMode::Backtrack(error_node_position!( &a, ErrorKind::Alt, ErrorStr("abcd".to_string()) ))) ); assert_eq!(alt2(a), Ok((&b""[..], a))); assert_eq!(alt3(a), Ok((a, &b""[..]))); fn alt4(i: &[u8]) -> IResult<&[u8], &[u8]> { alt(("abcd", "efgh")).parse_peek(i) } let b = &b"efgh"[..]; assert_eq!(alt4(a), Ok((&b""[..], a))); assert_eq!(alt4(b), Ok((&b""[..], b))); } #[test] fn alt_incomplete() { fn alt1(i: Partial<&[u8]>) -> IResult, &[u8]> { alt(("a", "bc", "def")).parse_peek(i) } let a = &b""[..]; assert_eq!( alt1(Partial::new(a)), Err(ErrMode::Incomplete(Needed::new(1))) ); let a = &b"b"[..]; assert_eq!( alt1(Partial::new(a)), Err(ErrMode::Incomplete(Needed::new(1))) ); let a = &b"bcd"[..]; assert_eq!( alt1(Partial::new(a)), Ok((Partial::new(&b"d"[..]), &b"bc"[..])) ); let a = &b"cde"[..]; assert_eq!( alt1(Partial::new(a)), Err(ErrMode::Backtrack(error_position!( &Partial::new(a), ErrorKind::Tag ))) ); let a = &b"de"[..]; assert_eq!( alt1(Partial::new(a)), Err(ErrMode::Incomplete(Needed::new(1))) ); let a = &b"defg"[..]; assert_eq!( alt1(Partial::new(a)), Ok((Partial::new(&b"g"[..]), &b"def"[..])) ); } #[test] fn alt_array() { fn alt1<'i>(i: &mut &'i [u8]) -> PResult<&'i [u8]> { alt(["a", "bc", "def"]).parse_next(i) } let i = &b"a"[..]; assert_eq!(alt1.parse_peek(i), Ok((&b""[..], (&b"a"[..])))); let i = &b"bc"[..]; assert_eq!(alt1.parse_peek(i), Ok((&b""[..], (&b"bc"[..])))); let i = &b"defg"[..]; assert_eq!(alt1.parse_peek(i), Ok((&b"g"[..], (&b"def"[..])))); let i = &b"z"[..]; assert_eq!( alt1.parse_peek(i), Err(ErrMode::Backtrack(error_position!(&i, ErrorKind::Tag))) ); } #[test] fn permutation_test() { #[allow(clippy::type_complexity)] fn perm(i: Partial<&[u8]>) -> IResult, (&[u8], &[u8], &[u8])> { permutation(("abcd", "efg", "hi")).parse_peek(i) } let expected = (&b"abcd"[..], &b"efg"[..], &b"hi"[..]); let a = &b"abcdefghijk"[..]; assert_eq!( perm(Partial::new(a)), Ok((Partial::new(&b"jk"[..]), expected)) ); let b = &b"efgabcdhijk"[..]; assert_eq!( perm(Partial::new(b)), Ok((Partial::new(&b"jk"[..]), expected)) ); let c = &b"hiefgabcdjk"[..]; assert_eq!( perm(Partial::new(c)), Ok((Partial::new(&b"jk"[..]), expected)) ); let d = &b"efgxyzabcdefghi"[..]; assert_eq!( perm(Partial::new(d)), Err(ErrMode::Backtrack(error_node_position!( &Partial::new(&b"efgxyzabcdefghi"[..]), ErrorKind::Alt, error_position!(&Partial::new(&b"xyzabcdefghi"[..]), ErrorKind::Tag) ))) ); let e = &b"efgabc"[..]; assert_eq!( perm(Partial::new(e)), Err(ErrMode::Incomplete(Needed::new(1))) ); } #[test] #[cfg(feature = "alloc")] fn separated0_test() { fn multi(i: Partial<&[u8]>) -> IResult, Vec<&[u8]>> { separated0("abcd", ",").parse_peek(i) } fn multi_empty(i: Partial<&[u8]>) -> IResult, Vec<&[u8]>> { separated0("", ",").parse_peek(i) } fn multi_longsep(i: Partial<&[u8]>) -> IResult, Vec<&[u8]>> { separated0("abcd", "..").parse_peek(i) } let a = &b"abcdef"[..]; let b = &b"abcd,abcdef"[..]; let c = &b"azerty"[..]; let d = &b",,abc"[..]; let e = &b"abcd,abcd,ef"[..]; let f = &b"abc"[..]; let g = &b"abcd."[..]; let h = &b"abcd,abc"[..]; let res1 = vec![&b"abcd"[..]]; assert_eq!(multi(Partial::new(a)), Ok((Partial::new(&b"ef"[..]), res1))); let res2 = vec![&b"abcd"[..], &b"abcd"[..]]; assert_eq!(multi(Partial::new(b)), Ok((Partial::new(&b"ef"[..]), res2))); assert_eq!( multi(Partial::new(c)), Ok((Partial::new(&b"azerty"[..]), Vec::new())) ); let res3 = vec![&b""[..], &b""[..], &b""[..]]; assert_eq!( multi_empty(Partial::new(d)), Ok((Partial::new(&b"abc"[..]), res3)) ); let res4 = vec![&b"abcd"[..], &b"abcd"[..]]; assert_eq!( multi(Partial::new(e)), Ok((Partial::new(&b",ef"[..]), res4)) ); assert_eq!( multi(Partial::new(f)), Err(ErrMode::Incomplete(Needed::new(1))) ); assert_eq!( multi_longsep(Partial::new(g)), Err(ErrMode::Incomplete(Needed::new(1))) ); assert_eq!( multi(Partial::new(h)), Err(ErrMode::Incomplete(Needed::new(1))) ); } #[test] #[cfg(feature = "alloc")] #[cfg_attr(debug_assertions, should_panic)] fn separated0_empty_sep_test() { fn empty_sep(i: Partial<&[u8]>) -> IResult, Vec<&[u8]>> { separated0("abc", "").parse_peek(i) } let i = &b"abcabc"[..]; let i_err_pos = &i[3..]; assert_eq!( empty_sep(Partial::new(i)), Err(ErrMode::Backtrack(error_position!( &Partial::new(i_err_pos), ErrorKind::Assert ))) ); } #[test] #[cfg(feature = "alloc")] fn separated1_test() { fn multi(i: Partial<&[u8]>) -> IResult, Vec<&[u8]>> { separated1("abcd", ",").parse_peek(i) } fn multi_longsep(i: Partial<&[u8]>) -> IResult, Vec<&[u8]>> { separated1("abcd", "..").parse_peek(i) } let a = &b"abcdef"[..]; let b = &b"abcd,abcdef"[..]; let c = &b"azerty"[..]; let d = &b"abcd,abcd,ef"[..]; let f = &b"abc"[..]; let g = &b"abcd."[..]; let h = &b"abcd,abc"[..]; let res1 = vec![&b"abcd"[..]]; assert_eq!(multi(Partial::new(a)), Ok((Partial::new(&b"ef"[..]), res1))); let res2 = vec![&b"abcd"[..], &b"abcd"[..]]; assert_eq!(multi(Partial::new(b)), Ok((Partial::new(&b"ef"[..]), res2))); assert_eq!( multi(Partial::new(c)), Err(ErrMode::Backtrack(error_position!( &Partial::new(c), ErrorKind::Tag ))) ); let res3 = vec![&b"abcd"[..], &b"abcd"[..]]; assert_eq!( multi(Partial::new(d)), Ok((Partial::new(&b",ef"[..]), res3)) ); assert_eq!( multi(Partial::new(f)), Err(ErrMode::Incomplete(Needed::new(1))) ); assert_eq!( multi_longsep(Partial::new(g)), Err(ErrMode::Incomplete(Needed::new(1))) ); assert_eq!( multi(Partial::new(h)), Err(ErrMode::Incomplete(Needed::new(1))) ); } #[test] #[cfg(feature = "alloc")] fn repeat0_test() { fn multi(i: Partial<&[u8]>) -> IResult, Vec<&[u8]>> { repeat(0.., "abcd").parse_peek(i) } assert_eq!( multi(Partial::new(&b"abcdef"[..])), Ok((Partial::new(&b"ef"[..]), vec![&b"abcd"[..]])) ); assert_eq!( multi(Partial::new(&b"abcdabcdefgh"[..])), Ok((Partial::new(&b"efgh"[..]), vec![&b"abcd"[..], &b"abcd"[..]])) ); assert_eq!( multi(Partial::new(&b"azerty"[..])), Ok((Partial::new(&b"azerty"[..]), Vec::new())) ); assert_eq!( multi(Partial::new(&b"abcdab"[..])), Err(ErrMode::Incomplete(Needed::new(2))) ); assert_eq!( multi(Partial::new(&b"abcd"[..])), Err(ErrMode::Incomplete(Needed::new(4))) ); assert_eq!( multi(Partial::new(&b""[..])), Err(ErrMode::Incomplete(Needed::new(4))) ); } #[test] #[cfg(feature = "alloc")] #[cfg_attr(debug_assertions, should_panic)] fn repeat0_empty_test() { fn multi_empty(i: Partial<&[u8]>) -> IResult, Vec<&[u8]>> { repeat(0.., "").parse_peek(i) } assert_eq!( multi_empty(Partial::new(&b"abcdef"[..])), Err(ErrMode::Backtrack(error_position!( &Partial::new(&b"abcdef"[..]), ErrorKind::Assert ))) ); } #[test] #[cfg(feature = "alloc")] fn repeat1_test() { fn multi(i: Partial<&[u8]>) -> IResult, Vec<&[u8]>> { repeat(1.., "abcd").parse_peek(i) } let a = &b"abcdef"[..]; let b = &b"abcdabcdefgh"[..]; let c = &b"azerty"[..]; let d = &b"abcdab"[..]; let res1 = vec![&b"abcd"[..]]; assert_eq!(multi(Partial::new(a)), Ok((Partial::new(&b"ef"[..]), res1))); let res2 = vec![&b"abcd"[..], &b"abcd"[..]]; assert_eq!( multi(Partial::new(b)), Ok((Partial::new(&b"efgh"[..]), res2)) ); assert_eq!( multi(Partial::new(c)), Err(ErrMode::Backtrack(error_position!( &Partial::new(c), ErrorKind::Tag ))) ); assert_eq!( multi(Partial::new(d)), Err(ErrMode::Incomplete(Needed::new(2))) ); } #[test] #[cfg(feature = "alloc")] fn repeat_till_test() { #[allow(clippy::type_complexity)] fn multi(i: &[u8]) -> IResult<&[u8], (Vec<&[u8]>, &[u8])> { repeat_till0("abcd", "efgh").parse_peek(i) } let a = b"abcdabcdefghabcd"; let b = b"efghabcd"; let c = b"azerty"; let res_a = (vec![&b"abcd"[..], &b"abcd"[..]], &b"efgh"[..]); let res_b: (Vec<&[u8]>, &[u8]) = (Vec::new(), &b"efgh"[..]); assert_eq!(multi(&a[..]), Ok((&b"abcd"[..], res_a))); assert_eq!(multi(&b[..]), Ok((&b"abcd"[..], res_b))); assert_eq!( multi(&c[..]), Err(ErrMode::Backtrack(error_node_position!( &&c[..], ErrorKind::Many, error_position!(&&c[..], ErrorKind::Tag) ))) ); } #[test] #[cfg(feature = "std")] fn infinite_many() { fn tst(input: &[u8]) -> IResult<&[u8], &[u8]> { println!("input: {:?}", input); Err(ErrMode::Backtrack(error_position!(&input, ErrorKind::Tag))) } // should not go into an infinite loop fn multi0(i: &[u8]) -> IResult<&[u8], Vec<&[u8]>> { repeat(0.., unpeek(tst)).parse_peek(i) } let a = &b"abcdef"[..]; assert_eq!(multi0(a), Ok((a, Vec::new()))); fn multi1(i: &[u8]) -> IResult<&[u8], Vec<&[u8]>> { repeat(1.., unpeek(tst)).parse_peek(i) } let a = &b"abcdef"[..]; assert_eq!( multi1(a), Err(ErrMode::Backtrack(error_position!(&a, ErrorKind::Tag))) ); } #[test] #[cfg(feature = "alloc")] fn repeat_test() { fn multi(i: Partial<&[u8]>) -> IResult, Vec<&[u8]>> { repeat(2..=4, "Abcd").parse_peek(i) } let a = &b"Abcdef"[..]; let b = &b"AbcdAbcdefgh"[..]; let c = &b"AbcdAbcdAbcdAbcdefgh"[..]; let d = &b"AbcdAbcdAbcdAbcdAbcdefgh"[..]; let e = &b"AbcdAb"[..]; assert_eq!( multi(Partial::new(a)), Err(ErrMode::Backtrack(error_position!( &Partial::new(&b"ef"[..]), ErrorKind::Tag ))) ); let res1 = vec![&b"Abcd"[..], &b"Abcd"[..]]; assert_eq!( multi(Partial::new(b)), Ok((Partial::new(&b"efgh"[..]), res1)) ); let res2 = vec![&b"Abcd"[..], &b"Abcd"[..], &b"Abcd"[..], &b"Abcd"[..]]; assert_eq!( multi(Partial::new(c)), Ok((Partial::new(&b"efgh"[..]), res2)) ); let res3 = vec![&b"Abcd"[..], &b"Abcd"[..], &b"Abcd"[..], &b"Abcd"[..]]; assert_eq!( multi(Partial::new(d)), Ok((Partial::new(&b"Abcdefgh"[..]), res3)) ); assert_eq!( multi(Partial::new(e)), Err(ErrMode::Incomplete(Needed::new(2))) ); } #[test] #[cfg(feature = "alloc")] fn count_test() { const TIMES: usize = 2; fn cnt_2(i: Partial<&[u8]>) -> IResult, Vec<&[u8]>> { repeat(TIMES, "abc").parse_peek(i) } assert_eq!( cnt_2(Partial::new(&b"abcabcabcdef"[..])), Ok((Partial::new(&b"abcdef"[..]), vec![&b"abc"[..], &b"abc"[..]])) ); assert_eq!( cnt_2(Partial::new(&b"ab"[..])), Err(ErrMode::Incomplete(Needed::new(1))) ); assert_eq!( cnt_2(Partial::new(&b"abcab"[..])), Err(ErrMode::Incomplete(Needed::new(1))) ); assert_eq!( cnt_2(Partial::new(&b"xxx"[..])), Err(ErrMode::Backtrack(error_position!( &Partial::new(&b"xxx"[..]), ErrorKind::Tag ))) ); assert_eq!( cnt_2(Partial::new(&b"xxxabcabcdef"[..])), Err(ErrMode::Backtrack(error_position!( &Partial::new(&b"xxxabcabcdef"[..]), ErrorKind::Tag ))) ); assert_eq!( cnt_2(Partial::new(&b"abcxxxabcdef"[..])), Err(ErrMode::Backtrack(error_position!( &Partial::new(&b"xxxabcdef"[..]), ErrorKind::Tag ))) ); } #[test] #[cfg(feature = "alloc")] fn count_zero() { const TIMES: usize = 0; fn counter_2(i: &[u8]) -> IResult<&[u8], Vec<&[u8]>> { repeat(TIMES, "abc").parse_peek(i) } let done = &b"abcabcabcdef"[..]; let parsed_done = Vec::new(); let rest = done; let incomplete_1 = &b"ab"[..]; let parsed_incompl_1 = Vec::new(); let incomplete_2 = &b"abcab"[..]; let parsed_incompl_2 = Vec::new(); let error = &b"xxx"[..]; let error_remain = &b"xxx"[..]; let parsed_err = Vec::new(); let error_1 = &b"xxxabcabcdef"[..]; let parsed_err_1 = Vec::new(); let error_1_remain = &b"xxxabcabcdef"[..]; let error_2 = &b"abcxxxabcdef"[..]; let parsed_err_2 = Vec::new(); let error_2_remain = &b"abcxxxabcdef"[..]; assert_eq!(counter_2(done), Ok((rest, parsed_done))); assert_eq!( counter_2(incomplete_1), Ok((incomplete_1, parsed_incompl_1)) ); assert_eq!( counter_2(incomplete_2), Ok((incomplete_2, parsed_incompl_2)) ); assert_eq!(counter_2(error), Ok((error_remain, parsed_err))); assert_eq!(counter_2(error_1), Ok((error_1_remain, parsed_err_1))); assert_eq!(counter_2(error_2), Ok((error_2_remain, parsed_err_2))); } #[derive(Debug, Clone, Eq, PartialEq)] pub struct NilError; impl From<(I, ErrorKind)> for NilError { fn from(_: (I, ErrorKind)) -> Self { NilError } } impl ParserError for NilError { fn from_error_kind(_: &I, _: ErrorKind) -> NilError { NilError } fn append(self, _: &I, _: ErrorKind) -> NilError { NilError } } #[test] #[cfg(feature = "alloc")] fn fold_repeat0_test() { fn fold_into_vec(mut acc: Vec, item: T) -> Vec { acc.push(item); acc } fn multi(i: Partial<&[u8]>) -> IResult, Vec<&[u8]>> { fold_repeat(0.., "abcd", Vec::new, fold_into_vec).parse_peek(i) } assert_eq!( multi(Partial::new(&b"abcdef"[..])), Ok((Partial::new(&b"ef"[..]), vec![&b"abcd"[..]])) ); assert_eq!( multi(Partial::new(&b"abcdabcdefgh"[..])), Ok((Partial::new(&b"efgh"[..]), vec![&b"abcd"[..], &b"abcd"[..]])) ); assert_eq!( multi(Partial::new(&b"azerty"[..])), Ok((Partial::new(&b"azerty"[..]), Vec::new())) ); assert_eq!( multi(Partial::new(&b"abcdab"[..])), Err(ErrMode::Incomplete(Needed::new(2))) ); assert_eq!( multi(Partial::new(&b"abcd"[..])), Err(ErrMode::Incomplete(Needed::new(4))) ); assert_eq!( multi(Partial::new(&b""[..])), Err(ErrMode::Incomplete(Needed::new(4))) ); } #[test] #[cfg(feature = "alloc")] #[cfg_attr(debug_assertions, should_panic)] fn fold_repeat0_empty_test() { fn fold_into_vec(mut acc: Vec, item: T) -> Vec { acc.push(item); acc } fn multi_empty(i: Partial<&[u8]>) -> IResult, Vec<&[u8]>> { fold_repeat(0.., "", Vec::new, fold_into_vec).parse_peek(i) } assert_eq!( multi_empty(Partial::new(&b"abcdef"[..])), Err(ErrMode::Backtrack(error_position!( &Partial::new(&b"abcdef"[..]), ErrorKind::Assert ))) ); } #[test] #[cfg(feature = "alloc")] fn fold_repeat1_test() { fn fold_into_vec(mut acc: Vec, item: T) -> Vec { acc.push(item); acc } fn multi(i: Partial<&[u8]>) -> IResult, Vec<&[u8]>> { fold_repeat(1.., "abcd", Vec::new, fold_into_vec).parse_peek(i) } let a = &b"abcdef"[..]; let b = &b"abcdabcdefgh"[..]; let c = &b"azerty"[..]; let d = &b"abcdab"[..]; let res1 = vec![&b"abcd"[..]]; assert_eq!(multi(Partial::new(a)), Ok((Partial::new(&b"ef"[..]), res1))); let res2 = vec![&b"abcd"[..], &b"abcd"[..]]; assert_eq!( multi(Partial::new(b)), Ok((Partial::new(&b"efgh"[..]), res2)) ); assert_eq!( multi(Partial::new(c)), Err(ErrMode::Backtrack(error_position!( &Partial::new(c), ErrorKind::Many ))) ); assert_eq!( multi(Partial::new(d)), Err(ErrMode::Incomplete(Needed::new(2))) ); } #[test] #[cfg(feature = "alloc")] fn fold_repeat_test() { fn fold_into_vec(mut acc: Vec, item: T) -> Vec { acc.push(item); acc } fn multi(i: Partial<&[u8]>) -> IResult, Vec<&[u8]>> { fold_repeat(2..=4, "Abcd", Vec::new, fold_into_vec).parse_peek(i) } let a = &b"Abcdef"[..]; let b = &b"AbcdAbcdefgh"[..]; let c = &b"AbcdAbcdAbcdAbcdefgh"[..]; let d = &b"AbcdAbcdAbcdAbcdAbcdefgh"[..]; let e = &b"AbcdAb"[..]; assert_eq!( multi(Partial::new(a)), Err(ErrMode::Backtrack(error_position!( &Partial::new(&b"ef"[..]), ErrorKind::Tag ))) ); let res1 = vec![&b"Abcd"[..], &b"Abcd"[..]]; assert_eq!( multi(Partial::new(b)), Ok((Partial::new(&b"efgh"[..]), res1)) ); let res2 = vec![&b"Abcd"[..], &b"Abcd"[..], &b"Abcd"[..], &b"Abcd"[..]]; assert_eq!( multi(Partial::new(c)), Ok((Partial::new(&b"efgh"[..]), res2)) ); let res3 = vec![&b"Abcd"[..], &b"Abcd"[..], &b"Abcd"[..], &b"Abcd"[..]]; assert_eq!( multi(Partial::new(d)), Ok((Partial::new(&b"Abcdefgh"[..]), res3)) ); assert_eq!( multi(Partial::new(e)), Err(ErrMode::Incomplete(Needed::new(2))) ); } #[test] fn repeat0_count_test() { fn count0_nums(i: &[u8]) -> IResult<&[u8], usize> { repeat(0.., (digit, ",")).parse_peek(i) } assert_eq!(count0_nums(&b"123,junk"[..]), Ok((&b"junk"[..], 1))); assert_eq!(count0_nums(&b"123,45,junk"[..]), Ok((&b"junk"[..], 2))); assert_eq!( count0_nums(&b"1,2,3,4,5,6,7,8,9,0,junk"[..]), Ok((&b"junk"[..], 10)) ); assert_eq!(count0_nums(&b"hello"[..]), Ok((&b"hello"[..], 0))); } #[test] fn repeat1_count_test() { fn count1_nums(i: &[u8]) -> IResult<&[u8], usize> { repeat(1.., (digit, ",")).parse_peek(i) } assert_eq!(count1_nums(&b"123,45,junk"[..]), Ok((&b"junk"[..], 2))); assert_eq!( count1_nums(&b"1,2,3,4,5,6,7,8,9,0,junk"[..]), Ok((&b"junk"[..], 10)) ); assert_eq!( count1_nums(&b"hello"[..]), Err(ErrMode::Backtrack(error_position!( &&b"hello"[..], ErrorKind::Slice ))) ); }