Add test to delete batch

tembo-io · Aug 23, 2023 · e81d0bc · e81d0bc
1 parent 0cc5af5
commit e81d0bc
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/core/tests/integration_test.rs b/core/tests/integration_test.rs
@@ -870,6 +870,15 @@ async fn test_extension_api() {
         .await
         .expect("failed to delete");
     assert!(!deleted);
+
+    // delete a batch of messages
+    let m1 = queue.send(&test_queue, &del_msg).await.unwrap();
+    let m2 = queue.send(&test_queue, &del_msg).await.unwrap();
+    let m3 = queue.send(&test_queue, &del_msg).await.unwrap();
+    let delete_result = queue.delete_batch(&test_queue, &[m1, m2, m3]).await.expect("delete batch error");
+    let rowcount = rowcount(&test_queue, &queue.connection).await;
+    assert_eq!(rowcount, 0);
+    assert_eq!(delete_result, true);
 }
 
 #[tokio::test]