UIAbsSecurityTest.java

/*!
 * Copyright 2010 - 2015 Pentaho Corporation.  All rights reserved.
 *
 * 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.pentaho.di.ui.repository.repositoryexplorer.abs.model;

import org.junit.Assert;
import org.junit.Before;
import org.junit.Test;
import org.pentaho.di.ui.repository.pur.repositoryexplorer.UIEEObjectRegistery;
import org.pentaho.di.ui.repository.pur.repositoryexplorer.abs.model.UIAbsRepositoryRole;
import org.pentaho.di.ui.repository.pur.repositoryexplorer.abs.model.UIAbsSecurity;
import org.pentaho.di.ui.repository.pur.services.IRoleSupportSecurityManager;

public class UIAbsSecurityTest implements java.io.Serializable {

  static final long serialVersionUID = -8052299792314313987L; /* EESOURCE: UPDATE SERIALVERUID */

  IRoleSupportSecurityManager sm;
  public final static String CREATE_CONTENT = "org.pentaho.di.creator"; //$NON-NLS-1$
  public final static String READ_CONTENT = "org.pentaho.di.reader";//$NON-NLS-1$
  public final static String ADMINISTER_SECURITY = "org.pentaho.di.securityAdministrator";//$NON-NLS-1$

  @Before
  public void init() {
    sm = new RepsitoryUserTestImpl();
  }

  @Test
  public void testUIAbsSecurity() throws Exception {
    UIEEObjectRegistery.getInstance().registerUIRepositoryRoleClass( UIAbsRepositoryRole.class );
    UIAbsSecurity security = new UIAbsSecurity( sm );
    security.setSelectedRole( new UIAbsRepositoryRole( sm.getRoles().get( 0 ) ) );
    Assert.assertEquals( ( (UIAbsRepositoryRole) security.getSelectedRole() ).getLogicalRoles().size(), 0 );
    security.addLogicalRole( CREATE_CONTENT );
    Assert.assertEquals( ( (UIAbsRepositoryRole) security.getSelectedRole() ).getLogicalRoles().size(), 1 );
    security.removeLogicalRole( CREATE_CONTENT );
    Assert.assertEquals( ( (UIAbsRepositoryRole) security.getSelectedRole() ).getLogicalRoles().size(), 0 );
  }
}