Search in sources :

Example 1 with AlarmCallbackConfigurationService

use of org.graylog2.alarmcallbacks.AlarmCallbackConfigurationService in project graylog2-server by Graylog2.

the class V20161125142400_EmailAlarmCallbackMigrationTest method setUp.

@Before
public void setUp() throws Exception {
    final User localAdmin = mock(User.class);
    when(localAdmin.getId()).thenReturn(localAdminId);
    when(userService.getAdminUser()).thenReturn(localAdmin);
    this.emailAlarmCallbackMigrationPeriodical = new V20161125142400_EmailAlarmCallbackMigration(clusterConfigService, streamService, alarmCallbackConfigurationService, emailAlarmCallback, userService);
}
Also used : User(org.graylog2.plugin.database.users.User) Before(org.junit.Before)

Example 2 with AlarmCallbackConfigurationService

use of org.graylog2.alarmcallbacks.AlarmCallbackConfigurationService in project graylog2-server by Graylog2.

the class StreamCatalogTest method setUp.

@Before
@SuppressForbidden("Using Executors.newSingleThreadExecutor() is okay in tests")
public void setUp() throws Exception {
    final MongoConnection mongoConnection = mongodb.mongoConnection();
    final ClusterEventBus clusterEventBus = new ClusterEventBus("cluster-event-bus", Executors.newSingleThreadExecutor());
    final StreamRuleService streamRuleService = new StreamRuleServiceImpl(mongoConnection, clusterEventBus);
    final StreamService streamService = new StreamServiceImpl(mongoConnection, streamRuleService, alertService, outputService, indexSetService, mongoIndexSetFactory, notificationService, entityOwnershipService, clusterEventBus, alarmCallbackConfigurationService);
    when(outputService.load("5adf239e4b900a0fdb4e5197")).thenReturn(OutputImpl.create("5adf239e4b900a0fdb4e5197", "Title", "Type", "admin", Collections.emptyMap(), new Date(1524654085L), null));
    facade = new StreamFacade(objectMapper, streamService, streamRuleService, alertService, alarmCallbackConfigurationService, legacyAlertConditionMigration, indexSetService, userService);
}
Also used : StreamRuleServiceImpl(org.graylog2.streams.StreamRuleServiceImpl) StreamService(org.graylog2.streams.StreamService) StreamRuleService(org.graylog2.streams.StreamRuleService) StreamServiceImpl(org.graylog2.streams.StreamServiceImpl) MongoConnection(org.graylog2.database.MongoConnection) ClusterEventBus(org.graylog2.events.ClusterEventBus) Date(java.util.Date) Before(org.junit.Before) SuppressForbidden(org.graylog2.shared.SuppressForbidden)

Example 3 with AlarmCallbackConfigurationService

use of org.graylog2.alarmcallbacks.AlarmCallbackConfigurationService in project graylog2-server by Graylog2.

the class ContentPackServiceTest method setUp.

