org.jtalks.common.security.acl.AclDataProvider.java Source code

Introduction

Here is the source code for org.jtalks.common.security.acl.AclDataProvider.java

Source

/**
 * 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.common.security.acl;

import com.google.common.collect.Lists;
import org.jtalks.common.model.entity.Entity;
import org.jtalks.common.model.permissions.BranchPermission;
import org.jtalks.common.model.permissions.GeneralPermission;
import org.jtalks.common.security.acl.sids.UserGroupSid;
import org.jtalks.common.security.acl.sids.UserSid;
import org.springframework.security.acls.domain.AccessControlEntryImpl;
import org.springframework.security.acls.model.AccessControlEntry;
import org.springframework.security.acls.model.MutableAcl;
import org.springframework.security.acls.model.Permission;
import org.springframework.security.acls.model.Sid;
import org.testng.annotations.DataProvider;

import java.util.ArrayList;
import java.util.List;

import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;
import static org.testng.Assert.assertEquals;

/**
 * @author stanislav bashkirtsev
 */
public class AclDataProvider {
    @DataProvider(name = "randomSidsAndPermissions")
    public static Object[][] provideRandomSidsAndPermissionsAndAcl() {
        List<Permission> permissions = Lists.<Permission>newArrayList(BranchPermission.VIEW_TOPICS,
                BranchPermission.CREATE_POSTS, BranchPermission.DELETE_OTHERS_POSTS);
        List sids = (List) provideRandomSids()[0][0];
        return new Object[][] { { sids, permissions } };
    }

    @DataProvider(name = "randomSidsAndPermissionsAndEntity")
    public static Object[][] provideRandomSidsAndPermissionsAndEntity() {
        List<Permission> permissions = Lists.<Permission>newArrayList(BranchPermission.VIEW_TOPICS,
                BranchPermission.CREATE_POSTS, BranchPermission.DELETE_OWN_POSTS);
        List sids = (List) provideRandomSids()[0][0];
        return new Object[][] { { sids, permissions, new EntityImpl(3L) } };
    }

    @DataProvider(name = "randomSids")
    public static Object[][] provideRandomSids() {
        List<Sid> sids = Lists.<Sid>newArrayList(new UserGroupSid(1L), new UserGroupSid(2L), new UserGroupSid(3L));
        return new Object[][] { { sids } };
    }

    @DataProvider(name = "randomEntity")
    public static Object[][] provideRandomEntity() {
        return new Object[][] { { new EntityImpl(2L) } };
    }

    @DataProvider(name = "notSavedEntity")
    public static Object[][] provideNotSavedEntity() {
        return new Object[][] { { new EntityImpl(0L) } };
    }

    @DataProvider
    public static Object[][] provideAclWithMixedTypeSids() {
        List<AccessControlEntry> aces = new ArrayList<AccessControlEntry>();
        ExtendedMutableAcl acl = mock(ExtendedMutableAcl.class);
        when(acl.getEntries()).thenReturn(aces);

        aces.add(new AccessControlEntryImpl(1L, acl, new UserGroupSid(1L), GeneralPermission.READ, true, true,
                true));
        aces.add(new AccessControlEntryImpl(2L, acl, new UserSid(2L), BranchPermission.VIEW_TOPICS, false, true,
                true));
        aces.add(new AccessControlEntryImpl(3L, acl, new UserGroupSid(3L), GeneralPermission.WRITE, true, true,
                true));
        return new Object[][] { { acl } };
    }

    public static List<AccessControlEntry> createEntries(MutableAcl acl, List<Sid> sids,
            List<Permission> permissions) {
        assertEquals(sids.size(), permissions.size(), "Provided lists should have the same size");
        List<AccessControlEntry> toReturn = new ArrayList<AccessControlEntry>(sids.size());
        for (int i = 0; i < sids.size(); i++) {
            toReturn.add(createEntry(acl, sids.get(i), permissions.get(i)));
        }
        return toReturn;
    }

    public static AccessControlEntry createEntry(MutableAcl acl, Sid sid, Permission permission) {
        return new AccessControlEntryImpl(1L, acl, sid, permission, true, true, true);
    }

    private static class EntityImpl extends Entity {
        private EntityImpl(Long id) {
            setId(id);
        }
    }
}