feat: Implement is_error() for ResponseType

martindisch · Jan 22, 2025 · 41e3bea · 41e3bea
1 parent 4a45bd5
commit 41e3bea
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/src/header.rs b/src/header.rs
@@ -243,6 +243,13 @@ pub enum ResponseType {
     UnKnown,
 }
 
+impl ResponseType {
+    pub fn is_error(&self) -> bool {
+        MessageClass::Response(*self)
+            >= MessageClass::Response(ResponseType::BadRequest)
+    }
+}
+
 /// The message types.
 #[derive(Debug, Clone, Copy, PartialEq)]
 pub enum MessageType {
@@ -418,4 +425,13 @@ mod test {
         h.set_type(MessageType::Reset);
         assert_eq!(MessageType::Reset, h.get_type());
     }
+
+    #[test]
+    fn is_error() {
+        assert!(!ResponseType::Created.is_error());
+        assert!(!ResponseType::Continue.is_error());
+        assert!(ResponseType::BadRequest.is_error());
+        assert!(ResponseType::TooManyRequests.is_error());
+        assert!(ResponseType::HopLimitReached.is_error());
+    }
 }