Package org.apache.ambari.server.orm.entities

Examples of org.apache.ambari.server.orm.entities.RoleEntity


  /**
   * Creates new role
   */
  public void createRole(String role) {
    RoleEntity roleEntity = new RoleEntity();
    roleEntity.setRoleName(role);
    roleDAO.create(roleEntity);
  }
View Full Code Here


    UserEntity userEntity = new UserEntity();
    userEntity.setUserName(userName);
    userEntity.setUserPassword(passwordEncoder.encode(password));
    userEntity.setRoleEntities(new HashSet<RoleEntity>());

    RoleEntity roleEntity = roleDAO.findByName(getUserRole());
    if (roleEntity == null) {
      createRole(getUserRole());
    }
    roleEntity = roleDAO.findByName(getUserRole());

    userEntity.getRoleEntities().add(roleEntity);
    userDAO.create(userEntity);

    roleEntity.getUserEntities().add(userEntity);
    roleDAO.merge(roleEntity);
  }
View Full Code Here

        Collections.singleton(new UserRequest(null)));

    Assert.assertEquals(0, responses.size());

    RoleDAO roleDao = injector.getInstance(RoleDAO.class);
    RoleEntity re1 = roleDao.findByName("user");
    RoleEntity re2 = roleDao.findByName("admin");
    Assert.assertNotNull(re1);
    Assert.assertNotNull(re2);
  }
View Full Code Here


  }

  private void createLdapUser() {
    RoleEntity role = roleDAO.findByName(users.getUserRole());
    UserEntity ldapUser = new UserEntity();

    ldapUser.setUserName("ldapUser");
    ldapUser.setLdapUser(true);

    userDAO.create(ldapUser);

    UserEntity userEntity = userDAO.findLdapUserByName("ldapUser");

    userEntity.getRoleEntities().add(role);
    role.getUserEntities().add(ldapUser);

    userDAO.merge(ldapUser);
    roleDAO.merge(role);
  }
View Full Code Here

        new UsernamePasswordAuthenticationToken("allowedUser", "password");
    authenticationProvider.authenticate(authentication);
    UserEntity allowedUserEntity = userDAO.findLdapUserByName("allowedUser");


    RoleEntity adminRole = roleDAO.findByName(
        configuration.getConfigsMap().get(Configuration.ADMIN_ROLE_NAME_KEY));
    RoleEntity userRole = roleDAO.findByName(
        configuration.getConfigsMap().get(Configuration.USER_ROLE_NAME_KEY));


    assertTrue(allowedAdminEntity.getRoleEntities().contains(userRole));
    assertTrue(allowedAdminEntity.getRoleEntities().contains(adminRole));
View Full Code Here

  @Test
  public void testAddRole() throws Exception {
    AmbariLdapAuthoritiesPopulator populator =
        new AmbariLdapAuthoritiesPopulator(configuration, helper, userDAO, roleDAO);

    RoleEntity roleEntity = createMock(RoleEntity.class);
    Set<UserEntity> userEntities = createMock(Set.class);
    Set<RoleEntity> roleEntities = createMock(Set.class);

    Capture<RoleEntity> createdRole = new Capture<RoleEntity>();

    expect(roleDAO.findByName(adminRole)).andReturn(null).andReturn(roleEntity);
    expect(roleDAO.findByName(adminRole)).andReturn(roleEntity);

    roleDAO.create(capture(createdRole));
    expectLastCall();

    expect(userEntity.getUserName()).andReturn(username).anyTimes();
    expect(userEntity.getRoleEntities()).andReturn(roleEntities).anyTimes();

    expect(roleEntity.getUserEntities()).andReturn(userEntities).anyTimes();

    expect(roleEntities.contains(roleEntity)).andReturn(false);
    expect(roleEntities.contains(roleEntity)).andReturn(true);

    expect(userEntities.add(userEntity)).andReturn(true);
View Full Code Here

    int userId = 123;

    AmbariLdapAuthoritiesPopulator populator =
        new AmbariLdapAuthoritiesPopulator(configuration, helper, userDAO, roleDAO);

    RoleEntity roleEntity = createMock(RoleEntity.class);
    Set<UserEntity> userEntities = createMock(Set.class);
    Set<RoleEntity> roleEntities = createMock(Set.class);

    expect(userEntity.getUserId()).andReturn(userId);

    expect(userDAO.findByPK(userId)).andReturn(userEntity);

    expect(roleDAO.findByName(adminRole)).andReturn(roleEntity);

    expect(userEntity.getRoleEntities()).andReturn(roleEntities);

    expect(roleEntities.contains(roleEntity)).andReturn(true);

    expect(userEntity.getUserName()).andReturn(username);

    expect(userEntity.getRoleEntities()).andReturn(roleEntities);
    expect(roleEntity.getUserEntities()).andReturn(userEntities);

    expect(userEntities.remove(userEntity)).andReturn(true);
    expect(roleEntities.remove(roleEntity)).andReturn(true);

    expect(userDAO.merge(userEntity)).andReturn(userEntity);
View Full Code Here

        new UsernamePasswordAuthenticationToken("the allowedUser", "password");
    authenticationProvider.authenticate(authentication);
    UserEntity allowedUserEntity = userDAO.findLdapUserByName("the allowedUser");


    RoleEntity adminRole = roleDAO.findByName(
        configuration.getConfigsMap().get(Configuration.ADMIN_ROLE_NAME_KEY));
    RoleEntity userRole = roleDAO.findByName(
        configuration.getConfigsMap().get(Configuration.USER_ROLE_NAME_KEY));


    assertTrue(allowedAdminEntity.getRoleEntities().contains(userRole));
    assertTrue(allowedAdminEntity.getRoleEntities().contains(adminRole));
View Full Code Here

public class AuthorizationHelperTest {

  @Test
  public void testConvertRolesToAuthorities() throws Exception {
    Collection<RoleEntity> roles = new ArrayList<RoleEntity>();
    RoleEntity role = new RoleEntity();
    role.setRoleName("admin");
    roles.add(role);
    role = new RoleEntity();
    role.setRoleName("user");
    roles.add(role);

    Collection<GrantedAuthority> authorities = new AuthorizationHelper().convertRolesToAuthorities(roles);

    assertEquals("Wrong number of authorities", 2, authorities.size());
View Full Code Here

        Collections.singleton(new UserRequest(null)));

    Assert.assertEquals(0, responses.size());

    RoleDAO roleDao = injector.getInstance(RoleDAO.class);
    RoleEntity re1 = roleDao.findByName("user");
    RoleEntity re2 = roleDao.findByName("admin");
    Assert.assertNotNull(re1);
    Assert.assertNotNull(re2);
  }
View Full Code Here

TOP

Related Classes of org.apache.ambari.server.orm.entities.RoleEntity

Copyright © 2018 www.massapicom. 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.