diff --git a/src/client.rs b/src/client.rs index 3d1b20c..e06d09e 100644 --- a/src/client.rs +++ b/src/client.rs @@ -2556,9 +2556,10 @@ mod tests { #[test] fn get_quota_with_limits() { - let response = b"* QUOTA my_root (STORAGE 10 500)\r\n\ - a1 OK completed\r\n" - .to_vec(); + let response = b"* QUOTA my_root (STORAGE 10 500)\r +a1 OK completed\r +" + .to_vec(); let mock_stream = MockStream::new(response); let mut session = mock_session!(mock_stream); let quota = session.get_quota("my_root").unwrap(); diff --git a/src/types/quota.rs b/src/types/quota.rs index a6de6c2..84f399f 100644 --- a/src/types/quota.rs +++ b/src/types/quota.rs @@ -49,12 +49,10 @@ pub enum QuotaResourceName<'a> { impl<'a> From<&'a str> for QuotaResourceName<'a> { fn from(input: &'a str) -> Self { - if input == "STORAGE" { - QuotaResourceName::Storage - } else if input == "MESSAGE" { - QuotaResourceName::Message - } else { - QuotaResourceName::Atom(Cow::from(input)) + match input { + "STORAGE" => QuotaResourceName::Storage, + "MESSAGE" => QuotaResourceName::Message, + _ => QuotaResourceName::Atom(Cow::from(input)), } } }