Examples with ConfigBuilder org.apache.drill.test.ConfigBuilder used on opensource projects

Example 11 with ConfigBuilder

use of org.apache.drill.test.ConfigBuilder in project drill by apache.

the class TestHttpProxy method testConfigForUrl.

@Test
public void testConfigForUrl() {
    Config config = new ConfigBuilder().put(ExecConstants.HTTP_PROXY_URL, "http://bob:secret@foo.com:1234").put(ExecConstants.HTTPS_PROXY_URL, "http://alice:s3cr3t@bar.com:2345").build();
    doTestConfigForUrl(config);
}

Example 12 with ConfigBuilder

use of org.apache.drill.test.ConfigBuilder in project drill by apache.

the class TestHttpProxy method testHttpUrlConfig.

@Test
public void testHttpUrlConfig() {
    Config config = new ConfigBuilder().put(ExecConstants.HTTP_PROXY_URL, "").put(ExecConstants.HTTP_PROXY_TYPE, "socks").put(ExecConstants.HTTP_PROXY_HOST, "foo.com").put(ExecConstants.HTTP_PROXY_PORT, 1234).put(ExecConstants.HTTP_PROXY_USER_NAME, "bob").put(ExecConstants.HTTP_PROXY_PASSWORD, "secret").build();
    HttpProxyConfig proxy = HttpProxyConfig.builder().fromHttpConfig(config).build();
    assertEquals(ProxyType.SOCKS, proxy.type);
    assertEquals("foo.com", proxy.host);
    assertEquals(1234, proxy.port);
    assertEquals("bob", proxy.username);
    assertEquals("secret", proxy.password);
}

Example 13 with ConfigBuilder

use of org.apache.drill.test.ConfigBuilder in project drill by apache.

the class TestSSLConfig method testInvalidHadoopKeystore.

@Test
public void testInvalidHadoopKeystore() throws Exception {
    Configuration hadoopConfig = new Configuration();
    String hadoopSSLFileProp = MessageFormat.format(HADOOP_SSL_CONF_TPL_KEY, SSLConfig.Mode.SERVER.toString().toLowerCase());
    hadoopConfig.set(hadoopSSLFileProp, "ssl-server-invalid.xml");
    ConfigBuilder config = new ConfigBuilder();
    config.put(ExecConstants.USER_SSL_ENABLED, true);
    config.put(ExecConstants.SSL_USE_HADOOP_CONF, true);
    SSLConfig sslv;
    try {
        sslv = new SSLConfigBuilder().config(config.build()).mode(SSLConfig.Mode.SERVER).initializeSSLContext(false).validateKeyStore(true).hadoopConfig(hadoopConfig).build();
        fail();
    } catch (Exception e) {
        assertTrue(e instanceof DrillException);
    }
}

Example 14 with ConfigBuilder

use of org.apache.drill.test.ConfigBuilder in project drill by apache.

the class TestSSLConfig method testForTrustStore.

@Test
public void testForTrustStore() throws Exception {
    ConfigBuilder config = new ConfigBuilder();
    config.put(ExecConstants.HTTP_TRUSTSTORE_PATH, "/root");
    config.put(ExecConstants.HTTP_TRUSTSTORE_PASSWORD, "root");
    config.put(ExecConstants.SSL_USE_HADOOP_CONF, false);
    SSLConfig sslv = new SSLConfigBuilder().config(config.build()).mode(SSLConfig.Mode.SERVER).initializeSSLContext(false).validateKeyStore(true).build();
    assertEquals(true, sslv.hasTrustStorePath());
    assertEquals(true, sslv.hasTrustStorePassword());
    assertEquals("/root", sslv.getTrustStorePath());
    assertEquals("root", sslv.getTrustStorePassword());
}

Example 15 with ConfigBuilder

use of org.apache.drill.test.ConfigBuilder in project drill by apache.

the class TestSSLConfig method testMissingKeystorePassword.

@Test
public void testMissingKeystorePassword() throws Exception {
    ConfigBuilder config = new ConfigBuilder();
    config.put(ExecConstants.HTTP_KEYSTORE_PATH, "/root");
    config.put(ExecConstants.HTTP_KEYSTORE_PASSWORD, "");
    config.put(ExecConstants.SSL_USE_HADOOP_CONF, false);
    config.put(ExecConstants.USER_SSL_ENABLED, true);
    try {
        SSLConfig sslv = new SSLConfigBuilder().config(config.build()).mode(SSLConfig.Mode.SERVER).initializeSSLContext(false).validateKeyStore(true).build();
        fail();
    // Expected
    } catch (Exception e) {
        assertTrue(e instanceof DrillException);
    }
}