This is an automated email from the ASF dual-hosted git repository.

zhangliang pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/shardingsphere.git


The following commit(s) were added to refs/heads/master by this push:
     new 31c90574185 Add MySQLSQLFederationFunctionRegisterTest and 
MySQLSQLFederationFunctionRegisterTest (#37316)
31c90574185 is described below

commit 31c905741859d3870fa96643b04ec8a432ec1910
Author: Liang Zhang <[email protected]>
AuthorDate: Wed Dec 10 00:13:07 2025 +0800

    Add MySQLSQLFederationFunctionRegisterTest and 
MySQLSQLFederationFunctionRegisterTest (#37316)
---
 ...QLSQLFederationConnectionConfigBuilderTest.java | 52 +++++++++++++++++++
 .../MySQLSQLFederationFunctionRegisterTest.java    | 59 ++++++++++++++++++++++
 2 files changed, 111 insertions(+)

diff --git 
a/kernel/sql-federation/dialect/mysql/src/test/java/org/apache/shardingsphere/sqlfederation/mysql/MySQLSQLFederationConnectionConfigBuilderTest.java
 
b/kernel/sql-federation/dialect/mysql/src/test/java/org/apache/shardingsphere/sqlfederation/mysql/MySQLSQLFederationConnectionConfigBuilderTest.java
new file mode 100644
index 00000000000..b314af1e13f
--- /dev/null
+++ 
b/kernel/sql-federation/dialect/mysql/src/test/java/org/apache/shardingsphere/sqlfederation/mysql/MySQLSQLFederationConnectionConfigBuilderTest.java
@@ -0,0 +1,52 @@
+/*
+ * 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.
+ */
+
+package org.apache.shardingsphere.sqlfederation.mysql;
+
+import org.apache.calcite.config.CalciteConnectionConfig;
+import org.apache.calcite.config.Lex;
+import org.apache.calcite.config.NullCollation;
+import org.apache.calcite.sql.SqlOperatorTable;
+import org.apache.calcite.sql.validate.SqlConformanceEnum;
+import 
org.apache.shardingsphere.database.connector.core.spi.DatabaseTypedSPILoader;
+import org.apache.shardingsphere.database.connector.core.type.DatabaseType;
+import org.apache.shardingsphere.infra.spi.type.typed.TypedSPILoader;
+import 
org.apache.shardingsphere.sqlfederation.compiler.context.connection.config.DialectSQLFederationConnectionConfigBuilder;
+import org.junit.jupiter.api.Test;
+
+import static org.hamcrest.CoreMatchers.is;
+import static org.hamcrest.MatcherAssert.assertThat;
+import static org.junit.jupiter.api.Assertions.assertNotNull;
+import static org.junit.jupiter.api.Assertions.assertTrue;
+
+class MySQLSQLFederationConnectionConfigBuilderTest {
+    
+    private final DatabaseType databaseType = 
TypedSPILoader.getService(DatabaseType.class, "MySQL");
+    
+    private final DialectSQLFederationConnectionConfigBuilder builder = 
DatabaseTypedSPILoader.getService(DialectSQLFederationConnectionConfigBuilder.class,
 databaseType);
+    
+    @Test
+    void assertBuild() {
+        CalciteConnectionConfig actualConfig = builder.build();
+        assertThat(actualConfig.lex(), is(Lex.MYSQL));
+        assertThat(actualConfig.conformance(), is(SqlConformanceEnum.MYSQL_5));
+        assertNotNull(actualConfig.fun(SqlOperatorTable.class, null));
+        assertThat(actualConfig.caseSensitive(), is(Lex.MYSQL.caseSensitive));
+        assertThat(actualConfig.defaultNullCollation(), is(NullCollation.LOW));
+        assertTrue(builder.isDefault());
+    }
+}
diff --git 
a/kernel/sql-federation/dialect/mysql/src/test/java/org/apache/shardingsphere/sqlfederation/mysql/MySQLSQLFederationFunctionRegisterTest.java
 
b/kernel/sql-federation/dialect/mysql/src/test/java/org/apache/shardingsphere/sqlfederation/mysql/MySQLSQLFederationFunctionRegisterTest.java
new file mode 100644
index 00000000000..abd8c33d219
--- /dev/null
+++ 
b/kernel/sql-federation/dialect/mysql/src/test/java/org/apache/shardingsphere/sqlfederation/mysql/MySQLSQLFederationFunctionRegisterTest.java
@@ -0,0 +1,59 @@
+/*
+ * 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.
+ */
+
+package org.apache.shardingsphere.sqlfederation.mysql;
+
+import org.apache.calcite.runtime.SqlFunctions;
+import org.apache.calcite.schema.Function;
+import org.apache.calcite.schema.SchemaPlus;
+import org.apache.calcite.schema.impl.ScalarFunctionImpl;
+import org.apache.calcite.tools.Frameworks;
+import 
org.apache.shardingsphere.database.connector.core.spi.DatabaseTypedSPILoader;
+import org.apache.shardingsphere.database.connector.core.type.DatabaseType;
+import org.apache.shardingsphere.infra.spi.type.typed.TypedSPILoader;
+import 
org.apache.shardingsphere.sqlfederation.compiler.sql.function.DialectSQLFederationFunctionRegister;
+import 
org.apache.shardingsphere.sqlfederation.compiler.sql.function.mysql.impl.MySQLBinFunction;
+import org.junit.jupiter.api.Test;
+
+import java.util.Collection;
+
+import static org.hamcrest.CoreMatchers.is;
+import static org.hamcrest.MatcherAssert.assertThat;
+import static org.junit.jupiter.api.Assertions.assertTrue;
+
+class MySQLSQLFederationFunctionRegisterTest {
+    
+    private final DatabaseType databaseType = 
TypedSPILoader.getService(DatabaseType.class, "MySQL");
+    
+    private final DialectSQLFederationFunctionRegister register = 
DatabaseTypedSPILoader.getService(DialectSQLFederationFunctionRegister.class, 
databaseType);
+    
+    @Test
+    void assertRegisterFunction() {
+        SchemaPlus schemaPlus = Frameworks.createRootSchema(true);
+        register.registerFunction(schemaPlus, "schema");
+        assertFunction(schemaPlus.getFunctions("bin"), MySQLBinFunction.class, 
"bin");
+        assertFunction(schemaPlus.getFunctions("atan"), SqlFunctions.class, 
"atan2");
+        assertFunction(schemaPlus.getFunctions("atan2"), SqlFunctions.class, 
"atan");
+    }
+    
+    private void assertFunction(final Collection<Function> functions, final 
Class<?> expectedClass, final String expectedMethod) {
+        assertThat(functions.size(), is(1));
+        ScalarFunctionImpl actualFunction = (ScalarFunctionImpl) 
functions.iterator().next();
+        
assertTrue(expectedClass.isAssignableFrom(actualFunction.method.getDeclaringClass()));
+        assertThat(actualFunction.method.getName(), is(expectedMethod));
+    }
+}

Reply via email to