cover more cases

PostHog · Nov 23, 2023 · fc2ea91 · fc2ea91
1 parent 3dd5b3c
commit fc2ea91
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/capture/src/sink.rs b/capture/src/sink.rs
@@ -359,7 +359,7 @@ mod tests {
         cluster.clear_request_errors(RDKafkaApiKey::Produce);
         let err = [RDKafkaRespErr::RD_KAFKA_RESP_ERR_INVALID_PARTITIONS; 1];
         cluster.request_errors(RDKafkaApiKey::Produce, &err);
-        match sink.send(event.clone()).await {
+        match sink.send_batch(vec![event.clone(), event.clone()]).await {
             Err(CaptureError::RetryableSinkError) => {} // Expected
             Err(err) => panic!("wrong error code {}", err),
             Ok(()) => panic!("should have errored"),
@@ -388,5 +388,10 @@ mod tests {
             Err(err) => panic!("wrong error code {}", err),
             Ok(()) => panic!("should have errored"),
         };
+        match sink.send_batch(vec![event.clone(), event.clone()]).await {
+            Err(CaptureError::RetryableSinkError) => {} // Expected
+            Err(err) => panic!("wrong error code {}", err),
+            Ok(()) => panic!("should have errored"),
+        };
     }
 }