| 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; |
| #[cfg(feature = "alloc")] |
| use crate::lib::std::borrow::ToOwned; |
| 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))); |
| } |
| |
| use crate::lib::std::convert::From; |
| impl From<u32> for CustomError { |
| fn from(_: u32) -> Self { |
| CustomError |
| } |
| } |
| |
| impl<I: Stream> ParserError<I> for CustomError { |
| fn from_error_kind(_: &I, _: ErrorKind) -> Self { |
| CustomError |
| } |
| |
| fn append(self, _: &I, _: &<I as Stream>::Checkpoint, _: 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<u8>> = 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<Partial<&[u8]>, 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_literal(i: Partial<&[u8]>) -> IResult<Partial<&[u8]>, &[u8]> { |
| peek("abcd").parse_peek(i) |
| } |
| |
| assert_eq!( |
| peek_literal(Partial::new(&b"abcdef"[..])), |
| Ok((Partial::new(&b"abcdef"[..]), &b"abcd"[..])) |
| ); |
| assert_eq!( |
| peek_literal(Partial::new(&b"ab"[..])), |
| Err(ErrMode::Incomplete(Needed::new(2))) |
| ); |
| assert_eq!( |
| peek_literal(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<Partial<&[u8]>, ()> { |
| 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<Partial<&[u8]>, &[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<Partial<&[u8]>, (&[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<Partial<&[u8]>, &[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<Partial<&[u8]>, &[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<Partial<&[u8]>, &[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}, |
| }; |
| |
| #[cfg(feature = "alloc")] |
| #[derive(Debug, Clone, Eq, PartialEq)] |
| struct ErrorStr(String); |
| |
| #[cfg(feature = "alloc")] |
| impl From<u32> 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<I: Stream + Debug> ParserError<I> for ErrorStr { |
| fn from_error_kind(input: &I, kind: ErrorKind) -> Self { |
| ErrorStr(format!("custom error message: ({input:?}, {kind:?})")) |
| } |
| |
| fn append(self, input: &I, _: &<I as Stream>::Checkpoint, 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_owned()))) |
| } |
| |
| 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_owned()) |
| ))) |
| ); |
| 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<Partial<&[u8]>, &[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 alt_dynamic_array() { |
| fn alt1<'i>(i: &mut &'i [u8]) -> PResult<&'i [u8]> { |
| alt(&mut ["a", "bc", "def"][..]).parse_next(i) |
| } |
| |
| let a = &b"a"[..]; |
| assert_eq!(alt1.parse_peek(a), Ok((&b""[..], (&b"a"[..])))); |
| |
| let bc = &b"bc"[..]; |
| assert_eq!(alt1.parse_peek(bc), Ok((&b""[..], (&b"bc"[..])))); |
| |
| let defg = &b"defg"[..]; |
| assert_eq!(alt1.parse_peek(defg), Ok((&b"g"[..], (&b"def"[..])))); |
| } |
| |
| #[test] |
| fn permutation_test() { |
| #[allow(clippy::type_complexity)] |
| fn perm(i: Partial<&[u8]>) -> IResult<Partial<&[u8]>, (&[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<Partial<&[u8]>, Vec<&[u8]>> { |
| separated(0.., "abcd", ",").parse_peek(i) |
| } |
| fn multi_empty(i: Partial<&[u8]>) -> IResult<Partial<&[u8]>, Vec<&[u8]>> { |
| separated(0.., "", ",").parse_peek(i) |
| } |
| fn multi_longsep(i: Partial<&[u8]>) -> IResult<Partial<&[u8]>, Vec<&[u8]>> { |
| separated(0.., "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<Partial<&[u8]>, Vec<&[u8]>> { |
| separated(0.., "abc", "").parse_peek(i) |
| } |
| |
| let i = &b"abcabc"[..]; |
| |
| let i_err_pos = &i[3..]; |
| assert_eq!( |
| empty_sep(Partial::new(i)), |
| Err(ErrMode::Cut(error_position!( |
| &Partial::new(i_err_pos), |
| ErrorKind::Assert |
| ))) |
| ); |
| } |
| |
| #[test] |
| #[cfg(feature = "alloc")] |
| fn separated1_test() { |
| fn multi(i: Partial<&[u8]>) -> IResult<Partial<&[u8]>, Vec<&[u8]>> { |
| separated(1.., "abcd", ",").parse_peek(i) |
| } |
| fn multi_longsep(i: Partial<&[u8]>) -> IResult<Partial<&[u8]>, Vec<&[u8]>> { |
| separated(1.., "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 separated_test() { |
| fn multi(i: Partial<&[u8]>) -> IResult<Partial<&[u8]>, Vec<&[u8]>> { |
| separated(2..=4, "abcd", ",").parse_peek(i) |
| } |
| |
| let a = &b"abcd,ef"[..]; |
| let b = &b"abcd,abcd,efgh"[..]; |
| let c = &b"abcd,abcd,abcd,abcd,efgh"[..]; |
| let d = &b"abcd,abcd,abcd,abcd,abcd,efgh"[..]; |
| let e = &b"abcd,ab"[..]; |
| |
| 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",abcd,efgh"[..]), res3)) |
| ); |
| assert_eq!( |
| multi(Partial::new(e)), |
| Err(ErrMode::Incomplete(Needed::new(2))) |
| ); |
| } |
| |
| #[test] |
| #[cfg(feature = "alloc")] |
| fn repeat0_test() { |
| fn multi(i: Partial<&[u8]>) -> IResult<Partial<&[u8]>, 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<Partial<&[u8]>, Vec<&[u8]>> { |
| repeat(0.., "").parse_peek(i) |
| } |
| |
| assert_eq!( |
| multi_empty(Partial::new(&b"abcdef"[..])), |
| Err(ErrMode::Cut(error_position!( |
| &Partial::new(&b"abcdef"[..]), |
| ErrorKind::Assert |
| ))) |
| ); |
| } |
| |
| #[test] |
| #[cfg(feature = "alloc")] |
| fn repeat1_test() { |
| fn multi(i: Partial<&[u8]>) -> IResult<Partial<&[u8]>, 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_till(0.., "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 = "alloc")] |
| fn repeat_till_range_test() { |
| #[allow(clippy::type_complexity)] |
| fn multi(i: &str) -> IResult<&str, (Vec<&str>, &str)> { |
| repeat_till(2..=4, "ab", "cd").parse_peek(i) |
| } |
| |
| assert_eq!( |
| multi("cd"), |
| Err(ErrMode::Backtrack(error_node_position!( |
| &"cd", |
| ErrorKind::Many, |
| error_position!(&"cd", ErrorKind::Tag) |
| ))) |
| ); |
| assert_eq!( |
| multi("abcd"), |
| Err(ErrMode::Backtrack(error_node_position!( |
| &"cd", |
| ErrorKind::Many, |
| error_position!(&"cd", ErrorKind::Tag) |
| ))) |
| ); |
| assert_eq!(multi("ababcd"), Ok(("", (vec!["ab", "ab"], "cd")))); |
| assert_eq!(multi("abababcd"), Ok(("", (vec!["ab", "ab", "ab"], "cd")))); |
| assert_eq!( |
| multi("ababababcd"), |
| Ok(("", (vec!["ab", "ab", "ab", "ab"], "cd"))) |
| ); |
| assert_eq!( |
| multi("abababababcd"), |
| Err(ErrMode::Backtrack(error_node_position!( |
| &"cd", |
| ErrorKind::Many, |
| error_position!(&"abcd", 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<Partial<&[u8]>, 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<Partial<&[u8]>, 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)] |
| struct NilError; |
| |
| impl<I> From<(I, ErrorKind)> for NilError { |
| fn from(_: (I, ErrorKind)) -> Self { |
| NilError |
| } |
| } |
| |
| impl<I: Stream> ParserError<I> for NilError { |
| fn from_error_kind(_: &I, _: ErrorKind) -> NilError { |
| NilError |
| } |
| fn append(self, _: &I, _: &<I as Stream>::Checkpoint, _: ErrorKind) -> NilError { |
| NilError |
| } |
| } |
| |
| #[test] |
| #[cfg(feature = "alloc")] |
| fn fold_repeat0_test() { |
| fn fold_into_vec<T>(mut acc: Vec<T>, item: T) -> Vec<T> { |
| acc.push(item); |
| acc |
| } |
| fn multi(i: Partial<&[u8]>) -> IResult<Partial<&[u8]>, Vec<&[u8]>> { |
| repeat(0.., "abcd") |
| .fold(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<T>(mut acc: Vec<T>, item: T) -> Vec<T> { |
| acc.push(item); |
| acc |
| } |
| fn multi_empty(i: Partial<&[u8]>) -> IResult<Partial<&[u8]>, Vec<&[u8]>> { |
| repeat(0.., "").fold(Vec::new, fold_into_vec).parse_peek(i) |
| } |
| |
| assert_eq!( |
| multi_empty(Partial::new(&b"abcdef"[..])), |
| Err(ErrMode::Cut(error_position!( |
| &Partial::new(&b"abcdef"[..]), |
| ErrorKind::Assert |
| ))) |
| ); |
| } |
| |
| #[test] |
| #[cfg(feature = "alloc")] |
| fn fold_repeat1_test() { |
| fn fold_into_vec<T>(mut acc: Vec<T>, item: T) -> Vec<T> { |
| acc.push(item); |
| acc |
| } |
| fn multi(i: Partial<&[u8]>) -> IResult<Partial<&[u8]>, Vec<&[u8]>> { |
| repeat(1.., "abcd") |
| .fold(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<T>(mut acc: Vec<T>, item: T) -> Vec<T> { |
| acc.push(item); |
| acc |
| } |
| fn multi(i: Partial<&[u8]>) -> IResult<Partial<&[u8]>, Vec<&[u8]>> { |
| repeat(2..=4, "Abcd") |
| .fold(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 |
| ))) |
| ); |
| } |
