Package org.jtalks.jcommune.service.security

Source Code of org.jtalks.jcommune.service.security.TransactionalPermissionServiceTest

/**
* Copyright (C) 2011  JTalks.org Team
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either
* version 2.1 of the License, or (at your option) any later version.
* This library is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
* Lesser General Public License for more details.
* You should have received a copy of the GNU Lesser General Public
* License along with this library; if not, write to the Free Software
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
*/
package org.jtalks.jcommune.service.security;

import org.jtalks.common.model.entity.Component;
import org.jtalks.common.model.entity.Group;
import org.jtalks.common.model.permissions.BranchPermission;
import org.jtalks.common.model.permissions.GeneralPermission;
import org.jtalks.common.model.permissions.JtalksPermission;
import org.jtalks.common.service.security.SecurityContextHolderFacade;
import org.jtalks.jcommune.model.dto.GroupsPermissions;
import org.jtalks.jcommune.model.dto.PermissionChanges;
import org.jtalks.jcommune.model.entity.Branch;
import org.jtalks.jcommune.service.transactional.TransactionalPermissionService;
import org.mockito.Mock;
import org.springframework.security.access.AccessDeniedException;
import org.springframework.security.core.Authentication;
import org.springframework.security.core.context.SecurityContext;
import org.testng.annotations.BeforeMethod;
import org.testng.annotations.Test;

import java.io.Serializable;
import java.util.List;

import static org.mockito.Mockito.*;
import static org.mockito.MockitoAnnotations.initMocks;
import static org.testng.Assert.*;

public class TransactionalPermissionServiceTest {

    @Mock
    private SecurityContextHolderFacade contextFacade;
    @Mock
    private AclGroupPermissionEvaluator aclEvaluator;

    private PermissionService permissionService;
    private PermissionManager permissionManager;

    @BeforeMethod
    public void initEnvironmental() {
        initMocks(this);
        permissionManager = mock(PermissionManager.class);
        permissionService = spy(new TransactionalPermissionService(contextFacade, aclEvaluator, permissionManager));
    }

    @BeforeMethod
    public void prepareTestData() {
        Authentication authentication = mock(Authentication.class);
        SecurityContext securityContext = mock(SecurityContext.class);

        when(contextFacade.getContext()).thenReturn(securityContext);
        when(securityContext.getAuthentication()).thenReturn(authentication);
    }

    @Test
    public void testHasPermissionStringPermissionPermissionGranted() {
        doReturn(true).when(aclEvaluator)
                .hasPermission(any(Authentication.class), any(Serializable.class), anyString(), anyString());

        assertTrue(permissionService.hasPermission(0, null, ""));
    }

    @Test
    public void testHasPermissionStringPermissionPermissionNotGranted() {
        doReturn(false).when(aclEvaluator)
                .hasPermission(any(Authentication.class), any(Serializable.class), anyString(), anyString());

        assertFalse(permissionService.hasPermission(0, null, ""));
    }

    @Test
    public void testHasPermissionEnumPermissionPermissionGranted() {
        doReturn(true).when(aclEvaluator)
                .hasPermission(any(Authentication.class), any(Serializable.class), anyString(), anyString());

        assertTrue(permissionService.hasPermission(0, AclClassName.BRANCH, GeneralPermission.READ));
    }

    @Test
    public void testHasPermissionEnumPermissionPermissionNotGranted() {
        doReturn(false).when(aclEvaluator)
                .hasPermission(any(Authentication.class), any(Serializable.class), anyString(), anyString());

        assertFalse(permissionService.hasPermission(0, AclClassName.BRANCH, GeneralPermission.READ));
    }

    @Test
    public void testCheckPermissionPermissionGranted() {
        doReturn(true).when(aclEvaluator)
                .hasPermission(any(Authentication.class), any(Serializable.class), anyString(), anyString());

        permissionService.checkPermission(0, AclClassName.BRANCH, GeneralPermission.READ);
    }

    @Test(expectedExceptions = AccessDeniedException.class)
    public void testCheckPermissionPermissionNotGranted() {
        doReturn(false).when(aclEvaluator)
                .hasPermission(any(Authentication.class), any(Serializable.class), anyString(), anyString());

        permissionService.checkPermission(0, AclClassName.BRANCH, GeneralPermission.READ);
    }

    /**
     * Tests whether method constructs data that is going to be passed to the evaluator is correct and thus {@link
     * PermissionService#hasBranchPermission(long, JtalksPermission)} will return same result as evaluator.
     */
    @Test
    public void hasBranchPermissionIsTrueIfEvaluatorSaidSo() {
        doReturn(true).when(aclEvaluator).hasPermission(any(Authentication.class),
                eq(1L),
                eq(AclClassName.BRANCH.toString()),
                eq("BranchPermission.EDIT_OWN_POSTS"));

        permissionService.hasBranchPermission(1L, BranchPermission.EDIT_OWN_POSTS);
    }

    @Test
    public void testGetPermissionsFor() {
        Branch branch = mock(Branch.class);
        doReturn(new GroupsPermissions()).when(permissionManager).getPermissionsMapFor(branch);
        assertNotNull(permissionService.getPermissionsFor(branch));
    }

    @Test
    public void testChangeGrants() {
        PermissionChanges changes = mock(PermissionChanges.class);
        Branch branch = mock(Branch.class);
        permissionService.changeGrants(branch, changes);

        Component component = mock(Component.class);
        permissionService.changeGrants(component, changes);

        Group group = mock(Group.class);
        permissionService.changeGrants(group, changes);
    }

    @Test
    public void testChangeRestrictions() {
        PermissionChanges changes = mock(PermissionChanges.class);
        Branch branch = mock(Branch.class);
        permissionService.changeRestrictions(branch, changes);

        Component component = mock(Component.class);
        permissionService.changeRestrictions(component, changes);

        Group group = mock(Group.class);
        permissionService.changeRestrictions(group, changes);
    }

    @Test
    public void testGetPermissionsMapFor() {
        Component component = mock(Component.class);
        doReturn(new GroupsPermissions()).when(permissionManager).getPermissionsMapFor(component);
        assertNotNull(permissionService.getPermissionsMapFor(component));
    }

    @Test
    public void testGetPersonalPermissions() {
        List<Group> groups = mock(List.class);
        doReturn(new GroupsPermissions()).when(permissionManager).getPermissionsMapFor(groups);
        assertNotNull(permissionService.getPersonalPermissions(groups));
    }
}
TOP

Related Classes of org.jtalks.jcommune.service.security.TransactionalPermissionServiceTest

TOP
Copyright © 2018 www.massapi.com. All rights reserved.
All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.