CapabilityCalculatorTest.java

/**
 * Copyright 2010 JBoss Inc
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.drools.guvnor.server.security;

import java.util.ArrayList;
import java.util.List;
import java.util.HashMap;
import java.util.Map;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertTrue;
import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertSame;

import org.drools.guvnor.client.security.Capabilities;
import org.junit.Test;

public class CapabilityCalculatorTest {

	@Test
	public void testAdmin() {
		CapabilityCalculator loader = new CapabilityCalculator();
		List<RoleBasedPermission> perms = new ArrayList<RoleBasedPermission>();
		perms.add(new RoleBasedPermission("s", RoleTypes.ADMIN, null, null  ));

        Map<String,String> hm = new HashMap<String,String>();
		Capabilities caps = loader.calcCapabilities(perms, hm);
		assertEquals(7, caps.list.size());
        assertSame(hm, caps.prefs);
	}

	@Test
	public void testCapabilitiesCalculate() {
		Map<String,String> hm = new HashMap<String,String>();
		CapabilityCalculator loader = new CapabilityCalculator();
		List<RoleBasedPermission> perms = new ArrayList<RoleBasedPermission>();
		perms.add(new RoleBasedPermission("", RoleTypes.PACKAGE_DEVELOPER, null, null));
		perms.add(new RoleBasedPermission("", RoleTypes.ANALYST, null, null));
		Capabilities caps = loader.calcCapabilities(perms, hm);
		assertTrue(caps.list.contains(Capabilities.SHOW_PACKAGE_VIEW));
        assertSame(hm, caps.prefs);

		perms = new ArrayList<RoleBasedPermission>();
		perms.add(new RoleBasedPermission("", RoleTypes.PACKAGE_ADMIN, null, null));
		caps = loader.calcCapabilities(perms, hm);
		assertTrue(caps.list.contains(Capabilities.SHOW_PACKAGE_VIEW));
        assertSame(hm, caps.prefs);

		perms = new ArrayList<RoleBasedPermission>();
		perms.add(new RoleBasedPermission("", RoleTypes.PACKAGE_READONLY, null, null));
		caps = loader.calcCapabilities(perms, hm);
		assertTrue(caps.list.contains(Capabilities.SHOW_PACKAGE_VIEW));
		assertEquals(1, caps.list.size());
        assertSame(hm, caps.prefs);

		perms = new ArrayList<RoleBasedPermission>();
		perms.add(new RoleBasedPermission("", RoleTypes.PACKAGE_READONLY, null, null));
		perms.add(new RoleBasedPermission("", RoleTypes.PACKAGE_READONLY, null, null));
		perms.add(new RoleBasedPermission("", RoleTypes.ANALYST_READ, null, null));
		perms.add(new RoleBasedPermission("", RoleTypes.PACKAGE_DEVELOPER, null, null));
		caps = loader.calcCapabilities(perms, hm);
        assertSame(hm, caps.prefs);
		assertTrue(caps.list.contains(Capabilities.SHOW_PACKAGE_VIEW));
		assertTrue(caps.list.contains(Capabilities.SHOW_CREATE_NEW_ASSET));
		assertFalse(caps.list.contains(Capabilities.SHOW_CREATE_NEW_PACKAGE));
		assertTrue(caps.list.contains(Capabilities.SHOW_QA));
		assertEquals(3, caps.list.size());

		perms = new ArrayList<RoleBasedPermission>();
		perms.add(new RoleBasedPermission("", RoleTypes.PACKAGE_READONLY, null, null));
		perms.add(new RoleBasedPermission("", RoleTypes.PACKAGE_READONLY, null, null));
		perms.add(new RoleBasedPermission("", RoleTypes.ANALYST, null, null));
		perms.add(new RoleBasedPermission("", RoleTypes.PACKAGE_ADMIN, null, null));
		caps = loader.calcCapabilities(perms, hm);
        assertSame(hm, caps.prefs);
		assertTrue(caps.list.contains(Capabilities.SHOW_PACKAGE_VIEW));
		assertTrue(caps.list.contains(Capabilities.SHOW_CREATE_NEW_ASSET));
		assertTrue(caps.list.contains(Capabilities.SHOW_CREATE_NEW_PACKAGE));
		assertTrue(caps.list.contains(Capabilities.SHOW_DEPLOYMENT));
		assertTrue(caps.list.contains(Capabilities.SHOW_DEPLOYMENT_NEW));
		assertTrue(caps.list.contains(Capabilities.SHOW_QA));

		assertEquals(6, caps.list.size());

		perms = new ArrayList<RoleBasedPermission>();
		perms.add(new RoleBasedPermission("", RoleTypes.PACKAGE_READONLY, null, null));
		perms.add(new RoleBasedPermission("", RoleTypes.PACKAGE_READONLY, null, null));
		perms.add(new RoleBasedPermission("", RoleTypes.ADMIN, null, null));
		perms.add(new RoleBasedPermission("", RoleTypes.PACKAGE_ADMIN, null, null));
		caps = loader.calcCapabilities(perms, hm);
        assertSame(hm, caps.prefs);
		assertEquals(7, caps.list.size());

	}

}