use crate::ParseError::InvalidDomain;
use crate::{Domain, DomainRef, ParseError, doc_name_lowercase_required, impl_parse_ref};
impl_parse_ref!(DomainRef, doc_name_lowercase_required!());
impl<'a> TryFrom<&'a [u8]> for DomainRef<'a> {
type Error = ParseError;
#[doc = doc_name_lowercase_required!()]
fn try_from(name: &'a [u8]) -> Result<Self, Self::Error> {
if Domain::is_valid_name(name, false) {
let name: &str = unsafe { std::str::from_utf8_unchecked(name) };
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);
}
}
#[test]
fn round_trip() {
let canonical: &[&str] = &[
"localhost",
"example.com",
"a-b.c--d.example",
"xn--bcher-kva.example",
"123.example",
];
for input in canonical {
let value: DomainRef = DomainRef::try_from(*input).unwrap();
assert_eq!(value.to_string(), *input, "input={}", input);
}
}
}