diff --git a/plugin/trino-base-jdbc/src/test/java/io/trino/plugin/jdbc/BaseJdbcConnectorTest.java b/plugin/trino-base-jdbc/src/test/java/io/trino/plugin/jdbc/BaseJdbcConnectorTest.java new file mode 100644 index 000000000000..c58de234babc --- /dev/null +++ b/plugin/trino-base-jdbc/src/test/java/io/trino/plugin/jdbc/BaseJdbcConnectorTest.java @@ -0,0 +1,22 @@ +/* + * Licensed 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 io.trino.plugin.jdbc; + +import io.trino.testing.BaseConnectorTest; + +public abstract class BaseJdbcConnectorTest + extends BaseConnectorTest +{ + // TODO move common tests from connector-specific classes here +} diff --git a/plugin/trino-base-jdbc/src/test/java/io/trino/plugin/jdbc/TestJdbcConnectorTest.java b/plugin/trino-base-jdbc/src/test/java/io/trino/plugin/jdbc/TestJdbcConnectorTest.java index 3ebf3da1727f..425b7ebf4266 100644 --- a/plugin/trino-base-jdbc/src/test/java/io/trino/plugin/jdbc/TestJdbcConnectorTest.java +++ b/plugin/trino-base-jdbc/src/test/java/io/trino/plugin/jdbc/TestJdbcConnectorTest.java @@ -14,7 +14,6 @@ package io.trino.plugin.jdbc; import com.google.common.collect.ImmutableMap; -import io.trino.testing.BaseConnectorTest; import io.trino.testing.QueryRunner; import io.trino.testing.sql.JdbcSqlExecutor; import io.trino.testing.sql.TestTable; @@ -27,7 +26,7 @@ import static io.trino.plugin.jdbc.H2QueryRunner.createH2QueryRunner; public class TestJdbcConnectorTest - extends BaseConnectorTest + extends BaseJdbcConnectorTest { private Map properties; diff --git a/plugin/trino-clickhouse/pom.xml b/plugin/trino-clickhouse/pom.xml index 5fce19188ebc..d99d6c902c0a 100644 --- a/plugin/trino-clickhouse/pom.xml +++ b/plugin/trino-clickhouse/pom.xml @@ -100,6 +100,13 @@ + + io.trino + trino-base-jdbc + test-jar + test + + io.trino trino-main diff --git a/plugin/trino-clickhouse/src/test/java/io/trino/plugin/clickhouse/TestClickHouseConnectorTest.java b/plugin/trino-clickhouse/src/test/java/io/trino/plugin/clickhouse/TestClickHouseConnectorTest.java index 1a0328f41267..8586cbde11b9 100644 --- a/plugin/trino-clickhouse/src/test/java/io/trino/plugin/clickhouse/TestClickHouseConnectorTest.java +++ b/plugin/trino-clickhouse/src/test/java/io/trino/plugin/clickhouse/TestClickHouseConnectorTest.java @@ -14,11 +14,10 @@ package io.trino.plugin.clickhouse; import com.google.common.collect.ImmutableMap; -import io.trino.testing.BaseConnectorTest; +import io.trino.plugin.jdbc.BaseJdbcConnectorTest; import io.trino.testing.MaterializedResult; import io.trino.testing.QueryRunner; import io.trino.testing.sql.TestTable; -import io.trino.tpch.TpchTable; import org.testng.SkipException; import org.testng.annotations.Test; @@ -31,7 +30,7 @@ import static org.assertj.core.api.Assertions.assertThat; public class TestClickHouseConnectorTest - extends BaseConnectorTest + extends BaseJdbcConnectorTest { private TestingClickHouseServer clickhouseServer; @@ -40,10 +39,14 @@ protected QueryRunner createQueryRunner() throws Exception { this.clickhouseServer = closeAfterClass(new TestingClickHouseServer()); - return createClickHouseQueryRunner(clickhouseServer, ImmutableMap.of(), ImmutableMap.builder() - .put("clickhouse.map-string-as-varchar", "true") - .put("allow-drop-table", "true") - .build(), TpchTable.getTables()); + return createClickHouseQueryRunner( + clickhouseServer, + ImmutableMap.of(), + ImmutableMap.builder() + .put("clickhouse.map-string-as-varchar", "true") + .put("allow-drop-table", "true") + .build(), + REQUIRED_TPCH_TABLES); } @Override diff --git a/plugin/trino-memsql/pom.xml b/plugin/trino-memsql/pom.xml index 96b7c2d49912..76ea5f7f6d11 100644 --- a/plugin/trino-memsql/pom.xml +++ b/plugin/trino-memsql/pom.xml @@ -90,6 +90,13 @@ + + io.trino + trino-base-jdbc + test-jar + test + + io.trino trino-main diff --git a/plugin/trino-memsql/src/test/java/io/trino/plugin/memsql/TestMemSqlConnectorTest.java b/plugin/trino-memsql/src/test/java/io/trino/plugin/memsql/TestMemSqlConnectorTest.java index f3224088f59a..1d3a7c8063e0 100644 --- a/plugin/trino-memsql/src/test/java/io/trino/plugin/memsql/TestMemSqlConnectorTest.java +++ b/plugin/trino-memsql/src/test/java/io/trino/plugin/memsql/TestMemSqlConnectorTest.java @@ -14,14 +14,13 @@ package io.trino.plugin.memsql; import com.google.common.collect.ImmutableMap; +import io.trino.plugin.jdbc.BaseJdbcConnectorTest; import io.trino.sql.planner.plan.AggregationNode; import io.trino.sql.planner.plan.FilterNode; -import io.trino.testing.BaseConnectorTest; import io.trino.testing.MaterializedResult; import io.trino.testing.MaterializedRow; import io.trino.testing.QueryRunner; import io.trino.testing.sql.TestTable; -import io.trino.tpch.TpchTable; import org.intellij.lang.annotations.Language; import org.testng.SkipException; import org.testng.annotations.AfterClass; @@ -43,7 +42,7 @@ import static org.testng.Assert.assertTrue; public class TestMemSqlConnectorTest - extends BaseConnectorTest + extends BaseJdbcConnectorTest { protected TestingMemSqlServer memSqlServer; @@ -52,7 +51,7 @@ protected QueryRunner createQueryRunner() throws Exception { memSqlServer = new TestingMemSqlServer(); - return createMemSqlQueryRunner(memSqlServer, ImmutableMap.of(), TpchTable.getTables()); + return createMemSqlQueryRunner(memSqlServer, ImmutableMap.of(), REQUIRED_TPCH_TABLES); } @AfterClass(alwaysRun = true) diff --git a/plugin/trino-mysql/pom.xml b/plugin/trino-mysql/pom.xml index 4f1b47394777..b6067c513b65 100644 --- a/plugin/trino-mysql/pom.xml +++ b/plugin/trino-mysql/pom.xml @@ -104,6 +104,13 @@ + + io.trino + trino-base-jdbc + test-jar + test + + io.trino trino-main diff --git a/plugin/trino-mysql/src/test/java/io/trino/plugin/mysql/BaseMySqlConnectorTest.java b/plugin/trino-mysql/src/test/java/io/trino/plugin/mysql/BaseMySqlConnectorTest.java index 1cd5373f28d7..fb48372f0e64 100644 --- a/plugin/trino-mysql/src/test/java/io/trino/plugin/mysql/BaseMySqlConnectorTest.java +++ b/plugin/trino-mysql/src/test/java/io/trino/plugin/mysql/BaseMySqlConnectorTest.java @@ -15,12 +15,12 @@ import com.google.common.collect.ImmutableList; import io.trino.Session; +import io.trino.plugin.jdbc.BaseJdbcConnectorTest; import io.trino.sql.planner.plan.AggregationNode; import io.trino.sql.planner.plan.ExchangeNode; import io.trino.sql.planner.plan.FilterNode; import io.trino.sql.planner.plan.MarkDistinctNode; import io.trino.sql.planner.plan.ProjectNode; -import io.trino.testing.BaseConnectorTest; import io.trino.testing.MaterializedResult; import io.trino.testing.MaterializedRow; import io.trino.testing.sql.SqlExecutor; @@ -43,7 +43,7 @@ import static org.testng.Assert.assertTrue; public abstract class BaseMySqlConnectorTest - extends BaseConnectorTest + extends BaseJdbcConnectorTest { @Override protected boolean supportsDelete() diff --git a/plugin/trino-oracle/pom.xml b/plugin/trino-oracle/pom.xml index 6ef6f6d13d74..3f45930d6a4c 100644 --- a/plugin/trino-oracle/pom.xml +++ b/plugin/trino-oracle/pom.xml @@ -109,6 +109,13 @@ + + io.trino + trino-base-jdbc + test-jar + test + + io.trino trino-main diff --git a/plugin/trino-oracle/src/test/java/io/trino/plugin/oracle/BaseOracleConnectorTest.java b/plugin/trino-oracle/src/test/java/io/trino/plugin/oracle/BaseOracleConnectorTest.java index 6f58ed88c135..1e83bbff5c7a 100644 --- a/plugin/trino-oracle/src/test/java/io/trino/plugin/oracle/BaseOracleConnectorTest.java +++ b/plugin/trino-oracle/src/test/java/io/trino/plugin/oracle/BaseOracleConnectorTest.java @@ -16,10 +16,10 @@ import com.google.common.collect.ImmutableList; import io.trino.Session; import io.trino.execution.QueryInfo; +import io.trino.plugin.jdbc.BaseJdbcConnectorTest; import io.trino.sql.planner.plan.AggregationNode; import io.trino.sql.planner.plan.FilterNode; import io.trino.sql.planner.plan.ProjectNode; -import io.trino.testing.BaseConnectorTest; import io.trino.testing.MaterializedResult; import io.trino.testing.ResultWithQueryId; import io.trino.testing.sql.SqlExecutor; @@ -39,7 +39,7 @@ import static org.testng.Assert.assertTrue; public abstract class BaseOracleConnectorTest - extends BaseConnectorTest + extends BaseJdbcConnectorTest { @Override protected boolean supportsDelete() diff --git a/plugin/trino-oracle/src/test/java/io/trino/plugin/oracle/TestOracleConnectorTest.java b/plugin/trino-oracle/src/test/java/io/trino/plugin/oracle/TestOracleConnectorTest.java index 72c38ebe5c1c..366f27490b90 100644 --- a/plugin/trino-oracle/src/test/java/io/trino/plugin/oracle/TestOracleConnectorTest.java +++ b/plugin/trino-oracle/src/test/java/io/trino/plugin/oracle/TestOracleConnectorTest.java @@ -17,7 +17,6 @@ import io.airlift.testing.Closeables; import io.trino.testing.QueryRunner; import io.trino.testing.sql.SqlExecutor; -import io.trino.tpch.TpchTable; import org.testng.annotations.AfterClass; import org.testng.annotations.Test; @@ -53,7 +52,7 @@ protected QueryRunner createQueryRunner() .put("oracle.connection-pool.enabled", "false") .put("oracle.remarks-reporting.enabled", "false") .build(), - TpchTable.getTables()); + REQUIRED_TPCH_TABLES); } @AfterClass(alwaysRun = true) diff --git a/plugin/trino-oracle/src/test/java/io/trino/plugin/oracle/TestOraclePoolConnectorTest.java b/plugin/trino-oracle/src/test/java/io/trino/plugin/oracle/TestOraclePoolConnectorTest.java index 48fb5452898d..dbc6cd11497e 100644 --- a/plugin/trino-oracle/src/test/java/io/trino/plugin/oracle/TestOraclePoolConnectorTest.java +++ b/plugin/trino-oracle/src/test/java/io/trino/plugin/oracle/TestOraclePoolConnectorTest.java @@ -17,7 +17,6 @@ import io.airlift.testing.Closeables; import io.trino.testing.QueryRunner; import io.trino.testing.sql.SqlExecutor; -import io.trino.tpch.TpchTable; import org.testng.annotations.AfterClass; import org.testng.annotations.Test; @@ -49,7 +48,7 @@ protected QueryRunner createQueryRunner() .put("oracle.connection-pool.enabled", "true") .put("oracle.remarks-reporting.enabled", "false") .build(), - TpchTable.getTables()); + REQUIRED_TPCH_TABLES); } @AfterClass(alwaysRun = true) diff --git a/plugin/trino-oracle/src/test/java/io/trino/plugin/oracle/TestRemarksReportingOraclePoolConnectorTest.java b/plugin/trino-oracle/src/test/java/io/trino/plugin/oracle/TestRemarksReportingOraclePoolConnectorTest.java index 24a21bbe6fde..b69513491ca1 100644 --- a/plugin/trino-oracle/src/test/java/io/trino/plugin/oracle/TestRemarksReportingOraclePoolConnectorTest.java +++ b/plugin/trino-oracle/src/test/java/io/trino/plugin/oracle/TestRemarksReportingOraclePoolConnectorTest.java @@ -17,7 +17,6 @@ import io.airlift.testing.Closeables; import io.trino.testing.QueryRunner; import io.trino.testing.sql.SqlExecutor; -import io.trino.tpch.TpchTable; import org.testng.annotations.AfterClass; import java.io.IOException; @@ -46,7 +45,7 @@ protected QueryRunner createQueryRunner() .put("oracle.connection-pool.enabled", "true") .put("oracle.remarks-reporting.enabled", "true") .build(), - TpchTable.getTables()); + REQUIRED_TPCH_TABLES); } @AfterClass(alwaysRun = true) diff --git a/plugin/trino-postgresql/pom.xml b/plugin/trino-postgresql/pom.xml index d4803d0fa5ba..01ded25b7f80 100644 --- a/plugin/trino-postgresql/pom.xml +++ b/plugin/trino-postgresql/pom.xml @@ -112,6 +112,13 @@ + + io.trino + trino-base-jdbc + test-jar + test + + io.trino trino-jmx diff --git a/plugin/trino-postgresql/src/test/java/io/trino/plugin/postgresql/TestPostgreSqlConnectorTest.java b/plugin/trino-postgresql/src/test/java/io/trino/plugin/postgresql/TestPostgreSqlConnectorTest.java index 53f9cc7c400a..cc9ca6c27855 100644 --- a/plugin/trino-postgresql/src/test/java/io/trino/plugin/postgresql/TestPostgreSqlConnectorTest.java +++ b/plugin/trino-postgresql/src/test/java/io/trino/plugin/postgresql/TestPostgreSqlConnectorTest.java @@ -15,12 +15,12 @@ import com.google.common.collect.ImmutableList; import io.trino.Session; +import io.trino.plugin.jdbc.BaseJdbcConnectorTest; import io.trino.sql.planner.plan.AggregationNode; import io.trino.sql.planner.plan.ExchangeNode; import io.trino.sql.planner.plan.FilterNode; import io.trino.sql.planner.plan.MarkDistinctNode; import io.trino.sql.planner.plan.ProjectNode; -import io.trino.testing.BaseConnectorTest; import io.trino.testing.QueryRunner; import io.trino.testing.sql.JdbcSqlExecutor; import io.trino.testing.sql.TestTable; @@ -48,7 +48,7 @@ import static org.testng.Assert.assertTrue; public class TestPostgreSqlConnectorTest - extends BaseConnectorTest + extends BaseJdbcConnectorTest { protected TestingPostgreSqlServer postgreSqlServer; diff --git a/plugin/trino-sqlserver/pom.xml b/plugin/trino-sqlserver/pom.xml index b902e5f37979..f89fe75bd02e 100644 --- a/plugin/trino-sqlserver/pom.xml +++ b/plugin/trino-sqlserver/pom.xml @@ -95,6 +95,13 @@ + + io.trino + trino-base-jdbc + test-jar + test + + io.trino trino-main diff --git a/plugin/trino-sqlserver/src/test/java/io/trino/plugin/sqlserver/TestSqlServerConnectorTest.java b/plugin/trino-sqlserver/src/test/java/io/trino/plugin/sqlserver/TestSqlServerConnectorTest.java index 9e2c3e151c1e..45ea665c22dd 100644 --- a/plugin/trino-sqlserver/src/test/java/io/trino/plugin/sqlserver/TestSqlServerConnectorTest.java +++ b/plugin/trino-sqlserver/src/test/java/io/trino/plugin/sqlserver/TestSqlServerConnectorTest.java @@ -15,15 +15,14 @@ import com.google.common.collect.ImmutableMap; import io.trino.Session; +import io.trino.plugin.jdbc.BaseJdbcConnectorTest; import io.trino.sql.planner.plan.AggregationNode; import io.trino.sql.planner.plan.ExchangeNode; import io.trino.sql.planner.plan.FilterNode; import io.trino.sql.planner.plan.MarkDistinctNode; import io.trino.sql.planner.plan.ProjectNode; -import io.trino.testing.BaseConnectorTest; import io.trino.testing.QueryRunner; import io.trino.testing.sql.TestTable; -import io.trino.tpch.TpchTable; import org.testng.annotations.DataProvider; import org.testng.annotations.Test; @@ -43,7 +42,7 @@ import static org.testng.Assert.assertTrue; public class TestSqlServerConnectorTest - extends BaseConnectorTest + extends BaseJdbcConnectorTest { protected TestingSqlServer sqlServer; @@ -53,7 +52,7 @@ protected QueryRunner createQueryRunner() { sqlServer = closeAfterClass(new TestingSqlServer()); sqlServer.start(); - return createSqlServerQueryRunner(sqlServer, ImmutableMap.of(), ImmutableMap.of(), TpchTable.getTables()); + return createSqlServerQueryRunner(sqlServer, ImmutableMap.of(), ImmutableMap.of(), REQUIRED_TPCH_TABLES); } @Override