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 bfa45205935 Add PostgreSQLSQLFederationConnectionConfigBuilderTest and
PostgreSQLSQLFederationFunctionRegisterTest (#37318)
bfa45205935 is described below
commit bfa4520593549def7d480d19f7a997cc3571ed68
Author: Liang Zhang <[email protected]>
AuthorDate: Wed Dec 10 00:42:36 2025 +0800
Add PostgreSQLSQLFederationConnectionConfigBuilderTest and
PostgreSQLSQLFederationFunctionRegisterTest (#37318)
---
...QLSQLFederationConnectionConfigBuilderTest.java | 49 ++++++++++++++++
...ostgreSQLSQLFederationFunctionRegisterTest.java | 66 ++++++++++++++++++++++
2 files changed, 115 insertions(+)
diff --git
a/kernel/sql-federation/dialect/postgresql/src/test/java/org/apache/shardingsphere/sqlfederation/postgresql/PostgreSQLSQLFederationConnectionConfigBuilderTest.java
b/kernel/sql-federation/dialect/postgresql/src/test/java/org/apache/shardingsphere/sqlfederation/postgresql/PostgreSQLSQLFederationConnectionConfigBuilderTest.java
new file mode 100644
index 00000000000..c06ff2c1884
--- /dev/null
+++
b/kernel/sql-federation/dialect/postgresql/src/test/java/org/apache/shardingsphere/sqlfederation/postgresql/PostgreSQLSQLFederationConnectionConfigBuilderTest.java
@@ -0,0 +1,49 @@
+/*
+ * 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.postgresql;
+
+import org.apache.calcite.config.CalciteConnectionConfig;
+import org.apache.calcite.config.Lex;
+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.assertFalse;
+import static org.junit.jupiter.api.Assertions.assertNotNull;
+
+class PostgreSQLSQLFederationConnectionConfigBuilderTest {
+
+ private final DatabaseType databaseType =
TypedSPILoader.getService(DatabaseType.class, "PostgreSQL");
+
+ private final DialectSQLFederationConnectionConfigBuilder builder =
DatabaseTypedSPILoader.getService(DialectSQLFederationConnectionConfigBuilder.class,
databaseType);
+
+ @Test
+ void assertBuild() {
+ CalciteConnectionConfig actualConfig = builder.build();
+ assertThat(actualConfig.lex(), is(Lex.JAVA));
+ assertThat(actualConfig.conformance(), is(SqlConformanceEnum.BABEL));
+ assertNotNull(actualConfig.fun(SqlOperatorTable.class, null));
+ assertFalse(actualConfig.caseSensitive());
+ }
+}
diff --git
a/kernel/sql-federation/dialect/postgresql/src/test/java/org/apache/shardingsphere/sqlfederation/postgresql/PostgreSQLSQLFederationFunctionRegisterTest.java
b/kernel/sql-federation/dialect/postgresql/src/test/java/org/apache/shardingsphere/sqlfederation/postgresql/PostgreSQLSQLFederationFunctionRegisterTest.java
new file mode 100644
index 00000000000..b1f47150d3f
--- /dev/null
+++
b/kernel/sql-federation/dialect/postgresql/src/test/java/org/apache/shardingsphere/sqlfederation/postgresql/PostgreSQLSQLFederationFunctionRegisterTest.java
@@ -0,0 +1,66 @@
+/*
+ * 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.postgresql;
+
+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.postgresql.impl.PostgreSQLSystemFunction;
+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 PostgreSQLSQLFederationFunctionRegisterTest {
+
+ private final DatabaseType databaseType =
TypedSPILoader.getService(DatabaseType.class, "PostgreSQL");
+
+ private final DialectSQLFederationFunctionRegister register =
DatabaseTypedSPILoader.getService(DialectSQLFederationFunctionRegister.class,
databaseType);
+
+ @Test
+ void assertRegisterPgCatalogFunctions() {
+ SchemaPlus schemaPlus = Frameworks.createRootSchema(true);
+ register.registerFunction(schemaPlus, "pg_catalog");
+ assertThat(register.getDatabaseType(), is("PostgreSQL"));
+ assertFunction(schemaPlus.getFunctions("pg_table_is_visible"),
"pgTableIsVisible");
+ assertFunction(schemaPlus.getFunctions("pg_get_userbyid"),
"pgGetUserById");
+ }
+
+ @Test
+ void assertNoRegistrationWhenNotPgCatalog() {
+ SchemaPlus schemaPlus = Frameworks.createRootSchema(true);
+ register.registerFunction(schemaPlus, "public");
+ assertThat(schemaPlus.getFunctions("pg_table_is_visible").size(),
is(0));
+ assertThat(schemaPlus.getFunctions("pg_get_userbyid").size(), is(0));
+ }
+
+ private void assertFunction(final Collection<Function> functions, final
String expectedMethod) {
+ assertThat(functions.size(), is(1));
+ ScalarFunctionImpl actualFunction = (ScalarFunctionImpl)
functions.iterator().next();
+
assertTrue(PostgreSQLSystemFunction.class.isAssignableFrom(actualFunction.method.getDeclaringClass()));
+ assertThat(actualFunction.method.getName(), is(expectedMethod));
+ }
+}
