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

Introduction

Here is the source code for org.jtalks.common.security.acl.JtalksSidFactoryTest.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 org.jtalks.common.model.entity.Entity;
import org.jtalks.common.model.entity.Group;
import org.jtalks.common.model.entity.User;
import org.jtalks.common.security.acl.sids.JtalksSidFactory;
import org.jtalks.common.security.acl.sids.UniversalSid;
import org.jtalks.common.security.acl.sids.UserGroupSid;
import org.jtalks.common.security.acl.sids.UserSid;
import org.springframework.security.acls.domain.GrantedAuthoritySid;
import org.springframework.security.acls.domain.PrincipalSid;
import org.springframework.security.acls.model.Sid;
import org.springframework.security.authentication.UsernamePasswordAuthenticationToken;
import org.testng.annotations.BeforeMethod;
import org.testng.annotations.Test;

import java.util.Arrays;
import java.util.List;

import static org.testng.Assert.assertEquals;
import static org.testng.Assert.assertSame;

/**
 * @author stanislav bashkirtsev
 */
public class JtalksSidFactoryTest {
    private JtalksSidFactory sidFactory;

    @BeforeMethod
    public void setUp() throws Exception {
        sidFactory = new JtalksSidFactory();
    }

    @Test
    public void testCreatePrincipal_anonymousUser() throws Exception {
        String username = "anonymousUser";
        Sid anonymousUser = sidFactory.createPrincipal(new UsernamePasswordAuthenticationToken(username, ""));
        assertSame(anonymousUser, UserSid.createAnonymous());
    }

    @Test
    public void testCreate_customSidByChildEntity() throws Exception {
        Group receiver = new GroupChild();
        receiver.setId(1L);
        UserGroupSid sid = (UserGroupSid) sidFactory.create(receiver);
        assertEquals(sid.getGroupId(), "1");
        assertEquals(sid.getSidId(), "usergroup:1");
    }

    @Test
    public void testCreate_customSidByEntityies() throws Exception {
        Group group = new Group();
        group.setId(1L);
        User user = new User("", "", "", "");
        user.setId(2L);
        List<Entity> receivers = Arrays.asList(group, user);
        List<Sid> sids = sidFactory.create(receivers);
        assertEquals(((UniversalSid) sids.get(0)).getSidId(), "usergroup:1");
        assertEquals(((UniversalSid) sids.get(1)).getSidId(), "user:2");
    }

    @Test
    public void testCreate_customSidByEntityt() throws Exception {
        Group receiver = new Group();
        receiver.setId(1L);
        UserGroupSid sid = (UserGroupSid) sidFactory.create(receiver);
        assertEquals(sid.getGroupId(), "1");
        assertEquals(sid.getSidId(), "usergroup:1");
    }

    @Test
    public void testCreate_customSid() throws Exception {
        UserGroupSid sid = (UserGroupSid) sidFactory.create("usergroup:2", true);
        assertEquals(sid.getGroupId(), "2");
        assertEquals(sid.getSidId(), "usergroup:2");
    }

    @Test
    public void testCreate_principalSid() throws Exception {
        PrincipalSid sid = (PrincipalSid) sidFactory.create("uncle toby", true);
        assertEquals(sid.getPrincipal(), "uncle toby");
    }

    @Test
    public void testCreate_grantedAuthoritySid() throws Exception {
        GrantedAuthoritySid sid = (GrantedAuthoritySid) sidFactory.create("ROLE_ADMIN", false);
        assertEquals(sid.getGrantedAuthority(), "ROLE_ADMIN");
    }

    private static class SidWithoutRequiredConstructor implements UniversalSid {
        @Override
        public String getSidId() {
            return null;
        }

        @Override
        public boolean isPrincipal() {
            return false;
        }
    }

    private static class GroupChild extends Group {

    }
}