mirror of
https://github.com/sile/hls_m3u8.git
synced 2024-06-02 07:50:30 +00:00
c53e9e33f1
This will allow for better troubleshooting of failing test, because you don't have to search for the difference (between left and right). This is especially helpful for larger assertions.
127 lines
3.6 KiB
Rust
127 lines
3.6 KiB
Rust
use crate::Error;
|
|
|
|
macro_rules! required_version {
|
|
( $( $tag:expr ),* ) => {
|
|
::core::iter::empty()
|
|
$(
|
|
.chain(::core::iter::once($tag.required_version()))
|
|
)*
|
|
.max()
|
|
.unwrap_or_default()
|
|
}
|
|
}
|
|
|
|
macro_rules! impl_from {
|
|
( $($( $type:tt ),* => $target:path ),* ) => {
|
|
use ::core::convert::From;
|
|
|
|
$( // repeat $target
|
|
$( // repeat $type
|
|
impl From<$type> for $target {
|
|
fn from(value: $type) -> Self {
|
|
Self::from_f64_unchecked(value.into())
|
|
}
|
|
}
|
|
)*
|
|
)*
|
|
};
|
|
}
|
|
|
|
impl_from![
|
|
u8, u16, u32 => crate::types::DecimalFloatingPoint,
|
|
u8, i8, u16, i16, u32, i32, f32, f64 => crate::types::SignedDecimalFloatingPoint
|
|
];
|
|
|
|
pub(crate) fn parse_yes_or_no<T: AsRef<str>>(s: T) -> crate::Result<bool> {
|
|
match s.as_ref() {
|
|
"YES" => Ok(true),
|
|
"NO" => Ok(false),
|
|
_ => Err(Error::invalid_input()),
|
|
}
|
|
}
|
|
|
|
/// According to the documentation the following characters are forbidden
|
|
/// inside a quoted string:
|
|
/// - carriage return (`\r`)
|
|
/// - new line (`\n`)
|
|
/// - double quotes (`"`)
|
|
///
|
|
/// Therefore it is safe to simply remove any occurence of those characters.
|
|
/// [rfc8216#section-4.2](https://tools.ietf.org/html/rfc8216#section-4.2)
|
|
pub(crate) fn unquote<T: ToString>(value: T) -> String {
|
|
value
|
|
.to_string()
|
|
.replace("\"", "")
|
|
.replace("\n", "")
|
|
.replace("\r", "")
|
|
}
|
|
|
|
/// Puts a string inside quotes.
|
|
pub(crate) fn quote<T: ToString>(value: T) -> String {
|
|
// the replace is for the case, that quote is called on an already quoted
|
|
// string, which could cause problems!
|
|
format!("\"{}\"", value.to_string().replace("\"", ""))
|
|
}
|
|
|
|
/// Checks, if the given tag is at the start of the input. If this is the case,
|
|
/// it will remove it and return the rest of the input.
|
|
///
|
|
/// # Error
|
|
/// This function will return `Error::MissingTag`, if the input doesn't start
|
|
/// with the tag, that has been passed to this function.
|
|
pub(crate) fn tag<T>(input: &str, tag: T) -> crate::Result<&str>
|
|
where
|
|
T: AsRef<str>,
|
|
{
|
|
if !input.trim().starts_with(tag.as_ref()) {
|
|
return Err(Error::missing_tag(tag.as_ref(), input));
|
|
}
|
|
let result = input.split_at(tag.as_ref().len()).1;
|
|
Ok(result)
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use super::*;
|
|
use pretty_assertions::assert_eq;
|
|
|
|
#[test]
|
|
fn test_parse_yes_or_no() {
|
|
assert!(parse_yes_or_no("YES").unwrap());
|
|
assert!(!parse_yes_or_no("NO").unwrap());
|
|
assert!(parse_yes_or_no("garbage").is_err());
|
|
}
|
|
|
|
#[test]
|
|
fn test_unquote() {
|
|
assert_eq!(unquote("\"TestValue\""), "TestValue".to_string());
|
|
assert_eq!(unquote("\"TestValue\n\""), "TestValue".to_string());
|
|
assert_eq!(unquote("\"TestValue\n\r\""), "TestValue".to_string());
|
|
}
|
|
|
|
#[test]
|
|
fn test_quote() {
|
|
assert_eq!(quote("value"), "\"value\"".to_string());
|
|
assert_eq!(quote("\"value\""), "\"value\"".to_string());
|
|
}
|
|
|
|
#[test]
|
|
fn test_tag() {
|
|
let input = "HelloMyFriendThisIsASampleString";
|
|
|
|
let input = tag(input, "Hello").unwrap();
|
|
assert_eq!(input, "MyFriendThisIsASampleString");
|
|
|
|
let input = tag(input, "My").unwrap();
|
|
assert_eq!(input, "FriendThisIsASampleString");
|
|
|
|
let input = tag(input, "FriendThisIs").unwrap();
|
|
assert_eq!(input, "ASampleString");
|
|
|
|
let input = tag(input, "A").unwrap();
|
|
assert_eq!(input, "SampleString");
|
|
|
|
assert!(tag(input, "B").is_err());
|
|
}
|
|
}
|