Expose more components from sqllogictest (apache#14249)

Author: xudong963

datafusion/sqllogictest/src/engines/datafusion_engine/mod.rs

@@ -21,4 +21,5 @@ mod normalize;
 mod runner;
 
 pub use error::*;
+pub use normalize::*;
 pub use runner::*;

datafusion/sqllogictest/src/engines/datafusion_engine/normalize.rs

@@ -28,8 +28,8 @@ use std::sync::OnceLock;
 use super::super::conversion::*;
 use super::error::{DFSqlLogicTestError, Result};
 
-/// Converts `batches` to a result as expected by sqllogicteset.
-pub(crate) fn convert_batches(batches: Vec<RecordBatch>) -> Result<Vec<Vec<String>>> {
+/// Converts `batches` to a result as expected by sqllogictest.
+pub fn convert_batches(batches: Vec<RecordBatch>) -> Result<Vec<Vec<String>>> {
     if batches.is_empty() {
         Ok(vec![])
     } else {

datafusion/sqllogictest/src/engines/mod.rs

@@ -20,7 +20,10 @@ mod conversion;
 mod datafusion_engine;
 mod output;
 
+pub use datafusion_engine::convert_batches;
 pub use datafusion_engine::DataFusion;
+pub use output::DFColumnType;
+pub use output::DFOutput;
 
 #[cfg(feature = "postgres")]
 mod postgres_engine;

datafusion/sqllogictest/src/engines/output.rs

@@ -54,4 +54,4 @@ impl ColumnType for DFColumnType {
     }
 }
 
-pub(crate) type DFOutput = DBOutput<DFColumnType>;
+pub type DFOutput = DBOutput<DFColumnType>;

datafusion/sqllogictest/src/lib.rs

@@ -19,6 +19,9 @@
 
 mod engines;
 
+pub use engines::convert_batches;
+pub use engines::DFColumnType;
+pub use engines::DFOutput;
 pub use engines::DataFusion;
 
 #[cfg(feature = "postgres")]