@Before
public void setUp() throws Exception {
    final ContentPackInstallationPersistenceService contentPackInstallationPersistenceService = contentPackInstallService;
    final Set<ConstraintChecker> constraintCheckers = Collections.emptySet();
    pluginMetaData = new HashSet<>();
    outputFactories = new HashMap<>();
    outputFactories2 = new HashMap<>();
    final Map<ModelType, EntityWithExcerptFacade<?, ?>> entityFacades = ImmutableMap.of(ModelTypes.GROK_PATTERN_V1, new GrokPatternFacade(objectMapper, patternService), ModelTypes.STREAM_V1, new StreamFacade(objectMapper, streamService, streamRuleService, alertService, alarmCallbackConfigurationService, legacyAlertConditionMigration, indexSetService, userService), ModelTypes.OUTPUT_V1, new OutputFacade(objectMapper, outputService, pluginMetaData, outputFactories, outputFactories2));
    contentPackService = new ContentPackService(contentPackInstallationPersistenceService, constraintCheckers, entityFacades);
    Map<String, String> entityData = new HashMap<>(2);
    entityData.put("name", "NAME");
    entityData.put("pattern", "\\w");
    grokPattern = GrokPattern.builder().pattern("\\w").name("NAME").build();
    JsonNode jsonData = objectMapper.convertValue(entityData, JsonNode.class);
    EntityV1 entityV1 = EntityV1.builder().id(ModelId.of("12345")).type(ModelTypes.GROK_PATTERN_V1).data(jsonData).build();
    ImmutableSet<Entity> entities = ImmutableSet.of(entityV1);
    NativeEntityDescriptor nativeEntityDescriptor = NativeEntityDescriptor.create(ModelId.of("12345"), "dead-beef1", ModelTypes.GROK_PATTERN_V1, "NAME");
    nativeEntityDescriptors = ImmutableSet.of(nativeEntityDescriptor);
    contentPack = ContentPackV1.builder().description("test").entities(entities).name("test").revision(1).summary("").vendor("").url(URI.create("http://graylog.com")).id(ModelId.of("dead-beef")).build();
    contentPackInstallation = ContentPackInstallation.builder().contentPackId(ModelId.of("dead-beef")).contentPackRevision(1).entities(nativeEntityDescriptors).comment("Installed").parameters(ImmutableMap.copyOf(Collections.emptyMap())).createdAt(Instant.now()).createdBy("me").build();
}
Also used : Entity(org.graylog2.contentpacks.model.entities.Entity) StreamFacade(org.graylog2.contentpacks.facades.StreamFacade) HashMap(java.util.HashMap) JsonNode(com.fasterxml.jackson.databind.JsonNode) EntityV1(org.graylog2.contentpacks.model.entities.EntityV1) EntityWithExcerptFacade(org.graylog2.contentpacks.facades.EntityWithExcerptFacade) NativeEntityDescriptor(org.graylog2.contentpacks.model.entities.NativeEntityDescriptor) ModelType(org.graylog2.contentpacks.model.ModelType) ConstraintChecker(org.graylog2.contentpacks.constraints.ConstraintChecker) GrokPatternFacade(org.graylog2.contentpacks.facades.GrokPatternFacade) OutputFacade(org.graylog2.contentpacks.facades.OutputFacade) Before(org.junit.Before)

Example 4 with AlarmCallbackConfigurationService

use of org.graylog2.alarmcallbacks.AlarmCallbackConfigurationService in project graylog2-server by Graylog2.

the class V20161125161400_AlertReceiversMigrationTest method setUp.

@Before
@SuppressWarnings("deprecation")
public void setUp() throws Exception {
    final MongoConnection mongoConnection = mock(MongoConnection.class);
    final DB database = mock(DB.class);
    when(mongoConnection.getDatabase()).thenReturn(database);
    when(database.getCollection(eq("streams"))).thenReturn(dbCollection);
    this.alertReceiversMigration = new V20161125161400_AlertReceiversMigration(clusterConfigService, streamService, alarmCallbackConfigurationService, mongoConnection);
}
Also used : MongoConnection(org.graylog2.database.MongoConnection) DB(com.mongodb.DB) Before(org.junit.Before)

Aggregations

Before (org.junit.Before)4 MongoConnection (org.graylog2.database.MongoConnection)2 JsonNode (com.fasterxml.jackson.databind.JsonNode)1 DB (com.mongodb.DB)1 Date (java.util.Date)1 HashMap (java.util.HashMap)1 ConstraintChecker (org.graylog2.contentpacks.constraints.ConstraintChecker)1 EntityWithExcerptFacade (org.graylog2.contentpacks.facades.EntityWithExcerptFacade)1 GrokPatternFacade (org.graylog2.contentpacks.facades.GrokPatternFacade)1 OutputFacade (org.graylog2.contentpacks.facades.OutputFacade)1 StreamFacade (org.graylog2.contentpacks.facades.StreamFacade)1 ModelType (org.graylog2.contentpacks.model.ModelType)1 Entity (org.graylog2.contentpacks.model.entities.Entity)1 EntityV1 (org.graylog2.contentpacks.model.entities.EntityV1)1 NativeEntityDescriptor (org.graylog2.contentpacks.model.entities.NativeEntityDescriptor)1 ClusterEventBus (org.graylog2.events.ClusterEventBus)1 User (org.graylog2.plugin.database.users.User)1 SuppressForbidden (org.graylog2.shared.SuppressForbidden)1 StreamRuleService (org.graylog2.streams.StreamRuleService)1 StreamRuleServiceImpl (org.graylog2.streams.StreamRuleServiceImpl)1