alamb commented on code in PR #6201:
URL: https://github.com/apache/arrow-rs/pull/6201#discussion_r1718870736
##########
arrow-flight/tests/flight_sql_client.rs:
##########
@@ -63,12 +68,83 @@ pub async fn test_begin_end_transaction() {
.is_err());
}
+#[tokio::test]
+pub async fn test_execute_ingest() {
+ let test_server = FlightSqlServiceImpl::new();
+ let fixture = TestFixture::new(test_server.service()).await;
+ let channel = fixture.channel().await;
+ let mut flight_sql_client = FlightSqlServiceClient::new(channel);
+ let cmd = make_ingest_command();
+ let expected_rows = 10;
+ let batches = vec![
+ make_primitive_batch(5),
+ make_primitive_batch(3),
+ make_primitive_batch(2),
+ ];
+ let actual_rows = flight_sql_client
+ .execute_ingest(cmd, futures::stream::iter(batches.clone()).map(Ok))
+ .await
+ .expect("ingest should succeed");
+ assert_eq!(actual_rows, expected_rows);
+ // make sure the batches made it through to the server
+ let ingested_batches = test_server.ingested_batches.lock().await.clone();
+ assert_eq!(ingested_batches, batches);
+}
+
+#[tokio::test]
+pub async fn test_execute_ingest_error() {
+ let test_server = FlightSqlServiceImpl::new();
+ let fixture = TestFixture::new(test_server.service()).await;
+ let channel = fixture.channel().await;
+ let mut flight_sql_client = FlightSqlServiceClient::new(channel);
+ let cmd = make_ingest_command();
+ // send an error from the client
+ let batches = vec![
Review Comment:
I also added a test showing that the error passed to the stream is returned
back
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
