Examples with ConfigHandler org.apache.metron.common.configuration.enrichment.handler.ConfigHandler used on opensource projects

Example 16 with ConfigHandler

use of org.apache.metron.common.configuration.enrichment.handler.ConfigHandler in project metron by apache.

the class StellarAdapterTest method test_default.

@Test
public void test_default() throws Exception {
    for (String c : DEFAULT_CONFIGS) {
        JSONObject message = getMessage();
        EnrichmentConfig enrichmentConfig = JSONUtils.INSTANCE.load(c, EnrichmentConfig.class);
        Assert.assertNotNull(enrichmentConfig.getEnrichmentConfigs().get("stellar"));
        ConfigHandler handler = enrichmentConfig.getEnrichmentConfigs().get("stellar");
        JSONObject enriched = enrich(message, "", handler);
        Assert.assertEquals("STELLAR_TEST", enriched.get("stmt1"));
        Assert.assertEquals("stellar_test", enriched.get("stmt2"));
        Assert.assertEquals("foo", enriched.get("stmt3"));
        Assert.assertEquals(3, enriched.size());
    }
}

Example 17 with ConfigHandler

use of org.apache.metron.common.configuration.enrichment.handler.ConfigHandler in project metron by apache.

the class StellarAdapterTest method test_grouped.

@Test
public void test_grouped() throws Exception {
    for (String c : GROUPED_CONFIGS) {
        JSONObject message = getMessage();
        EnrichmentConfig enrichmentConfig = JSONUtils.INSTANCE.load(c, EnrichmentConfig.class);
        Assert.assertNotNull(enrichmentConfig.getEnrichmentConfigs().get("stellar"));
        ConfigHandler handler = enrichmentConfig.getEnrichmentConfigs().get("stellar");
        {
            JSONObject enriched = enrich(message, "group1", handler);
            Assert.assertEquals("STELLAR_TEST", enriched.get("stmt1"));
            Assert.assertEquals("stellar_test", enriched.get("stmt2"));
            Assert.assertEquals(2, enriched.size());
        }
        {
            JSONObject enriched = enrich(message, "group2", handler);
            Assert.assertEquals("foo", enriched.get("stmt3"));
            Assert.assertEquals(1, enriched.size());
        }
    }
}