add integration test

Author: andygrove

rust/datafusion/src/datasource/parquet.rs

@@ -54,6 +54,7 @@ impl ParquetTable {
 }
 
 impl Table for ParquetTable {
+
     fn schema(&self) -> &Arc<Schema> {
         &self.schema
     }

rust/datafusion/tests/sql.rs

@@ -16,6 +16,7 @@
 // under the License.
 
 use std::cell::RefCell;
+use std::env;
 use std::rc::Rc;
 use std::sync::Arc;
 
@@ -27,9 +28,22 @@ use arrow::datatypes::{DataType, Field, Schema};
 
 use datafusion::execution::context::ExecutionContext;
 use datafusion::execution::relation::Relation;
+use datafusion::datasource::parquet::ParquetFile;
+use datafusion::datasource::parquet::ParquetTable;
+use datafusion::datasource::{RecordBatchIterator, Table};
 
 const DEFAULT_BATCH_SIZE: usize = 1024 * 1024;
 
+#[test]
+fn parquet_query() {
+    let mut ctx = ExecutionContext::new();
+    ctx.register_table("alltypes_plain", load_parquet_table("alltypes_plain.parquet"));
+    let sql = "SELECT id, string_col FROM alltypes_plain";
+    let actual = execute(&mut ctx, sql);
+    let expected = "tbd".to_string();
+    assert_eq!(expected, actual);
+}
+
 #[test]
 fn csv_query_with_predicate() {
     let mut ctx = ExecutionContext::new();
@@ -163,6 +177,14 @@ fn register_csv(
     ctx.register_csv(name, filename, &schema, true);
 }
 
+fn load_parquet_table(name: &str) -> Rc<Table> {
+    let testdata = env::var("PARQUET_TEST_DATA").unwrap();
+    let filename = format!("{}/{}", testdata, name);
+    let table = ParquetTable::new(&filename);
+    println!("{:?}", table.schema());
+    Rc::new(table)
+}
+
 /// Execute query and return result set as tab delimited string
 fn execute(ctx: &mut ExecutionContext, sql: &str) -> String {
     let results = ctx.sql(&sql, DEFAULT_BATCH_SIZE).unwrap();