delta & ballista tests pass

datafusion/core/src/datasource/datasource.rs

@@ -28,6 +28,7 @@ use crate::error::Result;
 use crate::execution::context::SessionState;
 use crate::logical_expr::Expr;
 use crate::physical_plan::ExecutionPlan;
+use crate::prelude::SessionContext;
 
 /// Source table
 #[async_trait]
@@ -83,6 +84,10 @@ pub trait TableProviderFactory: Sync + Send {
     async fn create(&self, url: &str) -> Result<Arc<dyn TableProvider>>;
 
     /// Create a TableProvider during execution with schema already known from planning
-    fn with_schema(&self, schema: SchemaRef, url: &str)
-        -> Result<Arc<dyn TableProvider>>;
+    fn with_schema(
+        &self,
+        ctx: &SessionContext,
+        schema: SchemaRef,
+        url: &str,
+    ) -> Result<Arc<dyn TableProvider>>;
 }

datafusion/core/src/test_util.rs

@@ -26,6 +26,7 @@ use crate::datasource::{empty::EmptyTable, provider_as_source, TableProvider};
 use crate::execution::context::SessionState;
 use crate::logical_expr::{LogicalPlanBuilder, UNNAMED_TABLE};
 use crate::physical_plan::ExecutionPlan;
+use crate::prelude::SessionContext;
 use arrow::datatypes::{DataType, Field, Schema, SchemaRef};
 use async_trait::async_trait;
 use datafusion_common::DataFusionError;
@@ -339,6 +340,7 @@ impl TableProviderFactory for TestTableFactory {
 
     fn with_schema(
         &self,
+        _ctx: &SessionContext,
         _schema: SchemaRef,
         url: &str,
     ) -> datafusion_common::Result<Arc<dyn TableProvider>> {

datafusion/proto/src/lib.rs

@@ -183,7 +183,7 @@ mod roundtrip_tests {
                 .table_factories
                 .get("testtable")
                 .expect("Unable to find testtable factory");
-            let provider = (*factory).with_schema(schema, msg.url.as_str())?;
+            let provider = (*factory).with_schema(ctx, schema, msg.url.as_str())?;
             Ok(provider)
         }