test: Add E2E unit tests for CLP connector. (#54)

Author: anlowee

pom.xml

@@ -2420,6 +2420,19 @@
                 <version>2.1.0-3</version>
             </dependency>
 
+            <dependency>
+                <groupId>com.facebook.presto</groupId>
+                <artifactId>presto-clp</artifactId>
+                <version>${project.version}</version>
+            </dependency>
+
+            <dependency>
+                <groupId>com.facebook.presto</groupId>
+                <artifactId>presto-clp</artifactId>
+                <version>${project.version}</version>
+                <type>test-jar</type>
+            </dependency>
+
             <!-- force newer version to be used for dependencies -->
             <dependency>
                 <groupId>org.javassist</groupId>

presto-clp/pom.xml

@@ -129,6 +129,12 @@
             <scope>test</scope>
         </dependency>
 
+        <dependency>
+            <groupId>com.facebook.presto</groupId>
+            <artifactId>presto-tests</artifactId>
+            <scope>test</scope>
+        </dependency>
+
         <dependency>
             <groupId>org.apache.commons</groupId>
             <artifactId>commons-math3</artifactId>

presto-clp/src/test/java/com/facebook/presto/plugin/clp/ClpQueryRunner.java

@@ -0,0 +1,101 @@
+/*
+ * 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 com.facebook.presto.plugin.clp;
+
+import com.facebook.airlift.log.Logger;
+import com.facebook.presto.Session;
+import com.facebook.presto.tests.DistributedQueryRunner;
+import com.google.common.collect.ImmutableMap;
+
+import java.net.URI;
+import java.util.Map;
+import java.util.Optional;
+import java.util.function.BiFunction;
+
+import static com.facebook.presto.testing.TestingSession.testSessionBuilder;
+
+public class ClpQueryRunner
+{
+    private static final Logger log = Logger.get(ClpQueryRunner.class);
+
+    public static final String CLP_CATALOG = "clp";
+    public static final String CLP_CONNECTOR = CLP_CATALOG;
+    public static final int DEFAULT_NUM_OF_WORKERS = 1;
+    public static final String DEFAULT_SCHEMA = "default";
+
+    public static DistributedQueryRunner createQueryRunner(
+            String metadataDbUrl,
+            String metadataDbUser,
+            String metadataDbPassword,
+            String metadataDbTablePrefix,
+            Optional<Integer> workerCount,
+            Optional<BiFunction<Integer, URI, Process>> externalWorkerLauncher)
+                throws Exception
+    {
+        log.info("Creating CLP query runner with default session");
+        return createQueryRunner(
+                createDefaultSession(),
+                metadataDbUrl,
+                metadataDbUser,
+                metadataDbPassword,
+                metadataDbTablePrefix,
+                workerCount,
+                externalWorkerLauncher);
+    }
+
+    public static DistributedQueryRunner createQueryRunner(
+            Session session,
+            String metadataDbUrl,
+            String metadataDbUser,
+            String metadataDbPassword,
+            String metadataDbTablePrefix,
+            Optional<Integer> workerCount,
+            Optional<BiFunction<Integer, URI, Process>> externalWorkerLauncher)
+                throws Exception
+    {
+        DistributedQueryRunner clpQueryRunner = DistributedQueryRunner.builder(session)
+                        .setNodeCount(workerCount.orElse(DEFAULT_NUM_OF_WORKERS))
+                        .setExternalWorkerLauncher(externalWorkerLauncher)
+                        .build();
+        Map<String, String> clpProperties = ImmutableMap.<String, String>builder()
+                .put("clp.metadata-provider-type", "mysql")
+                .put("clp.metadata-db-url", metadataDbUrl)
+                .put("clp.metadata-db-user", metadataDbUser)
+                .put("clp.metadata-db-password", metadataDbPassword)
+                .put("clp.metadata-table-prefix", metadataDbTablePrefix)
+                .put("clp.split-provider-type", "mysql")
+                .build();
+
+        clpQueryRunner.installPlugin(new ClpPlugin());
+        clpQueryRunner.createCatalog(CLP_CATALOG, CLP_CONNECTOR, clpProperties);
+        return clpQueryRunner;
+    }
+
+    /**
+     * Creates a default mock session for query use.
+     *
+     * @return a default session
+     */
+    private static Session createDefaultSession()
+    {
+        return testSessionBuilder()
+                .setCatalog(CLP_CATALOG)
+                .setSchema(DEFAULT_SCHEMA)
+                .build();
+    }
+
+    private ClpQueryRunner()
+    {
+    }
+}