Examples with ClusterBlock org.opensearch.cluster.block.ClusterBlock used on opensource projects

Example 31 with ClusterBlock

use of org.opensearch.cluster.block.ClusterBlock in project OpenSearch by opensearch-project.

the class MetadataIndexStateServiceTests method testCloseRoutingTable.

public void testCloseRoutingTable() {
    final Set<Index> nonBlockedIndices = new HashSet<>();
    final Map<Index, ClusterBlock> blockedIndices = new HashMap<>();
    final Map<Index, IndexResult> results = new HashMap<>();
    ClusterState state = ClusterState.builder(new ClusterName("testCloseRoutingTable")).build();
    for (int i = 0; i < randomIntBetween(1, 25); i++) {
        final String indexName = "index-" + i;
        if (randomBoolean()) {
            state = addOpenedIndex(indexName, randomIntBetween(1, 5), randomIntBetween(0, 5), state);
            nonBlockedIndices.add(state.metadata().index(indexName).getIndex());
        } else {
            final ClusterBlock closingBlock = MetadataIndexStateService.createIndexClosingBlock();
            state = addBlockedIndex(indexName, randomIntBetween(1, 5), randomIntBetween(0, 5), state, closingBlock);
            final Index index = state.metadata().index(indexName).getIndex();
            blockedIndices.put(index, closingBlock);
            if (randomBoolean()) {
                results.put(index, new CloseIndexResponse.IndexResult(index));
            } else {
                results.put(index, new CloseIndexResponse.IndexResult(index, new Exception("test")));
            }
        }
    }
    final ClusterState updatedState = MetadataIndexStateService.closeRoutingTable(state, blockedIndices, results).v1();
    assertThat(updatedState.metadata().indices().size(), equalTo(nonBlockedIndices.size() + blockedIndices.size()));
    for (Index nonBlockedIndex : nonBlockedIndices) {
        assertIsOpened(nonBlockedIndex.getName(), updatedState);
        assertThat(updatedState.blocks().hasIndexBlockWithId(nonBlockedIndex.getName(), INDEX_CLOSED_BLOCK_ID), is(false));
    }
    for (Index blockedIndex : blockedIndices.keySet()) {
        if (results.get(blockedIndex).hasFailures() == false) {
            assertIsClosed(blockedIndex.getName(), updatedState);
        } else {
            assertIsOpened(blockedIndex.getName(), updatedState);
            assertThat(updatedState.blocks().hasIndexBlockWithId(blockedIndex.getName(), INDEX_CLOSED_BLOCK_ID), is(true));
        }
    }
}