Added BaseConnectorSmokeTest as a base class for Pinot SmokeTest

plugin/trino-pinot/pom.xml

@@ -546,6 +546,18 @@
             </exclusions>
         </dependency>
 
+        <dependency>
+            <groupId>io.trino</groupId>
+            <artifactId>trino-tpch</artifactId>
+            <scope>test</scope>
+        </dependency>
+
+        <dependency>
+            <groupId>io.trino.tpch</groupId>
+            <artifactId>tpch</artifactId>
+            <scope>test</scope>
+        </dependency>
+
         <dependency>
             <groupId>io.airlift</groupId>
             <artifactId>testing</artifactId>

plugin/trino-pinot/src/test/java/io/trino/plugin/pinot/PinotQueryRunner.java

@@ -43,7 +43,6 @@ public static DistributedQueryRunner createPinotQueryRunner(Map<String, String>
             throws Exception
     {
         DistributedQueryRunner queryRunner = DistributedQueryRunner.builder(createSession("default"))
-                .setNodeCount(2)
                 .setExtraProperties(extraProperties)
                 .build();
         queryRunner.installPlugin(new PinotPlugin(extension));

plugin/trino-pinot/src/test/java/io/trino/plugin/pinot/TestPinotWithoutAuthenticationIntegrationSmokeTest.java → plugin/trino-pinot/src/test/java/io/trino/plugin/pinot/TestPinotWithoutAuthenticationIntegrationConnectorConnectorSmokeTest.java

@@ -13,8 +13,8 @@
  */
 package io.trino.plugin.pinot;
 
-public class TestPinotWithoutAuthenticationIntegrationSmokeTest
-        extends AbstractPinotIntegrationSmokeTest
+public class TestPinotWithoutAuthenticationIntegrationConnectorConnectorSmokeTest
+        extends BasePinotIntegrationConnectorSmokeTest
 {
     @Override
     protected boolean isSecured()

plugin/trino-pinot/src/test/java/io/trino/plugin/pinot/TestPinotWithoutAuthenticationIntegrationSmokeTestLatestVersion.java → plugin/trino-pinot/src/test/java/io/trino/plugin/pinot/TestPinotWithoutAuthenticationIntegrationLatestVersionConnectorConnectorSmokeTest.java

@@ -15,8 +15,8 @@
 
 import static io.trino.plugin.pinot.TestingPinotCluster.PINOT_LATEST_IMAGE_NAME;
 
-public class TestPinotWithoutAuthenticationIntegrationSmokeTestLatestVersion
-        extends AbstractPinotIntegrationSmokeTest
+public class TestPinotWithoutAuthenticationIntegrationLatestVersionConnectorConnectorSmokeTest
+        extends BasePinotIntegrationConnectorSmokeTest
 {
     @Override
     protected boolean isSecured()

plugin/trino-pinot/src/test/java/io/trino/plugin/pinot/TestPinotWithoutAuthenticationIntegrationSmokeTestLatestVersionNoGrpc.java → plugin/trino-pinot/src/test/java/io/trino/plugin/pinot/TestPinotWithoutAuthenticationIntegrationLatestVersionNoGrpcConnectorSmokeTest.java

@@ -15,8 +15,8 @@
 
 import static io.trino.plugin.pinot.TestingPinotCluster.PINOT_LATEST_IMAGE_NAME;
 
-public class TestPinotWithoutAuthenticationIntegrationSmokeTestLatestVersionNoGrpc
-        extends AbstractPinotIntegrationSmokeTest
+public class TestPinotWithoutAuthenticationIntegrationLatestVersionNoGrpcConnectorSmokeTest
+        extends BasePinotIntegrationConnectorSmokeTest
 {
     @Override
     protected boolean isSecured()

plugin/trino-pinot/src/test/java/io/trino/plugin/pinot/TestSecuredPinotIntegrationSmokeTest.java → plugin/trino-pinot/src/test/java/io/trino/plugin/pinot/TestSecuredPinotIntegrationConnectorSmokeTest.java

@@ -19,8 +19,8 @@
 
 import static io.trino.plugin.pinot.auth.PinotAuthenticationType.PASSWORD;
 
-public class TestSecuredPinotIntegrationSmokeTest
-        extends AbstractPinotIntegrationSmokeTest
+public class TestSecuredPinotIntegrationConnectorSmokeTest
+        extends BasePinotIntegrationConnectorSmokeTest
 {
     @Override
     protected boolean isSecured()

plugin/trino-pinot/src/test/resources/nation_realtimeSpec.json

@@ -0,0 +1,46 @@
+{
+    "tableName": "nation",
+    "tableType": "REALTIME",
+    "segmentsConfig": {
+        "timeColumnName": "updated_at_seconds",
+        "retentionTimeUnit": "DAYS",
+        "retentionTimeValue": "365",
+        "segmentPushType": "APPEND",
+        "segmentPushFrequency": "daily",
+        "segmentAssignmentStrategy": "BalanceNumSegmentAssignmentStrategy",
+        "schemaName": "nation",
+        "replicasPerPartition": "1"
+    },
+    "tenants": {
+        "broker": "DefaultTenant",
+        "server": "DefaultTenant"
+    },
+    "tableIndexConfig": {
+        "loadMode": "MMAP",
+        "noDictionaryColumns": [],
+        "sortedColumn": [
+            "updated_at_seconds"
+        ],
+        "aggregateMetrics": "false",
+        "nullHandlingEnabled": "true",
+        "streamConfigs": {
+            "streamType": "kafka",
+            "stream.kafka.consumer.type": "lowLevel",
+            "stream.kafka.topic.name": "nation",
+            "stream.kafka.decoder.class.name": "org.apache.pinot.plugin.inputformat.avro.confluent.KafkaConfluentSchemaRegistryAvroMessageDecoder",
+            "stream.kafka.consumer.factory.class.name": "org.apache.pinot.plugin.stream.kafka20.KafkaConsumerFactory",
+            "stream.kafka.decoder.prop.schema.registry.rest.url": "http://schema-registry:8081",
+            "stream.kafka.zk.broker.url": "zookeeper:2181/",
+            "stream.kafka.broker.list": "kafka:9092",
+            "realtime.segment.flush.threshold.time": "1m",
+            "realtime.segment.flush.threshold.size": "0",
+            "realtime.segment.flush.desired.size": "1M",
+            "isolation.level": "read_committed",
+            "stream.kafka.consumer.prop.auto.offset.reset": "smallest",
+            "stream.kafka.consumer.prop.group.id": "pinot_nation"
+        }
+    },
+    "metadata": {
+        "customConfigs": {}
+    }
+}

plugin/trino-pinot/src/test/resources/nation_schema.json

@@ -0,0 +1,31 @@
+{
+    "schemaName": "nation",
+    "dimensionFieldSpecs": [
+        {
+            "name": "nationkey",
+            "dataType": "LONG"
+        },
+        {
+            "name": "name",
+            "dataType": "STRING"
+        },
+        {
+            "name": "comment",
+            "dataType": "STRING"
+        },
+        {
+            "name": "regionkey",
+            "dataType": "LONG"
+        }
+    ],
+    "dateTimeFieldSpecs": [
+        {
+            "name": "updated_at_seconds",
+            "dataType": "LONG",
+            "defaultNullValue": 0,
+            "format": "1:SECONDS:EPOCH",
+            "transformFunction": "toEpochSeconds(updated_at)",
+            "granularity": "1:SECONDS"
+        }
+    ]
+}

plugin/trino-pinot/src/test/resources/region_realtimeSpec.json

@@ -0,0 +1,46 @@
+{
+    "tableName": "region",
+    "tableType": "REALTIME",
+    "segmentsConfig": {
+        "timeColumnName": "updated_at_seconds",
+        "retentionTimeUnit": "DAYS",
+        "retentionTimeValue": "365",
+        "segmentPushType": "APPEND",
+        "segmentPushFrequency": "daily",
+        "segmentAssignmentStrategy": "BalanceNumSegmentAssignmentStrategy",
+        "schemaName": "region",
+        "replicasPerPartition": "1"
+    },
+    "tenants": {
+        "broker": "DefaultTenant",
+        "server": "DefaultTenant"
+    },
+    "tableIndexConfig": {
+        "loadMode": "MMAP",
+        "noDictionaryColumns": [],
+        "sortedColumn": [
+            "updated_at_seconds"
+        ],
+        "aggregateMetrics": "false",
+        "nullHandlingEnabled": "true",
+        "streamConfigs": {
+            "streamType": "kafka",
+            "stream.kafka.consumer.type": "lowLevel",
+            "stream.kafka.topic.name": "region",
+            "stream.kafka.decoder.class.name": "org.apache.pinot.plugin.inputformat.avro.confluent.KafkaConfluentSchemaRegistryAvroMessageDecoder",
+            "stream.kafka.consumer.factory.class.name": "org.apache.pinot.plugin.stream.kafka20.KafkaConsumerFactory",
+            "stream.kafka.decoder.prop.schema.registry.rest.url": "http://schema-registry:8081",
+            "stream.kafka.zk.broker.url": "zookeeper:2181/",
+            "stream.kafka.broker.list": "kafka:9092",
+            "realtime.segment.flush.threshold.time": "1m",
+            "realtime.segment.flush.threshold.size": "0",
+            "realtime.segment.flush.desired.size": "1M",
+            "isolation.level": "read_committed",
+            "stream.kafka.consumer.prop.auto.offset.reset": "smallest",
+            "stream.kafka.consumer.prop.group.id": "pinot_region"
+        }
+    },
+    "metadata": {
+        "customConfigs": {}
+    }
+}

plugin/trino-pinot/src/test/resources/region_schema.json

@@ -0,0 +1,27 @@
+{
+    "schemaName": "region",
+    "dimensionFieldSpecs": [
+        {
+            "name": "regionkey",
+            "dataType": "LONG"
+        },
+        {
+            "name": "name",
+            "dataType": "STRING"
+        },
+        {
+            "name": "comment",
+            "dataType": "STRING"
+        }
+    ],
+    "dateTimeFieldSpecs": [
+        {
+            "name": "updated_at_seconds",
+            "dataType": "LONG",
+            "defaultNullValue": 0,
+            "format": "1:SECONDS:EPOCH",
+            "transformFunction": "toEpochSeconds(updated_at)",
+            "granularity": "1:SECONDS"
+        }
+    ]
+}