Examples with DBCollection com.mongodb.DBCollection used on opensource projects

Example 46 with DBCollection

use of com.mongodb.DBCollection in project graylog2-server by Graylog2.

the class ClusterEventPeriodicalTest method publishClusterEventHandlesAutoValueCorrectly.

@Test
@UsingDataSet(loadStrategy = LoadStrategyEnum.DELETE_ALL)
public void publishClusterEventHandlesAutoValueCorrectly() throws Exception {
    DBCollection collection = mongoConnection.getDatabase().getCollection(ClusterEventPeriodical.COLLECTION_NAME);
    DebugEvent event = DebugEvent.create("Node ID", "Test");
    assertThat(collection.count()).isEqualTo(0L);
    clusterEventPeriodical.publishClusterEvent(event);
    verify(clusterEventBus, never()).post(any());
    assertThat(collection.count()).isEqualTo(1L);
    DBObject dbObject = collection.findOne();
    assertThat((String) dbObject.get("producer")).isEqualTo(nodeId.toString());
    assertThat((String) dbObject.get("event_class")).isEqualTo(DebugEvent.class.getCanonicalName());
}

Example 47 with DBCollection

use of com.mongodb.DBCollection in project graylog2-server by Graylog2.

the class ClusterEventPeriodicalTest method prepareCollectionCreatesIndexesOnExistingCollection.

@Test
public void prepareCollectionCreatesIndexesOnExistingCollection() throws Exception {
    DBCollection original = mongoConnection.getDatabase().getCollection(ClusterEventPeriodical.COLLECTION_NAME);
    original.dropIndexes();
    assertThat(original.getName()).isEqualTo(ClusterEventPeriodical.COLLECTION_NAME);
    assertThat(original.getIndexInfo()).hasSize(1);
    DBCollection collection = ClusterEventPeriodical.prepareCollection(mongoConnection);
    assertThat(collection.getName()).isEqualTo(ClusterEventPeriodical.COLLECTION_NAME);
    assertThat(collection.getIndexInfo()).hasSize(2);
    assertThat(collection.getWriteConcern()).isEqualTo(WriteConcern.JOURNALED);
}

Example 48 with DBCollection

use of com.mongodb.DBCollection in project graylog2-server by Graylog2.

the class ClusterEventPeriodicalTest method testPublishClusterEvent.

@Test
@UsingDataSet(loadStrategy = LoadStrategyEnum.DELETE_ALL)
public void testPublishClusterEvent() throws Exception {
    DBCollection collection = mongoConnection.getDatabase().getCollection(ClusterEventPeriodical.COLLECTION_NAME);
    SimpleEvent event = new SimpleEvent("test");
    assertThat(collection.count()).isEqualTo(0L);
    clusterEventPeriodical.publishClusterEvent(event);
    verify(clusterEventBus, never()).post(any());
    assertThat(collection.count()).isEqualTo(1L);
    DBObject dbObject = collection.findOne();
    assertThat((String) dbObject.get("producer")).isEqualTo(nodeId.toString());
    assertThat((String) dbObject.get("event_class")).isEqualTo(SimpleEvent.class.getCanonicalName());
    @SuppressWarnings("unchecked") Map<String, Object> payload = (Map<String, Object>) dbObject.get("payload");
    assertThat(payload).containsEntry("payload", "test");
}

Example 49 with DBCollection

use of com.mongodb.DBCollection in project graylog2-server by Graylog2.

the class ClusterEventPeriodicalTest method serverEventBusDispatchesTypedEvents.

@Test
@UsingDataSet(loadStrategy = LoadStrategyEnum.DELETE_ALL)
public void serverEventBusDispatchesTypedEvents() throws Exception {
    final SimpleEventHandler handler = new SimpleEventHandler();
    serverEventBus.register(handler);
    DBObject event = new BasicDBObjectBuilder().add("timestamp", TIME.getMillis()).add("producer", "TEST-PRODUCER").add("consumers", Collections.emptyList()).add("event_class", SimpleEvent.class.getCanonicalName()).add("payload", ImmutableMap.of("payload", "test")).get();
    final DBCollection collection = mongoConnection.getDatabase().getCollection(ClusterEventPeriodical.COLLECTION_NAME);
    assertThat(collection.save(event).getN()).isEqualTo(1);
    assertThat(collection.count()).isEqualTo(1L);
    assertThat(handler.invocations).isEqualTo(0);
    clusterEventPeriodical.run();
    assertThat(handler.invocations).isEqualTo(1);
    assertThat(collection.count()).isEqualTo(1L);
    @SuppressWarnings("unchecked") final List<String> consumers = (List<String>) collection.findOne().get("consumers");
    assertThat(consumers).containsExactly(nodeId.toString());
    verify(serverEventBus, times(1)).post(any(SimpleEvent.class));
    verify(clusterEventBus, never()).post(any());
}

Example 50 with DBCollection

use of com.mongodb.DBCollection in project graylog2-server by Graylog2.

the class ClusterConfigServiceImplTest method prepareCollectionCreatesIndexesOnExistingCollection.

@Test
public void prepareCollectionCreatesIndexesOnExistingCollection() throws Exception {
    DBCollection original = mongoConnection.getDatabase().getCollection(COLLECTION_NAME);
    original.dropIndexes();
    assertThat(original.getName()).isEqualTo(COLLECTION_NAME);
    assertThat(original.getIndexInfo()).hasSize(1);
    DBCollection collection = ClusterConfigServiceImpl.prepareCollection(mongoConnection);
    assertThat(collection.getName()).isEqualTo(COLLECTION_NAME);
    assertThat(collection.getIndexInfo()).hasSize(2);
    assertThat(collection.getWriteConcern()).isEqualTo(WriteConcern.JOURNALED);
}