CI fixes

Author: Brent Gardner

datafusion/core/src/datasource/custom.rs

@@ -1,3 +1,20 @@
+// Licensed to the Apache Software Foundation (ASF) under one
+// or more contributor license agreements.  See the NOTICE file
+// distributed with this work for additional information
+// regarding copyright ownership.  The ASF licenses this file
+// to you under the Apache License, Version 2.0 (the
+// "License"); you may not use this file except in compliance
+// with the License.  You may obtain a copy of the License at
+//
+//   http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing,
+// software distributed under the License is distributed on an
+// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+// KIND, either express or implied.  See the License for the
+// specific language governing permissions and limitations
+// under the License.
+
 use crate::datasource::TableProvider;
 use crate::execution::context::SessionState;
 use crate::physical_plan::ExecutionPlan;

datafusion/core/tests/sql/create_drop.rs

@@ -405,13 +405,22 @@ impl TableProviderFactory for TestTableFactory {
     async fn create(&self, _name: &str, _url: &str) -> Result<Arc<dyn TableProvider>> {
         Ok(Arc::new(TestTableProvider {}))
     }
+
+    fn with_schema(
+        &self,
+        _schema: SchemaRef,
+        _table_type: &str,
+        _url: &str,
+    ) -> Result<Arc<dyn TableProvider>> {
+        Ok(Arc::new(TestTableProvider {}))
+    }
 }
 
 #[tokio::test]
 async fn create_custom_table() -> Result<()> {
     let mut table_factories: HashMap<String, Arc<dyn TableProviderFactory>> =
         HashMap::new();
-    table_factories.insert("DELTATABLE".to_string(), Arc::new(TestTableFactory {}));
+    table_factories.insert("deltatable".to_string(), Arc::new(TestTableFactory {}));
     let cfg = RuntimeConfig::new().with_table_factories(table_factories);
     let env = RuntimeEnv::new(cfg).unwrap();
     let ses = SessionConfig::new();