Search in sources :

Example 1 with HdfsInputSourceConfig

use of org.apache.druid.inputsource.hdfs.HdfsInputSourceConfig in project druid by druid-io.

the class HdfsStorageDruidModuleTest method testHdfsInputSourceConfigDefaultAllowedProtocols.

@Test
public void testHdfsInputSourceConfigDefaultAllowedProtocols() {
    Properties props = new Properties();
    Injector injector = makeInjectorWithProperties(props);
    HdfsInputSourceConfig instance = injector.getInstance(HdfsInputSourceConfig.class);
    Assert.assertEquals(ImmutableSet.of("hdfs"), instance.getAllowedProtocols());
}
Also used : HdfsInputSourceConfig(org.apache.druid.inputsource.hdfs.HdfsInputSourceConfig) Injector(com.google.inject.Injector) Properties(java.util.Properties) Test(org.junit.Test)

Example 2 with HdfsInputSourceConfig

use of org.apache.druid.inputsource.hdfs.HdfsInputSourceConfig in project druid by druid-io.

the class HdfsFirehoseFactoryTest method testConstructorAllowsOnlyCustomProtocol.

@Test
public void testConstructorAllowsOnlyCustomProtocol() {
    final Configuration conf = new Configuration();
    conf.set("fs.ftp.impl", "org.apache.hadoop.fs.ftp.FTPFileSystem");
    new HdfsFirehoseFactory("ftp://localhost:21/foo/bar", null, null, null, null, null, DEFAULT_CONFIGURATION, new HdfsInputSourceConfig(ImmutableSet.of("ftp")));
    expectedException.expect(IllegalArgumentException.class);
    expectedException.expectMessage("Only [druid] protocols are allowed");
    new HdfsFirehoseFactory("hdfs://localhost:7020/foo/bar", null, null, null, null, null, DEFAULT_CONFIGURATION, new HdfsInputSourceConfig(ImmutableSet.of("druid")));
}
Also used : Configuration(org.apache.hadoop.conf.Configuration) HdfsInputSourceConfig(org.apache.druid.inputsource.hdfs.HdfsInputSourceConfig) Test(org.junit.Test)

Example 3 with HdfsInputSourceConfig

use of org.apache.druid.inputsource.hdfs.HdfsInputSourceConfig in project druid by druid-io.

the class HdfsStorageDruidModuleTest method testHdfsInputSourceConfigCustomAllowedProtocols.

@Test
public void testHdfsInputSourceConfigCustomAllowedProtocols() {
    Properties props = new Properties();
    props.setProperty("druid.ingestion.hdfs.allowedProtocols", "[\"webhdfs\"]");
    Injector injector = makeInjectorWithProperties(props);
    HdfsInputSourceConfig instance = injector.getInstance(HdfsInputSourceConfig.class);
    Assert.assertEquals(ImmutableSet.of("webhdfs"), instance.getAllowedProtocols());
}
Also used : HdfsInputSourceConfig(org.apache.druid.inputsource.hdfs.HdfsInputSourceConfig) Injector(com.google.inject.Injector) Properties(java.util.Properties) Test(org.junit.Test)

Aggregations

HdfsInputSourceConfig (org.apache.druid.inputsource.hdfs.HdfsInputSourceConfig)3 Test (org.junit.Test)3 Injector (com.google.inject.Injector)2 Properties (java.util.Properties)2 Configuration (org.apache.hadoop.conf.Configuration)1