format files

src/integration_test.rs

@@ -189,9 +189,9 @@ impl IntegrationTest {
 
 #[cfg(test)]
 mod tests {
+    use super::*;
     use crate::integration_test::IntegrationTest;
     use crate::parser::ExecutionPlanParser;
-    use super::*;
     use datafusion::arrow::array::{Int32Array, RecordBatch};
     use datafusion::arrow::datatypes::{DataType, Field, Schema};
     use std::path::PathBuf;

src/query_graph.rs

@@ -123,7 +123,10 @@ impl QueryGraph {
                     let outputs = stage.outputs.clone();
 
                     if outputs.is_empty() {
-                        println!("QueryGraph::update_stage_status: Query {} is done.", self.query_id);
+                        println!(
+                            "QueryGraph::update_stage_status: Query {} is done.",
+                            self.query_id
+                        );
                         self.status = QueryStatus::Done;
                         return Ok(());
                     }

src/queue.rs

@@ -156,12 +156,15 @@ impl State {
 #[cfg(test)]
 mod tests {
     use rand::Rng;
-    use std::{fs, time::{Duration, SystemTime}};
+    use std::{
+        fs,
+        time::{Duration, SystemTime},
+    };
     use tokio::{sync::Notify, time::sleep};
 
-    use crate::{parser::ExecutionPlanParser, query_graph::QueryGraph};
     use crate::queue::State;
     use crate::task::TaskStatus;
+    use crate::{parser::ExecutionPlanParser, query_graph::QueryGraph};
     use std::{cmp::min, sync::Arc};
 
     // Deprecated, use test_queue_conc instead
@@ -290,9 +293,7 @@ mod tests {
         // Add a bunch of queries with staggered submission time
         let start_enqueue = SystemTime::now();
         for plan in long_plans {
-            queue
-                .add_query(Arc::clone(&plan))
-                .await;
+            queue.add_query(Arc::clone(&plan)).await;
             sleep(Duration::from_millis(10)).await;
         }
         let enq_time = SystemTime::now().duration_since(start_enqueue).unwrap();

src/task_queue.rs

@@ -30,7 +30,6 @@ impl TaskQueue {
     pub fn next_task(&mut self) -> Task {
         self.queue.pop_front().expect("Queue has no tasks.")
     }
-
 }
 
 #[cfg(test)]