Examples with Plugin io.trino.spi.Plugin used on opensource projects

Example 6 with Plugin

use of io.trino.spi.Plugin in project trino by trinodb.

the class TestKuduPlugin method testCreateConnector.

@Test
public void testCreateConnector() {
    Plugin plugin = new KuduPlugin();
    ConnectorFactory factory = getOnlyElement(plugin.getConnectorFactories());
    factory.create("test", ImmutableMap.of("kudu.client.master-addresses", "localhost:7051"), new TestingConnectorContext()).shutdown();
}

Example 7 with Plugin

use of io.trino.spi.Plugin in project trino by trinodb.

the class TestRaptorPlugin method testPlugin.

@Test
public void testPlugin() throws Exception {
    Plugin plugin = new RaptorPlugin();
    ConnectorFactory factory = getOnlyElement(plugin.getConnectorFactories());
    assertInstanceOf(factory, RaptorConnectorFactory.class);
    File tmpDir = Files.createTempDir();
    try {
        Map<String, String> config = ImmutableMap.<String, String>builder().put("metadata.db.type", "h2").put("metadata.db.filename", tmpDir.getAbsolutePath()).put("storage.data-directory", tmpDir.getAbsolutePath()).buildOrThrow();
        factory.create("test", config, new TestingConnectorContext()).shutdown();
    } finally {
        deleteRecursively(tmpDir.toPath(), ALLOW_INSECURE);
    }
}

Example 8 with Plugin

use of io.trino.spi.Plugin in project trino by trinodb.

the class TestSingleStorePlugin method testCreateConnector.

@Test
public void testCreateConnector() {
    Plugin plugin = new SingleStorePlugin();
    ConnectorFactory factory = stream(plugin.getConnectorFactories()).filter(connectorFactory -> connectorFactory.getName().equals("singlestore")).collect(toOptional()).orElseThrow();
    factory.create("test", ImmutableMap.of("connection-url", "jdbc:singlestore://test"), new TestingConnectorContext()).shutdown();
}

Example 9 with Plugin

use of io.trino.spi.Plugin in project trino by trinodb.

the class TestRedshiftPlugin method testCreateConnector.

@Test
public void testCreateConnector() {
    Plugin plugin = new RedshiftPlugin();
    ConnectorFactory factory = getOnlyElement(plugin.getConnectorFactories());
    factory.create("test", ImmutableMap.of("connection-url", "jdbc:redshift:test"), new TestingConnectorContext()).shutdown();
}

Example 10 with Plugin

use of io.trino.spi.Plugin in project trino by trinodb.

the class TestQueryTracker method createQueryRunner.

@Override
protected QueryRunner createQueryRunner() throws Exception {
    Session defaultSession = testSessionBuilder().setCatalog("mock").setSchema("default").setSystemProperty(QUERY_MAX_PLANNING_TIME, "2s").build();
    DistributedQueryRunner queryRunner = DistributedQueryRunner.builder(defaultSession).build();
    queryRunner.installPlugin(new Plugin() {

        @Override
        public Iterable<ConnectorFactory> getConnectorFactories() {
            return ImmutableList.of(MockConnectorFactory.builder().withGetColumns(ignored -> ImmutableList.of(new ColumnMetadata("col", VARCHAR))).withApplyFilter((ignored1, ignored2, ignored3) -> freeze()).build());
        }
    });
    queryRunner.createCatalog("mock", "mock");
    return queryRunner;
}