chore(transformers): improve test coverage

bosun-ai · Jun 23, 2024 · 7767588 · 7767588
1 parent 4d5c68e
commit 7767588
Show file tree

Hide file tree

Showing 2 changed files with 52 additions and 0 deletions.
diff --git a/swiftide/src/transformers/metadata_qa_code.rs b/swiftide/src/transformers/metadata_qa_code.rs
@@ -142,3 +142,29 @@ impl Transformer for MetadataQACode {
         self.concurrency
     }
 }
+
+#[cfg(test)]
+mod test {
+    use crate::MockSimplePrompt;
+
+    use super::*;
+
+    #[tokio::test]
+    async fn test_metadata_qacode() {
+        let mut client = MockSimplePrompt::new();
+
+        client
+            .expect_prompt()
+            .returning(|_| Ok("Q1: Hello\nA1: World".to_string()));
+
+        let transformer = MetadataQACode::builder().client(client).build().unwrap();
+        let node = IngestionNode::new("Some text");
+
+        let result = transformer.transform_node(node).await.unwrap();
+
+        assert_eq!(
+            result.metadata.get("Questions and Answers").unwrap(),
+            "Q1: Hello\nA1: World"
+        );
+    }
+}
diff --git a/swiftide/src/transformers/metadata_qa_text.rs b/swiftide/src/transformers/metadata_qa_text.rs
@@ -143,3 +143,29 @@ impl Transformer for MetadataQAText {
         self.concurrency
     }
 }
+
+#[cfg(test)]
+mod test {
+    use crate::MockSimplePrompt;
+
+    use super::*;
+
+    #[tokio::test]
+    async fn test_metadata_qacode() {
+        let mut client = MockSimplePrompt::new();
+
+        client
+            .expect_prompt()
+            .returning(|_| Ok("Q1: Hello\nA1: World".to_string()));
+
+        let transformer = MetadataQAText::builder().client(client).build().unwrap();
+        let node = IngestionNode::new("Some text");
+
+        let result = transformer.transform_node(node).await.unwrap();
+
+        assert_eq!(
+            result.metadata.get("Questions and Answers").unwrap(),
+            "Q1: Hello\nA1: World"
+        );
+    }
+}