Search in sources :

Example 16 with Group

use of com.sequenceiq.freeipa.client.model.Group 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();
}
Also used : Assertions.assertThrows(org.junit.jupiter.api.Assertions.assertThrows) ArgumentMatchers.any(org.mockito.ArgumentMatchers.any) ArrayListMultimap(com.google.common.collect.ArrayListMultimap) BeforeEach(org.junit.jupiter.api.BeforeEach) Matchers.aMapWithSize(org.hamcrest.Matchers.aMapWithSize) FmsGroup(com.sequenceiq.freeipa.service.freeipa.user.model.FmsGroup) ArgumentMatchers.eq(org.mockito.ArgumentMatchers.eq) Mock(org.mockito.Mock) RPCResponse(com.sequenceiq.cloudbreak.client.RPCResponse) TimeoutException(java.util.concurrent.TimeoutException) Multimap(com.google.common.collect.Multimap) Function(java.util.function.Function) Group(com.sequenceiq.freeipa.client.model.Group) Mockito.verifyNoInteractions(org.mockito.Mockito.verifyNoInteractions) Mockito.doThrow(org.mockito.Mockito.doThrow) ArgumentCaptor(org.mockito.ArgumentCaptor) HashMultimap(com.google.common.collect.HashMultimap) Matchers.everyItem(org.hamcrest.Matchers.everyItem) Assertions.assertFalse(org.junit.jupiter.api.Assertions.assertFalse) ExtendWith(org.junit.jupiter.api.extension.ExtendWith) Map(java.util.Map) BatchPartitionSizeProperties(com.sequenceiq.freeipa.configuration.BatchPartitionSizeProperties) ThreadInterruptChecker(com.sequenceiq.freeipa.util.ThreadInterruptChecker) MatcherAssert.assertThat(org.hamcrest.MatcherAssert.assertThat) Assertions.assertEquals(org.junit.jupiter.api.Assertions.assertEquals) Matchers.hasEntry(org.hamcrest.Matchers.hasEntry) InjectMocks(org.mockito.InjectMocks) MockitoExtension(org.mockito.junit.jupiter.MockitoExtension) Matchers.allOf(org.hamcrest.Matchers.allOf) Collection(java.util.Collection) Set(java.util.Set) Matchers(org.hamcrest.Matchers) ReflectionTestUtils(org.springframework.test.util.ReflectionTestUtils) User(com.sequenceiq.freeipa.client.model.User) Mockito.times(org.mockito.Mockito.times) FreeIpaClientException(com.sequenceiq.freeipa.client.FreeIpaClientException) Mockito.when(org.mockito.Mockito.when) FreeIpaClient(com.sequenceiq.freeipa.client.FreeIpaClient) FmsUser(com.sequenceiq.freeipa.service.freeipa.user.model.FmsUser) Collectors(java.util.stream.Collectors) FreeIpaErrorCodes(com.sequenceiq.freeipa.client.FreeIpaErrorCodes) Mockito.verify(org.mockito.Mockito.verify) Test(org.junit.jupiter.api.Test) Mockito.never(org.mockito.Mockito.never) List(java.util.List) Matchers.hasItem(org.hamcrest.Matchers.hasItem) Assertions.assertTrue(org.junit.jupiter.api.Assertions.assertTrue) JsonRpcClientException(com.googlecode.jsonrpc4j.JsonRpcClientException) FmsGroup(com.sequenceiq.freeipa.service.freeipa.user.model.FmsGroup) Group(com.sequenceiq.freeipa.client.model.Group) RPCResponse(com.sequenceiq.cloudbreak.client.RPCResponse) Collection(java.util.Collection) List(java.util.List) Map(java.util.Map) Test(org.junit.jupiter.api.Test)

Example 17 with Group

use of com.sequenceiq.freeipa.client.model.Group in project cloudbreak by hortonworks.

the class UserSyncOperationsTest method testAddGroupsSingle.

@Test
public void testAddGroupsSingle() throws FreeIpaClientException, TimeoutException {
    Multimap<String, String> warnings = ArrayListMultimap.create();
    FmsGroup group1 = new FmsGroup().withName(UserSyncConstants.CDP_USERSYNC_INTERNAL_GROUP);
    FmsGroup group2 = new FmsGroup().withName("group2");
    Set<FmsGroup> groups = Set.of(group1, group2);
    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_add"), flagsCaptor.capture(), paramsCaptor.capture(), eq(Group.class))).thenReturn(response1, response2);
    underTest.addGroups(false, freeIpaClient, groups, warnings::put);
    assertTrue(warnings.isEmpty());
    verifyNoInteractions(batchPartitionSizeProperties);
    verify(freeIpaClient, never()).callBatch(any(), any(), any(), any(), any());
    List<List<Object>> flagsList = flagsCaptor.getAllValues();
    assertThat(flagsList, allOf(hasItem(hasItem(UserSyncConstants.CDP_USERSYNC_INTERNAL_GROUP)), hasItem(hasItem("group2"))));
    List<Map<String, Object>> paramsList = paramsCaptor.getAllValues();
    assertTrue(paramsList.get(0).isEmpty());
    assertEquals(true, paramsList.get(1).get("nonposix"));
    verify(interruptChecker, times(4)).throwTimeoutExIfInterrupted();
}
Also used : FmsGroup(com.sequenceiq.freeipa.service.freeipa.user.model.FmsGroup) Group(com.sequenceiq.freeipa.client.model.Group) FmsGroup(com.sequenceiq.freeipa.service.freeipa.user.model.FmsGroup) RPCResponse(com.sequenceiq.cloudbreak.client.RPCResponse) List(java.util.List) Map(java.util.Map) Test(org.junit.jupiter.api.Test)

