Examples with AuditInfo org.apache.druid.audit.AuditInfo used on opensource projects

Example 16 with AuditInfo

use of org.apache.druid.audit.AuditInfo in project druid by druid-io.

the class LookupCoordinatorManagerTest method testDeleteLookup.

@Test
public void testDeleteLookup() {
    final LookupExtractorFactoryMapContainer ignore = new LookupExtractorFactoryMapContainer("v0", ImmutableMap.of("lookup", "ignore"));
    final LookupExtractorFactoryMapContainer lookup = new LookupExtractorFactoryMapContainer("v0", ImmutableMap.of("lookup", "foo"));
    final LookupCoordinatorManager manager = new LookupCoordinatorManager(client, druidNodeDiscoveryProvider, mapper, configManager, lookupCoordinatorManagerConfig) {

        @Override
        public Map<String, Map<String, LookupExtractorFactoryMapContainer>> getKnownLookups() {
            return ImmutableMap.of(LOOKUP_TIER, ImmutableMap.of("foo", lookup, "ignore", ignore));
        }
    };
    manager.start();
    final AuditInfo auditInfo = new AuditInfo("author", "comment", "localhost");
    EasyMock.reset(configManager);
    EasyMock.expect(configManager.set(EasyMock.eq(LookupCoordinatorManager.LOOKUP_CONFIG_KEY), EasyMock.eq(ImmutableMap.of(LOOKUP_TIER, ImmutableMap.of("ignore", ignore))), EasyMock.eq(auditInfo))).andReturn(SetResult.ok()).once();
    EasyMock.replay(configManager);
    Assert.assertTrue(manager.deleteLookup(LOOKUP_TIER, "foo", auditInfo));
    EasyMock.verify(configManager);
}

Example 17 with AuditInfo

use of org.apache.druid.audit.AuditInfo in project druid by druid-io.

the class LookupCoordinatorManagerTest method testDeleteLastLookup.

@Test
public void testDeleteLastLookup() {
    final LookupExtractorFactoryMapContainer lookup = new LookupExtractorFactoryMapContainer("v0", ImmutableMap.of("lookup", "foo"));
    final LookupCoordinatorManager manager = new LookupCoordinatorManager(client, druidNodeDiscoveryProvider, mapper, configManager, lookupCoordinatorManagerConfig) {

        @Override
        public Map<String, Map<String, LookupExtractorFactoryMapContainer>> getKnownLookups() {
            return ImmutableMap.of(LOOKUP_TIER, ImmutableMap.of("foo", lookup));
        }
    };
    manager.start();
    final AuditInfo auditInfo = new AuditInfo("author", "comment", "localhost");
    EasyMock.reset(configManager);
    EasyMock.expect(configManager.set(EasyMock.eq(LookupCoordinatorManager.LOOKUP_CONFIG_KEY), EasyMock.eq(ImmutableMap.<String, Map<String, LookupExtractorFactoryMapContainer>>of()), EasyMock.eq(auditInfo))).andReturn(SetResult.ok()).once();
    EasyMock.replay(configManager);
    Assert.assertTrue(manager.deleteLookup(LOOKUP_TIER, "foo", auditInfo));
    EasyMock.verify(configManager);
}

Example 18 with AuditInfo

use of org.apache.druid.audit.AuditInfo in project druid by druid-io.

the class LookupCoordinatorManagerTest method testDeleteLookupIgnoresNotReady.

@Test
public void testDeleteLookupIgnoresNotReady() {
    final LookupCoordinatorManager manager = new LookupCoordinatorManager(client, druidNodeDiscoveryProvider, mapper, configManager, lookupCoordinatorManagerConfig) {

        @Override
        public Map<String, Map<String, LookupExtractorFactoryMapContainer>> getKnownLookups() {
            return null;
        }
    };
    manager.start();
    final AuditInfo auditInfo = new AuditInfo("author", "comment", "localhost");
    Assert.assertFalse(manager.deleteLookup(LOOKUP_TIER, "foo", auditInfo));
}

Example 19 with AuditInfo

use of org.apache.druid.audit.AuditInfo in project druid by druid-io.

the class LookupCoordinatorManagerTest method testUpdateLookupsOnlyUpdatesToTier.

@Test
public void testUpdateLookupsOnlyUpdatesToTier() {
    final LookupExtractorFactoryMapContainer ignore = new LookupExtractorFactoryMapContainer("v0", ImmutableMap.of("prop", "old"));
    final AuditInfo auditInfo = new AuditInfo("author", "comment", "localhost");
    final LookupCoordinatorManager manager = new LookupCoordinatorManager(client, druidNodeDiscoveryProvider, mapper, configManager, lookupCoordinatorManagerConfig) {

        @Override
        public Map<String, Map<String, LookupExtractorFactoryMapContainer>> getKnownLookups() {
            return ImmutableMap.of(LOOKUP_TIER + "1", ImmutableMap.of("foo", new LookupExtractorFactoryMapContainer("v0", ImmutableMap.of("prop", "new"))), LOOKUP_TIER + "2", ImmutableMap.of("ignore", ignore));
        }
    };
    manager.start();
    final LookupExtractorFactoryMapContainer newSpec = new LookupExtractorFactoryMapContainer("v1", ImmutableMap.of("prop", "new"));
    EasyMock.reset(configManager);
    EasyMock.expect(configManager.set(EasyMock.eq(LookupCoordinatorManager.LOOKUP_CONFIG_KEY), EasyMock.eq(ImmutableMap.<String, Map<String, LookupExtractorFactoryMapContainer>>of(LOOKUP_TIER + "1", ImmutableMap.of("foo", newSpec), LOOKUP_TIER + "2", ImmutableMap.of("ignore", ignore))), EasyMock.eq(auditInfo))).andReturn(SetResult.ok()).once();
    EasyMock.replay(configManager);
    Assert.assertTrue(manager.updateLookups(ImmutableMap.of(LOOKUP_TIER + "1", ImmutableMap.of("foo", newSpec)), auditInfo));
    EasyMock.verify(configManager);
}

Example 20 with AuditInfo

use of org.apache.druid.audit.AuditInfo in project druid by druid-io.

the class LookupCoordinatorManagerTest method testUpdateLookupAdds.

@Test
public void testUpdateLookupAdds() {
    final LookupCoordinatorManager manager = new LookupCoordinatorManager(client, druidNodeDiscoveryProvider, mapper, configManager, lookupCoordinatorManagerConfig) {

        @Override
        public Map<String, Map<String, LookupExtractorFactoryMapContainer>> getKnownLookups() {
            return EMPTY_TIERED_LOOKUP;
        }
    };
    manager.start();
    final AuditInfo auditInfo = new AuditInfo("author", "comment", "localhost");
    EasyMock.reset(configManager);
    EasyMock.expect(configManager.set(EasyMock.eq(LookupCoordinatorManager.LOOKUP_CONFIG_KEY), EasyMock.eq(TIERED_LOOKUP_MAP_V0), EasyMock.eq(auditInfo))).andReturn(SetResult.ok()).once();
    EasyMock.replay(configManager);
    manager.updateLookup(LOOKUP_TIER, SINGLE_LOOKUP_NAME, SINGLE_LOOKUP_SPEC_V0, auditInfo);
    EasyMock.verify(configManager);
}