use crate::ParseError::InvalidDomain;
use crate::{Domain, DomainRef, ParseError, impl_parse_ref};
impl_parse_ref!(
DomainRef,
"The name must already be in lowercase. Use [`Domain`](crate::Domain) to parse mixed-case input."
);
impl<'a> TryFrom<&'a [u8]> for DomainRef<'a> {
type Error = ParseError;
fn try_from(name: &'a [u8]) -> Result<Self, Self::Error> {
if Domain::is_valid_name(name, false) {
let name: &str = std::str::from_utf8(name).map_err(|_| InvalidDomain)?;
Ok(unsafe { Self::new_unchecked(name) })
} else {
Err(InvalidDomain)
}
}
}
#[cfg(test)]
mod tests {
use crate::ParseError::InvalidDomain;
use crate::{DomainRef, ParseError};
#[test]
fn try_from_str() {
let result: Result<DomainRef, ParseError> = DomainRef::try_from("localhost");
let expected: Result<DomainRef, ParseError> = Ok(DomainRef::LOCALHOST);
assert_eq!(result, expected);
let result: Result<DomainRef, ParseError> = DomainRef::try_from("LocalHost");
let expected: Result<DomainRef, ParseError> = Err(InvalidDomain);
assert_eq!(result, expected);
}
#[test]
fn try_from_slice() {
let test_cases: &[(&[u8], Result<DomainRef, ParseError>)] = &[
("localhost".as_bytes(), Ok(DomainRef::LOCALHOST)),
("LocalHost".as_bytes(), Err(InvalidDomain)),
(b"\xFF".as_slice(), Err(InvalidDomain)),
("ΓΌ".as_bytes(), Err(InvalidDomain)),
];
for (input, expected) in test_cases {
let result: Result<DomainRef, ParseError> = DomainRef::try_from(*input);
assert_eq!(result, *expected, "input={:?}", input);
}
}
}