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

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


        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


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

      newUser.setUserName(username);

      String roleName = (configuration.getConfigsMap().get(Configuration.USER_ROLE_NAME_KEY));
      log.info("Using default role name " + roleName);

      RoleEntity role = roleDAO.findByName(roleName);

      if (role == null) {
        log.info("Role " + roleName + " not present in local DB - creating");
        role = new RoleEntity();
        role.setRoleName(roleName);
        roleDAO.create(role);
        role = roleDAO.findByName(role.getRoleName());
      }

      userDAO.create(newUser);

      user = userDAO.findLdapUserByName(newUser.getUserName());

      user.getRoleEntities().add(role);
      role.getUserEntities().add(user);
      roleDAO.merge(role);
      userDAO.merge(user);
    }

    return authorizationHelper.convertRolesToAuthorities(user.getRoleEntities());
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

    UserEntity userEntity = userDAO.findByPK(user.getUserId());
    if (userEntity == null) {
      throw new AmbariException("User " + user + " doesn't exist");
    }

    RoleEntity roleEntity = roleDAO.findByName(role);
    if (roleEntity == null) {
      LOG.warn("Trying to add user to non-existent role"
          + ", user=" + user.getUserName()
          + ", role=" + role);
      throw new AmbariException("Role " + role + " doesn't exist");
    }

    if (!userEntity.getRoleEntities().contains(roleEntity)) {
      userEntity.getRoleEntities().add(roleEntity);
      roleEntity.getUserEntities().add(userEntity);
      userDAO.merge(userEntity);
      roleDAO.merge(roleEntity);
    } else {
      throw new AmbariException("User " + user + " already owns role " + role);
    }
View Full Code Here

    UserEntity userEntity = userDAO.findByPK(user.getUserId());
    if (userEntity == null) {
      throw new AmbariException("User " + user + " doesn't exist");
    }

    RoleEntity roleEntity = roleDAO.findByName(role);
    if (roleEntity == null) {
      throw new AmbariException("Role " + role + " doesn't exist");
    }

    if (userEntity.getRoleEntities().contains(roleEntity)) {
      userEntity.getRoleEntities().remove(roleEntity);
      roleEntity.getUserEntities().remove(userEntity);
      userDAO.merge(userEntity);
      roleDAO.merge(roleEntity);
    } else {
      throw new AmbariException("User " + user + " doesn't own role " + role);
    }
View Full Code Here

  /**
   * Creates new role
   */
  public void createRole(String role) {
    RoleEntity roleEntity = new RoleEntity();
    roleEntity.setRoleName(role);
    roleDAO.create(roleEntity);
  }
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

   * @param roleName - the role to add to user's roleEntities
   */
  private void addRole(UserEntity user, String roleName) {
    log.info("Using default role name " + roleName);

    RoleEntity roleEntity = roleDAO.findByName(roleName);

    if (roleEntity == null) {
      log.info("Role " + roleName + " not present in local DB - creating");
      roleEntity = new RoleEntity();
      roleEntity.setRoleName(roleName);
      roleDAO.create(roleEntity);
      roleEntity = roleDAO.findByName(roleEntity.getRoleName());
    }

    UserEntity userEntity = userDAO.findLdapUserByName(user.getUserName());
    if (userEntity == null) {
      userDAO.create(user);
      userEntity = userDAO.findLdapUserByName(user.getUserName());
    }

    if (!userEntity.getRoleEntities().contains(roleEntity)) {
      userEntity.getRoleEntities().add(roleEntity);
      roleEntity.getUserEntities().add(userEntity);
      roleDAO.merge(roleEntity);
      userDAO.merge(userEntity);
    }
  }
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.