public static EnrichmentConfigurations getSampleEnrichmentConfigs() throws IOException { EnrichmentConfigurations configurations = new EnrichmentConfigurations(); configurations.updateGlobalConfig(ConfigurationsUtils.readGlobalConfigFromFile(TestConstants.SAMPLE_CONFIG_PATH)); Map<String, byte[]> sensorEnrichmentConfigs = ConfigurationsUtils.readSensorEnrichmentConfigsFromFile(TestConstants.SAMPLE_CONFIG_PATH); for(String sensorType: sensorEnrichmentConfigs.keySet()) { configurations.updateSensorEnrichmentConfig(sensorType, sensorEnrichmentConfigs.get(sensorType)); } return configurations; }
@Test public void gets_batch_size_and_timeout_from_global_config() throws IOException { EnrichmentConfigurations configs = new EnrichmentConfigurations(); configs.updateGlobalConfig(globalJson.getBytes()); EnrichmentWriterConfiguration writerConfig = new EnrichmentWriterConfiguration(configs); assertThat("batch timeout should match global config setting", writerConfig.getBatchTimeout(null), equalTo(555)); assertThat("list should have single batch timeout matching global config setting", writerConfig.getAllConfiguredTimeouts(), equalTo(asList(555))); assertThat("batch size should match global config setting", writerConfig.getBatchSize(null), equalTo(12345)); }
File asnHdfsFile = new File(new File(baseDir), "GeoLite2-ASN.tar.gz"); globalConfig.put(GeoLiteAsnDatabase.ASN_HDFS_FILE, asnHdfsFile.getAbsolutePath()); threatIntelJoinBolt.getConfigurations().updateGlobalConfig(globalConfig); threatIntelJoinBolt.withMaxCacheSize(100); threatIntelJoinBolt.withMaxTimeRetain(10000);
File geoHdfsFile = new File(new File(baseDir), "GeoLite2-City.mmdb.gz"); globalConfig.put(GeoLiteCityDatabase.GEO_HDFS_FILE, geoHdfsFile.getAbsolutePath()); genericEnrichmentBolt.getConfigurations().updateGlobalConfig(globalConfig);
sampleConfigurations.updateGlobalConfig(ConfigurationsUtils.readGlobalConfigFromFile(TestConstants.SAMPLE_CONFIG_PATH)); Map<String, byte[]> sensorEnrichmentConfigs = ConfigurationsUtils.readSensorEnrichmentConfigsFromFile(TestConstants.ENRICHMENTS_CONFIGS_PATH); for (String sensorType : sensorEnrichmentConfigs.keySet()) {