From e082d2f4f40493b119510dc178847ee4414bf3d2 Mon Sep 17 00:00:00 2001
From: Edward Rudd <urkle@outoforder.cc>
Date: Tue, 25 Oct 2022 15:26:36 -0400
Subject: [PATCH] fixup! add support for the imap quota extension (RFC 2087)

---
 src/client.rs      |  7 ++++---
 src/types/quota.rs | 10 ++++------
 2 files changed, 8 insertions(+), 9 deletions(-)

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)),
         }
     }
 }