Examples with BatchPartitionSizeProperties com.sequenceiq.freeipa.configuration.BatchPartitionSizeProperties used on opensource projects

Example 1 with BatchPartitionSizeProperties

use of com.sequenceiq.freeipa.configuration.BatchPartitionSizeProperties in project cloudbreak by hortonworks.

the class UserSyncOperationsTest method testRemoveUsersFromGroupsSingle.

@Test
public void testRemoveUsersFromGroupsSingle() throws Exception {
    Multimap<String, String> groupMapping = setupGroupMapping(5, MAX_SUBJECTS_PER_REQUEST * 2);
    Multimap<String, String> warnings = ArrayListMultimap.create();
    ArgumentCaptor<List<Object>> flagsCaptor = ArgumentCaptor.forClass(List.class);
    ArgumentCaptor<Map<String, Object>> paramsCaptor = ArgumentCaptor.forClass(Map.class);
    RPCResponse<Object> response1 = new RPCResponse<>();
    response1.setResult(new Group());
    RPCResponse<Object> response2 = new RPCResponse<>();
    response2.setResult(new Group());
    when(freeIpaClient.invoke(eq("group_remove_member"), flagsCaptor.capture(), paramsCaptor.capture(), eq(Group.class))).thenReturn(response1, response2);
    underTest.removeUsersFromGroups(false, freeIpaClient, groupMapping, warnings::put);
    assertTrue(warnings.isEmpty());
    verifyNoInteractions(batchPartitionSizeProperties);
    verify(freeIpaClient, never()).callBatch(any(), any(), any(), any(), any());
    List<List<Object>> flagsList = flagsCaptor.getAllValues();
    Map<String, Long> flagCount = flagsList.stream().flatMap(Collection::stream).map(o -> (String) o).collect(Collectors.groupingBy(Function.identity(), Collectors.counting()));
    assertEquals(5, flagCount.size());
    assertTrue(flagCount.keySet().containsAll(Set.of("group1", "group2", "group3", "group4", "group0")));
    flagCount.forEach((flag, count) -> assertEquals(2, count, flag + " size should be 2"));
    List<Map<String, Object>> paramsList = paramsCaptor.getAllValues();
    assertEquals(10, paramsList.size());
    paramsList.forEach(map -> {
        assertEquals(10, ((List<String>) map.get("user")).size());
        assertTrue(((List<String>) map.get("user")).stream().allMatch(user -> user.startsWith("user")));
    });
    verify(interruptChecker, times(20)).throwTimeoutExIfInterrupted();
}

Example 2 with BatchPartitionSizeProperties

use of com.sequenceiq.freeipa.configuration.BatchPartitionSizeProperties in project cloudbreak by hortonworks.

the class UserSyncOperationsTest method testAddUsersToGroupsSingle.

@Test
public void testAddUsersToGroupsSingle() throws Exception {
    Multimap<String, String> groupMapping = setupGroupMapping(5, MAX_SUBJECTS_PER_REQUEST * 2);
    Multimap<String, String> warnings = ArrayListMultimap.create();
    ArgumentCaptor<List<Object>> flagsCaptor = ArgumentCaptor.forClass(List.class);
    ArgumentCaptor<Map<String, Object>> paramsCaptor = ArgumentCaptor.forClass(Map.class);
    RPCResponse<Object> response = new RPCResponse<>();
    Group result = new Group();
    result.setMemberUser(List.copyOf(groupMapping.values()));
    response.setResult(result);
    when(freeIpaClient.invoke(eq("group_add_member"), flagsCaptor.capture(), paramsCaptor.capture(), eq(Group.class))).thenReturn(response);
    underTest.addUsersToGroups(false, freeIpaClient, groupMapping, warnings::put);
    assertTrue(warnings.isEmpty());
    verifyNoInteractions(batchPartitionSizeProperties);
    verify(freeIpaClient, never()).callBatch(any(), any(), any(), any(), any());
    List<List<Object>> flagsList = flagsCaptor.getAllValues();
    Map<String, Long> flagCount = flagsList.stream().flatMap(Collection::stream).map(o -> (String) o).collect(Collectors.groupingBy(Function.identity(), Collectors.counting()));
    assertEquals(5, flagCount.size());
    assertTrue(flagCount.keySet().containsAll(Set.of("group1", "group2", "group3", "group4", "group0")));
    flagCount.forEach((flag, count) -> assertEquals(2, count, flag + " size should be 2"));
    List<Map<String, Object>> paramsList = paramsCaptor.getAllValues();
    assertEquals(10, paramsList.size());
    paramsList.forEach(map -> {
        assertEquals(10, ((List<String>) map.get("user")).size());
        assertTrue(((List<String>) map.get("user")).stream().allMatch(user -> user.startsWith("user")));
    });
    verify(interruptChecker, times(20)).throwTimeoutExIfInterrupted();
}