fix build error

argoengine · Jan 18, 2022 · 12ea08d · 12ea08d
1 parent 57e032e
commit 12ea08d
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/ci/tests/argo_engine_server_test.rs b/ci/tests/argo_engine_server_test.rs
@@ -1,3 +1,7 @@
+use datafusion::assert_batches_eq;
+use common::config::ArgoEngineConfig;
+use rust_client::argo_engine_context::ArgoEngineContext;
+
 #[cfg(feature = "standalone")]
 #[tokio::test]
 async fn test_argo_engine_server() {

diff --git a/ci/tests/stream_load_flight_service_test.rs b/ci/tests/stream_load_flight_service_test.rs
@@ -1,4 +1,8 @@
+use arrow_flight::{Action, Criteria, flight_descriptor, FlightDescriptor, Ticket};
 use tokio;
+use common::config::ArgoEngineConfig;
+use rust_client::argo_engine_context::ArgoEngineContext;
+use rust_client::steam_load_flight_client::StreamLoadFlightClient;
 
 #[cfg(feature = "standalone")]
 #[tokio::test]