org.elasticsearch.ResourceNotFoundException Java Examples

@Test
void testIndexThrowsResourceNotFoundException() {
  Index index = Index.create("index");

  when(document.getContent()).thenReturn(xContentBuilder);
  when(document.getId()).thenReturn("id");

  when(client.prepareIndex()).thenReturn(indexRequestBuilder);
  when(indexRequestBuilder.setIndex("index")).thenReturn(indexRequestBuilder);
  when(indexRequestBuilder.setType(any())).thenReturn(indexRequestBuilder);
  when(indexRequestBuilder.setId("id")).thenReturn(indexRequestBuilder);
  when(indexRequestBuilder.setSource(xContentBuilder)).thenReturn(indexRequestBuilder);

  when(indexRequestBuilder.get()).thenThrow(new ResourceNotFoundException("exception"));

  Exception exception =
      assertThrows(UnknownIndexException.class, () -> clientFacade.index(index, document));
  assertThat(exception.getMessage()).containsPattern("Index 'index' not found\\.");
}
 

@Test
void testDeleteResourceNotFound() {
  Index index = Index.create("index");

  when(document.getId()).thenReturn("id");

  when(client.prepareDelete()).thenReturn(deleteRequestBuilder);
  when(deleteRequestBuilder.setIndex("index")).thenReturn(deleteRequestBuilder);
  when(deleteRequestBuilder.setType(any())).thenReturn(deleteRequestBuilder);
  when(deleteRequestBuilder.setId("id")).thenReturn(deleteRequestBuilder);

  when(deleteRequestBuilder.get()).thenThrow(new ResourceNotFoundException("exception"));

  Exception exception =
      assertThrows(UnknownIndexException.class, () -> clientFacade.deleteById(index, document));
  assertThat(exception.getMessage()).containsPattern("Index 'index' not found\\.");
}
 

public static Iterable<SqlFeatureContext> loadFeatures() throws IOException {
    try (InputStream sqlFeatures = SqlFeatures.class.getResourceAsStream("/sql_features.tsv")) {
        if (sqlFeatures == null) {
            throw new ResourceNotFoundException("sql_features.tsv file not found");
        }
        ArrayList<SqlFeatureContext> featuresList = new ArrayList<>();
        try (BufferedReader reader = new BufferedReader(new InputStreamReader(sqlFeatures, StandardCharsets.UTF_8))) {
            String next;
            while ((next = reader.readLine()) != null) {
                List<String> parts = List.of(next.split("\t", NUM_COLS));
                var ctx = new SqlFeatureContext(parts.get(0),
                    parts.get(1),
                    parts.get(2),
                    parts.get(3),
                    parts.get(4).equals("YES"),
                    parts.get(5).isEmpty() ? null : parts.get(5),
                    parts.get(6).isEmpty() ? null : parts.get(6));
                featuresList.add(ctx);
            }
        }
        return featuresList;
    }
}
 

@VisibleForTesting
static ClusterState updateTemplate(NamedXContentRegistry xContentRegistry,
                                   ClusterState currentState,
                                   String templateName,
                                   Map<String, Object> newMapping) throws Exception {
    IndexTemplateMetaData template = currentState.metaData().templates().get(templateName);
    if (template == null) {
        throw new ResourceNotFoundException("Template \"" + templateName + "\" for partitioned table is missing");
    }

    IndexTemplateMetaData.Builder templateBuilder = new IndexTemplateMetaData.Builder(template);
    for (ObjectObjectCursor<String, CompressedXContent> cursor : template.mappings()) {
        Map<String, Object> source = parseMapping(xContentRegistry, cursor.value.toString());
        mergeIntoSource(source, newMapping);
        try (XContentBuilder xContentBuilder = JsonXContent.contentBuilder()) {
            templateBuilder.putMapping(cursor.key, Strings.toString(xContentBuilder.map(source)));
        }
    }
    MetaData.Builder builder = MetaData.builder(currentState.metaData()).put(templateBuilder);
    return ClusterState.builder(currentState).metaData(builder).build();
}
 

@Test
void testRefreshIndicesNotFound() {
  when(indicesAdminClient.prepareRefresh("_all")).thenReturn(refreshRequestBuilder);
  when(refreshRequestBuilder.get()).thenThrow(new ResourceNotFoundException("exception"));

  Exception exception =
      assertThrows(UnknownIndexException.class, () -> clientFacade.refreshIndexes());
  assertThat(exception.getMessage()).containsPattern("One or more indexes '_all' not found\\.");
}
 

@Test
void testGetCountThrowsResourceNotFoundException() {
  Index index = Index.create("index");

  when(client.prepareSearch("index")).thenReturn(searchRequestBuilder);
  when(searchRequestBuilder.get()).thenThrow(new ResourceNotFoundException("exception"));

  Exception exception =
      assertThrows(UnknownIndexException.class, () -> clientFacade.getCount(index));
  assertThat(exception.getMessage()).containsPattern("One or more indexes 'index' not found\\.");
}
 

@Test
void testSearchIndexNotFound() {
  Index index = Index.create("index");

  when(client.prepareSearch("index")).thenReturn(searchRequestBuilder);
  when(searchRequestBuilder.get()).thenThrow(new ResourceNotFoundException("Exception"));
  when(queryBuilder.toString()).thenReturn("a == b");

  Exception exception =
      assertThrows(
          UnknownIndexException.class,
          () -> clientFacade.search(queryBuilder, 0, 100, ImmutableList.of(index)));
  assertThat(exception.getMessage()).containsPattern("One or more indexes 'index' not found\\.");
}
 

@Test
void testAggregateThrowsResourceNotFoundException() {
  Index index = Index.create("index");

  when(client.prepareSearch("index")).thenReturn(searchRequestBuilder);
  when(searchRequestBuilder.get()).thenThrow(new ResourceNotFoundException("Exception"));
  when(queryBuilder.toString()).thenReturn("a == b");

  Exception exception =
      assertThrows(
          UnknownIndexException.class,
          () ->
              clientFacade.aggregate(ImmutableList.of(aggregationBuilder), queryBuilder, index));
  assertThat(exception.getMessage()).containsPattern("One or more indexes 'index' not found\\.");
}