Example 18 with Group

use of com.sequenceiq.freeipa.client.model.Group in project cloudbreak by hortonworks.

the class FreeipaClientTestServiceTest method createGroup.

private Group createGroup(String group, String user) {
    Group ipaGroup = new Group();
    ipaGroup.setCn(group);
    ipaGroup.setMemberUser(Lists.newArrayList(user));
    return ipaGroup;
}
Also used : Group(com.sequenceiq.freeipa.client.model.Group)

Example 19 with Group

use of com.sequenceiq.freeipa.client.model.Group in project cloudbreak by hortonworks.

the class GroupAddOperation method invoke.

@Override
public Optional<Group> invoke(FreeIpaClient freeIpaClient) throws FreeIpaClientException {
    FreeIpaChecks.checkGroupNotProtected(group, () -> String.format("Group '%s' is protected and cannot be added to FreeIPA", group));
    try {
        LOGGER.debug("adding group {}", group);
        Group groupAdded = invoke(freeIpaClient, Group.class);
        LOGGER.debug("Success: {}", groupAdded);
        return Optional.of(groupAdded);
    } catch (FreeIpaClientException e) {
        if (FreeIpaClientExceptionUtil.isDuplicateEntryException(e)) {
            LOGGER.debug("group '{}' already exists", group);
        } else {
            LOGGER.warn("Failed to add group {}", group, e);
            warnings.accept(group, "Failed to add group:" + e.getMessage());
            freeIpaClient.checkIfClientStillUsable(e);
        }
    }
    return Optional.empty();
}
Also used : Group(com.sequenceiq.freeipa.client.model.Group) FreeIpaClientException(com.sequenceiq.freeipa.client.FreeIpaClientException)

Example 20 with Group

use of com.sequenceiq.freeipa.client.model.Group in project cloudbreak by hortonworks.

the class GroupRemoveOperation method invoke.

@Override
public Optional<Group> invoke(FreeIpaClient freeIpaClient) throws FreeIpaClientException {
    FreeIpaChecks.checkGroupNotProtected(group, () -> String.format("Group '%s' is protected and cannot be deleted from FreeIPA", group));
    try {
        LOGGER.debug("Removing group {}", group);
        Group groupRemoved = invoke(freeIpaClient, Group.class);
        LOGGER.debug("Success: {}", groupRemoved);
        return Optional.of(groupRemoved);
    } catch (FreeIpaClientException e) {
        if (FreeIpaClientExceptionUtil.isNotFoundException(e)) {
            LOGGER.debug("group '{}' already does not exists", group);
        } else {
            LOGGER.error("Failed to delete {}", group, e);
            warnings.accept(group, "Failed to remove group: " + e.getMessage());
            freeIpaClient.checkIfClientStillUsable(e);
        }
    }
    return Optional.empty();
}
Also used : Group(com.sequenceiq.freeipa.client.model.Group) FreeIpaClientException(com.sequenceiq.freeipa.client.FreeIpaClientException)

Aggregations

Group (com.sequenceiq.freeipa.client.model.Group)21 Test (org.junit.jupiter.api.Test)14 RPCResponse (com.sequenceiq.cloudbreak.client.RPCResponse)12 Map (java.util.Map)12 FreeIpaClientException (com.sequenceiq.freeipa.client.FreeIpaClientException)5 FmsGroup (com.sequenceiq.freeipa.service.freeipa.user.model.FmsGroup)4 List (java.util.List)4 FreeIpaClient (com.sequenceiq.freeipa.client.FreeIpaClient)3 ArrayListMultimap (com.google.common.collect.ArrayListMultimap)2 HashMultimap (com.google.common.collect.HashMultimap)2 Multimap (com.google.common.collect.Multimap)2 JsonRpcClientException (com.googlecode.jsonrpc4j.JsonRpcClientException)2 FreeIpaErrorCodes (com.sequenceiq.freeipa.client.FreeIpaErrorCodes)2 SudoRule (com.sequenceiq.freeipa.client.model.SudoRule)2 User (com.sequenceiq.freeipa.client.model.User)2 BatchPartitionSizeProperties (com.sequenceiq.freeipa.configuration.BatchPartitionSizeProperties)2 FmsUser (com.sequenceiq.freeipa.service.freeipa.user.model.FmsUser)2 ThreadInterruptChecker (com.sequenceiq.freeipa.util.ThreadInterruptChecker)2 Collection (java.util.Collection)2 Set (java.util.Set)2