@Test public void testShardPartitions() throws IOException { List<List<Map<String, Object>>> shards = MAPPER.readValue(getClass().getResourceAsStream("search-shards-response.json"), ArrayList.class); List<PartitionDefinition> partitions = RestService.findShardPartitions(null, null, Collections.<String, NodeInfo>emptyMap(), shards, LOGGER); Collections.sort(partitions); assertEquals(partitions.size(), 34); assertEquals(new HashSet(partitions).size(), 34); assertArrayEquals(partitions.toArray(), EXPECTED_SHARDS_PARTITIONS); }
@Test public void testEmpty() { Settings settings = new PropertiesSettings(); settings.setMaxDocsPerPartition(10000); settings.setInternalVersion(EsMajorVersion.LATEST); settings.setProperty(ES_RESOURCE_READ, "_all"); assertEquals(RestService.findShardPartitions(settings, null, Collections.<String, NodeInfo>emptyMap(), Collections.<List<Map<String,Object>>>emptyList(), LOGGER).size(), 0); assertEquals(RestService.findSlicePartitions(null, settings, null, Collections.<String, NodeInfo>emptyMap(), Collections.<List<Map<String,Object>>>emptyList(), LOGGER).size(), 0); }
partitions = findSlicePartitions(client.getRestClient(), settings, mapping, nodesMap, shards, log); } else { partitions = findShardPartitions(settings, mapping, nodesMap, shards, log);
partitions = findSlicePartitions(client.getRestClient(), settings, mapping, nodesMap, shards, log); } else { partitions = findShardPartitions(settings, mapping, nodesMap, shards, log);
partitions = findSlicePartitions(client.getRestClient(), settings, mapping, nodesMap, shards, log); } else { partitions = findShardPartitions(settings, mapping, nodesMap, shards, log);
partitions = findSlicePartitions(client.getRestClient(), settings, mapping, nodesMap, shards, log); } else { partitions = findShardPartitions(settings, mapping, nodesMap, shards, log);