test: IsError trait

Author: kthui

lib/llm/src/protocols/common/llm_backend.rs

@@ -159,3 +159,26 @@ pub struct EmbeddingsEngineOutput {
     pub prompt_tokens: u32,
     pub total_tokens: u32,
 }
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn test_is_error() {
+        let output = LLMEngineOutput::stop();
+        assert!(output.err().is_none());
+        assert!(output.is_ok());
+        assert!(!output.is_err());
+
+        let output = LLMEngineOutput::error("Test error".to_string());
+        assert_eq!(format!("{}", output.err().unwrap()), "Test error");
+        assert!(!output.is_ok());
+        assert!(output.is_err());
+
+        let output = LLMEngineOutput::from_err(anyhow::Error::msg("Test error 2").into());
+        assert_eq!(format!("{}", output.err().unwrap()), "Test error 2");
+        assert!(!output.is_ok());
+        assert!(output.is_err());
+    }
+}

lib/runtime/src/protocols/annotated.rs

@@ -15,6 +15,7 @@
 
 use super::*;
 use crate::{error, Result};
+use is_error::IsError;
 
 pub trait AnnotationsProvider {
     fn annotations(&self) -> Option<Vec<String>>;
@@ -146,7 +147,7 @@ impl<R> Annotated<R> {
     }
 }
 
-impl<R> is_error::IsError for Annotated<R>
+impl<R> IsError for Annotated<R>
 where
     R: for<'de> Deserialize<'de> + Serialize,
 {
@@ -188,3 +189,27 @@ where
 //         Box::pin(stream)
 //     }
 // }
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn test_is_error() {
+        let annotated = Annotated::from_data("Test data".to_string());
+        assert!(annotated.err().is_none());
+        assert!(annotated.is_ok());
+        assert!(!annotated.is_err());
+
+        let annotated = Annotated::<String>::from_error("Test error 2".to_string());
+        assert_eq!(format!("{}", annotated.err().unwrap()), "Test error 2");
+        assert!(!annotated.is_ok());
+        assert!(annotated.is_err());
+
+        let annotated =
+            Annotated::<String>::from_err(anyhow::Error::msg("Test error 3".to_string()).into());
+        assert_eq!(format!("{}", annotated.err().unwrap()), "Test error 3");
+        assert!(!annotated.is_ok());
+        assert!(annotated.is_err());
+    }
+}

lib/runtime/src/protocols/is_error.rs

@@ -32,3 +32,30 @@ pub trait IsError {
         self.err().is_some()
     }
 }
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    struct TestError {
+        message: String,
+    }
+    impl IsError for TestError {
+        fn from_err(err: Box<dyn Error>) -> Self {
+            TestError {
+                message: err.to_string(),
+            }
+        }
+        fn err(&self) -> Option<Box<dyn Error>> {
+            Some(anyhow::Error::msg(self.message.clone()).into())
+        }
+    }
+
+    #[test]
+    fn test_is_error_default_implementations() {
+        let err = TestError::from_err(anyhow::Error::msg("Test error".to_string()).into());
+        assert_eq!(format!("{}", err.err().unwrap()), "Test error");
+        assert!(!err.is_ok());
+        assert!(err.is_err());
+    }
+}