chore: added test

evilsocket · Dec 10, 2024 · 6262a43 · 6262a43
1 parent a88f53c
commit 6262a43
Showing 1 changed file with 24 additions and 0 deletions.
diff --git a/src/agent/generator/history.rs b/src/agent/generator/history.rs
@@ -291,4 +291,28 @@ mod tests {
         let history = ChatHistory::create(conv, ConversationWindow::Summary);
         assert_eq!(history.history, expected);
     }
+
+    #[test]
+    fn test_last_n_strategy() {
+        let conv = vec![
+            Message::Agent("test1".to_string(), None),
+            Message::Feedback("feedback1".to_string(), None),
+            Message::Agent("test2".to_string(), None),
+            Message::Feedback("feedback2".to_string(), None),
+            Message::Agent("test3".to_string(), None),
+            Message::Feedback("feedback3".to_string(), None),
+            Message::Agent("test4".to_string(), None),
+            Message::Feedback("feedback4".to_string(), None),
+        ];
+
+        let expected = vec![
+            Message::Agent("test3".to_string(), None),
+            Message::Feedback("feedback3".to_string(), None),
+            Message::Agent("test4".to_string(), None),
+            Message::Feedback("feedback4".to_string(), None),
+        ];
+
+        let history = ChatHistory::create(conv, ConversationWindow::LastN(4));
+        assert_eq!(history.history, expected);
+    }
 }