Examples with IndexSet org.graylog2.indexer.IndexSet used on opensource projects

Example 61 with IndexSet

use of org.graylog2.indexer.IndexSet in project graylog2-server by Graylog2.

the class IndexSetsResourceTest method delete0.

@Test
public void delete0() throws Exception {
    final IndexSet indexSet = mock(IndexSet.class);
    final IndexSetConfig indexSetConfig = mock(IndexSetConfig.class);
    when(indexSet.getConfig()).thenReturn(indexSetConfig);
    when(indexSetRegistry.getDefault()).thenReturn(null);
    when(indexSetRegistry.get("id")).thenReturn(Optional.of(indexSet));
    when(indexSetService.delete("id")).thenReturn(0);
    expectedException.expect(NotFoundException.class);
    expectedException.expectMessage("Couldn't delete index set with ID <id>");
    try {
        indexSetsResource.delete("id", false);
    } finally {
        verify(indexSetRegistry, times(1)).getDefault();
        verify(indexSetService, times(1)).delete("id");
        verifyNoMoreInteractions(indexSetService);
    }
}

Example 62 with IndexSet

use of org.graylog2.indexer.IndexSet in project graylog2-server by Graylog2.

the class IndexSetsResourceTest method setDefaultMakesIndexDefaultIfWritable.

@Test
public void setDefaultMakesIndexDefaultIfWritable() throws Exception {
    final String indexSetId = "newDefaultIndexSetId";
    final IndexSet indexSet = mock(IndexSet.class);
    final IndexSetConfig indexSetConfig = IndexSetConfig.create(indexSetId, "title", "description", true, "prefix", 1, 0, MessageCountRotationStrategy.class.getCanonicalName(), MessageCountRotationStrategyConfig.create(1000), NoopRetentionStrategy.class.getCanonicalName(), NoopRetentionStrategyConfig.create(1), ZonedDateTime.of(2016, 10, 10, 12, 0, 0, 0, ZoneOffset.UTC), "standard", "index-template", 1, false);
    when(indexSet.getConfig()).thenReturn(indexSetConfig);
    when(indexSetService.get(indexSetId)).thenReturn(Optional.of(indexSetConfig));
    indexSetsResource.setDefault(indexSetId);
    final ArgumentCaptor<DefaultIndexSetConfig> defaultIndexSetIdCaptor = ArgumentCaptor.forClass(DefaultIndexSetConfig.class);
    verify(clusterConfigService, times(1)).write(defaultIndexSetIdCaptor.capture());
    final DefaultIndexSetConfig defaultIndexSetConfig = defaultIndexSetIdCaptor.getValue();
    assertThat(defaultIndexSetConfig).isNotNull();
    assertThat(defaultIndexSetConfig.defaultIndexSetId()).isEqualTo(indexSetId);
}

Example 63 with IndexSet

use of org.graylog2.indexer.IndexSet in project graylog2-server by Graylog2.

the class IndexSetsResourceTest method setDefaultDoesNotDoAnythingIfIndexSetIsNotWritable.

@Test
public void setDefaultDoesNotDoAnythingIfIndexSetIsNotWritable() throws Exception {
    final String readOnlyIndexSetId = "newDefaultIndexSetId";
    final IndexSet readOnlyIndexSet = mock(IndexSet.class);
    final IndexSetConfig readOnlyIndexSetConfig = IndexSetConfig.create(readOnlyIndexSetId, "title", "description", false, "prefix", 1, 0, MessageCountRotationStrategy.class.getCanonicalName(), MessageCountRotationStrategyConfig.create(1000), NoopRetentionStrategy.class.getCanonicalName(), NoopRetentionStrategyConfig.create(1), ZonedDateTime.of(2016, 10, 10, 12, 0, 0, 0, ZoneOffset.UTC), "standard", "index-template", 1, false);
    when(readOnlyIndexSet.getConfig()).thenReturn(readOnlyIndexSetConfig);
    when(indexSetService.get(readOnlyIndexSetId)).thenReturn(Optional.of(readOnlyIndexSetConfig));
    expectedException.expect(ClientErrorException.class);
    expectedException.expectMessage("Default index set must be writable.");
    try {
        indexSetsResource.setDefault(readOnlyIndexSetId);
    } finally {
        verifyZeroInteractions(clusterConfigService);
    }
}