Standardize RUST_LOG configuration test setup (#6506)

datafusion/core/tests/memory_limit.rs

@@ -39,6 +39,7 @@ use test_utils::AccessLogGenerator;
 #[cfg(test)]
 #[ctor::ctor]
 fn init() {
+    // Enable RUST_LOG logging configuration for test
     let _ = env_logger::try_init();
 }
 

datafusion/core/tests/parquet/filter_pushdown.rs

@@ -63,13 +63,6 @@ fn generate_file(tempdir: &TempDir, props: WriterProperties) -> TestParquetFile
     test_parquet_file
 }
 
-#[cfg(test)]
-#[ctor::ctor]
-fn init() {
-    // enable logging so RUST_LOG works
-    let _ = env_logger::try_init();
-}
-
 #[cfg(not(target_family = "windows"))]
 #[tokio::test]
 async fn single_file() {

datafusion/core/tests/parquet/mod.rs

@@ -47,6 +47,13 @@ mod filter_pushdown;
 mod page_pruning;
 mod row_group_pruning;
 
+#[cfg(test)]
+#[ctor::ctor]
+fn init() {
+    // Enable RUST_LOG logging configuration for test
+    let _ = env_logger::try_init();
+}
+
 // ----------------------
 // Begin test fixture
 // ----------------------

datafusion/core/tests/sql/subqueries.rs

@@ -21,12 +21,6 @@ use datafusion::assert_batches_eq;
 use datafusion::prelude::SessionContext;
 use log::debug;
 
-#[cfg(test)]
-#[ctor::ctor]
-fn init() {
-    let _ = env_logger::try_init();
-}
-
 #[tokio::test]
 async fn correlated_recursive_scalar_subquery() -> Result<()> {
     let ctx = SessionContext::new();

datafusion/core/tests/sql_integration.rs

@@ -17,3 +17,10 @@
 
 /// Run all tests that are found in the `sql` directory
 mod sql;
+
+#[cfg(test)]
+#[ctor::ctor]
+fn init() {
+    // Enable RUST_LOG logging configuration for test
+    let _ = env_logger::try_init();
+}

datafusion/expr/src/expr_rewriter/mod.rs

@@ -224,11 +224,6 @@ mod test {
     use datafusion_common::tree_node::{RewriteRecursion, TreeNode, TreeNodeRewriter};
     use datafusion_common::{DFField, DFSchema, ScalarValue};
 
-    #[ctor::ctor]
-    fn init() {
-        let _ = env_logger::try_init();
-    }
-
     #[derive(Default)]
     struct RecordingRewriter {
         v: Vec<String>,

datafusion/expr/src/lib.rs

@@ -76,3 +76,10 @@ pub use udaf::AggregateUDF;
 pub use udf::ScalarUDF;
 pub use window_frame::{WindowFrame, WindowFrameBound, WindowFrameUnits};
 pub use window_function::{BuiltInWindowFunction, WindowFunction};
+
+#[cfg(test)]
+#[ctor::ctor]
+fn init() {
+    // Enable RUST_LOG logging configuration for test
+    let _ = env_logger::try_init();
+}

datafusion/optimizer/src/lib.rs

@@ -53,6 +53,6 @@ mod plan_signature;
 #[cfg(test)]
 #[ctor::ctor]
 fn init() {
-    // Enable logging for tests
+    // Enable RUST_LOG logging configuration for test
     let _ = env_logger::try_init();
 }

datafusion/optimizer/tests/integration-test.rs

@@ -35,6 +35,7 @@ use std::sync::Arc;
 #[cfg(test)]
 #[ctor::ctor]
 fn init() {
+    // enable logging so RUST_LOG works
     let _ = env_logger::try_init();
 }
 

datafusion/sql/tests/integration_test.rs

@@ -40,6 +40,7 @@ use rstest::rstest;
 #[cfg(test)]
 #[ctor::ctor]
 fn init() {
+    // Enable RUST_LOG logging configuration for tests
     let _ = env_logger::try_init();
 }