Java 类org.apache.hadoop.mapreduce.QueueAclsInfo 实例源码

@Test(expected = YarnQueueAclsException.class)
public void testSubmitWhenUserHasNoPermissionsToSubmitJobInQueueShouldRaiseYarnQueueAclsException() throws IOException, InterruptedException, ClassNotFoundException{
  Mockito.spy( YarnQueueAclsVerifier.class );
  ConfigurationProxyV2 configurationProxyV2 = Mockito.mock( ConfigurationProxyV2.class );
  Cluster cluster = Mockito.mock( Cluster.class );
  Job job = Mockito.mock( Job.class );

  Mockito.when( configurationProxyV2.getJob() ).thenReturn( job );
  Mockito.when( configurationProxyV2.createClusterDescription( Mockito.any( Configuration.class ) ) ).thenReturn( cluster );
  Mockito.when( configurationProxyV2.submit() ).thenCallRealMethod();
  Mockito.when( cluster.getQueueAclsForCurrentUser() ).thenReturn( new QueueAclsInfo[]{
    new QueueAclsInfo( StringUtils.EMPTY, new String[]{
      "ANOTHER_RIGHTS"
    } ),
    new QueueAclsInfo( StringUtils.EMPTY, new String[]{})
  });

  configurationProxyV2.submit();
}

@Test
public void testSubmitWhenUserHasPermissionsToSubmitJobInQueueShouldExecuteSuccessfully() throws IOException, InterruptedException, ClassNotFoundException{
  Mockito.spy( YarnQueueAclsVerifier.class );
  ConfigurationProxyV2 configurationProxyV2 = Mockito.mock( ConfigurationProxyV2.class );
  Cluster cluster = Mockito.mock( Cluster.class );
  Job job = Mockito.mock( Job.class );

  Mockito.when( configurationProxyV2.getJob() ).thenReturn( job );
  Mockito.when( configurationProxyV2.createClusterDescription( Mockito.any( Configuration.class ) ) ).thenReturn( cluster );
  Mockito.when( configurationProxyV2.submit() ).thenCallRealMethod();
  Mockito.when( cluster.getQueueAclsForCurrentUser() ).thenReturn( new QueueAclsInfo[]{
    new QueueAclsInfo( StringUtils.EMPTY, new String[]{
      "SUBMIT_APPLICATIONS"
    } ),
    new QueueAclsInfo( StringUtils.EMPTY, new String[]{})
  });

  Assert.assertNotNull( configurationProxyV2.submit() );
}

public QueueAclsInfo[] getQueueAclsForCurrentUser() throws IOException,
    InterruptedException {
  try {
    return TypeConverter.fromYarnQueueUserAclsInfo(client
      .getQueueAclsInfo());
  } catch (YarnException e) {
    throw new IOException(e);
  }
}

public QueueAclsInfo[] getQueueAclsForCurrentUser() throws IOException,
    InterruptedException {
  try {
    return TypeConverter.fromYarnQueueUserAclsInfo(client
      .getQueueAclsInfo());
  } catch (YarnException e) {
    throw new IOException(e);
  }
}

public QueueAclsInfo[] getQueueAclsForCurrentUser() throws IOException,
    InterruptedException {
  try {
    return TypeConverter.fromYarnQueueUserAclsInfo(client
      .getQueueAclsInfo());
  } catch (YarnException e) {
    throw new IOException(e);
  }
}

public QueueAclsInfo[] getQueueAclsForCurrentUser() throws IOException,
    InterruptedException {
  try {
    return TypeConverter.fromYarnQueueUserAclsInfo(client
      .getQueueAclsInfo());
  } catch (YarnException e) {
    throw new IOException(e);
  }
}

public QueueAclsInfo[] getQueueAclsForCurrentUser() throws IOException,
    InterruptedException {
  try {
    return TypeConverter.fromYarnQueueUserAclsInfo(client
      .getQueueAclsInfo());
  } catch (YarnException e) {
    throw new IOException(e);
  }
}

public QueueAclsInfo[] getQueueAclsForCurrentUser() throws IOException,
    InterruptedException {
  try {
    return TypeConverter.fromYarnQueueUserAclsInfo(client
      .getQueueAclsInfo());
  } catch (YarnException e) {
    throw new IOException(e);
  }
}

public QueueAclsInfo[] getQueueAclsForCurrentUser() throws IOException,
    InterruptedException {
  try {
    return TypeConverter.fromYarnQueueUserAclsInfo(client
      .getQueueAclsInfo());
  } catch (YarnException e) {
    throw new IOException(e);
  }
}

public QueueAclsInfo[] getQueueAclsForCurrentUser() throws IOException,
    InterruptedException {
  try {
    return TypeConverter.fromYarnQueueUserAclsInfo(client
      .getQueueAclsInfo());
  } catch (YarnException e) {
    throw new IOException(e);
  }
}

public QueueAclsInfo[] getQueueAclsForCurrentUser() throws IOException,
    InterruptedException {
  try {
    return TypeConverter.fromYarnQueueUserAclsInfo(client
      .getQueueAclsInfo());
  } catch (YarnException e) {
    throw new IOException(e);
  }
}

public QueueAclsInfo[] getQueueAclsForCurrentUser() throws IOException,
    InterruptedException {
  try {
    return TypeConverter.fromYarnQueueUserAclsInfo(client
      .getQueueAclsInfo());
  } catch (YarnException e) {
    throw new IOException(e);
  }
}

@Test
public void testVerifyWhenUserHasNoPermissionsForSubmitInAnyQueueShouldReturnFalse() throws Exception {
  assertFalse( YarnQueueAclsVerifier.verify( new QueueAclsInfo[] {
    new QueueAclsInfo( StringUtils.EMPTY, new String[] {
      "ANOTHER_RIGHTS"
    } ),
    new QueueAclsInfo( StringUtils.EMPTY, new String[] {} )
  } ) );
}

@Test
public void testVerifyWhenUserHasPermissionsForSubmitInAnyQueueShouldReturnTrue() throws Exception {
  assertTrue( YarnQueueAclsVerifier.verify( new QueueAclsInfo[] {
    new QueueAclsInfo( StringUtils.EMPTY, new String[] {
      "SUBMIT_APPLICATIONS"
    } ),
    new QueueAclsInfo( StringUtils.EMPTY, new String[] {} )
  } ) );
}

public QueueAclsInfo[] getQueueAclsForCurrentUser() throws IOException,
    InterruptedException {
  try {
    return TypeConverter.fromYarnQueueUserAclsInfo(
        client.getQueueAclsInfo());
  } catch (YarnException e) {
    throw new IOException(e);
  }
}

public QueueAclsInfo[] getQueueAclsForCurrentUser() throws IOException,
    InterruptedException {
  try {
    return TypeConverter.fromYarnQueueUserAclsInfo(
        client.getQueueAclsInfo());
  } catch (YarnException e) {
    throw new IOException(e);
  }
}

@Override
public QueueAclsInfo[] getQueueAclsForCurrentUser() throws IOException, InterruptedException {
    return resMgrDelegate.getQueueAclsForCurrentUser();
}

@Override
public QueueAclsInfo[] getQueueAclsForCurrentUser() throws IOException,
    InterruptedException {
  return resMgrDelegate.getQueueAclsForCurrentUser();
}

@Override
public QueueAclsInfo[] getQueueAclsForCurrentUser() throws IOException,
    InterruptedException {
  return resMgrDelegate.getQueueAclsForCurrentUser();
}

@Override
public QueueAclsInfo[] getQueueAclsForCurrentUser() throws IOException,
    InterruptedException {
  return resMgrDelegate.getQueueAclsForCurrentUser();
}

@Override
public QueueAclsInfo[] getQueueAclsForCurrentUser() throws IOException,
    InterruptedException {
  return resMgrDelegate.getQueueAclsForCurrentUser();
}

@Override
public QueueAclsInfo[] getQueueAclsForCurrentUser() throws IOException,
    InterruptedException {
  throw new UnsupportedOperationException();

}

@Override
public QueueAclsInfo[] getQueueAclsForCurrentUser() throws IOException,
    InterruptedException {
  return resMgrDelegate.getQueueAclsForCurrentUser();
}

@Override
public QueueAclsInfo[] getQueueAclsForCurrentUser() throws IOException,
    InterruptedException {
  return resMgrDelegate.getQueueAclsForCurrentUser();
}

/** {@inheritDoc} */
@Override public QueueAclsInfo[] getQueueAclsForCurrentUser() throws IOException, InterruptedException {
    return new QueueAclsInfo[0];
}

@Override
public QueueAclsInfo[] getQueueAclsForCurrentUser() throws IOException,
    InterruptedException {
  return resMgrDelegate.getQueueAclsForCurrentUser();
}

@Override
public QueueAclsInfo[] getQueueAclsForCurrentUser() throws IOException,
    InterruptedException {
  return resMgrDelegate.getQueueAclsForCurrentUser();
}

@Override
public QueueAclsInfo[] getQueueAclsForCurrentUser() throws IOException,
    InterruptedException {
  return resMgrDelegate.getQueueAclsForCurrentUser();
}

@Override
public QueueAclsInfo[] getQueueAclsForCurrentUser() throws IOException,
    InterruptedException {
  return resMgrDelegate.getQueueAclsForCurrentUser();
}

@Override
public QueueAclsInfo[] getQueueAclsForCurrentUser() throws IOException,
    InterruptedException {
  throw new UnsupportedOperationException();

}

public static boolean verify( QueueAclsInfo[] queueAclsInfos ) throws IOException, InterruptedException {
  return queueAclsInfos != null && Arrays.stream( queueAclsInfos ).map( QueueAclsInfo::getOperations )
    .flatMap( Arrays::stream ).anyMatch( Predicate.isEqual( QueueACL.SUBMIT_APPLICATIONS.toString() ) );
}

@Override
public QueueAclsInfo[] getQueueAclsForCurrentUser() throws IOException,
    InterruptedException {
  return resMgrDelegate.getQueueAclsForCurrentUser();
}

@Override
public QueueAclsInfo[] getQueueAclsForCurrentUser() throws IOException,
    InterruptedException {
  throw new UnsupportedOperationException();

}

@Override
public QueueAclsInfo[] getQueueAclsForCurrentUser() throws IOException,
    InterruptedException {
  return resMgrDelegate.getQueueAclsForCurrentUser();
}

/**
 * Gets the Queue ACLs for current user
 * @return array of QueueAclsInfo object for current user.
 * @throws IOException
 */
public QueueAclsInfo[] getQueueAclsForCurrentUser() 
throws IOException, InterruptedException;

/**
 * Gets the Queue ACLs for current user
 * @return array of QueueAclsInfo object for current user.
 * @throws IOException
 */
public QueueAclsInfo[] getQueueAclsForCurrentUser() 
throws IOException, InterruptedException;

/**
 * Gets the Queue ACLs for current user
 * @return array of QueueAclsInfo object for current user.
 * @throws IOException
 */
public QueueAclsInfo[] getQueueAclsForCurrentUser() 
throws IOException, InterruptedException;

/**
 * Gets the Queue ACLs for current user
 * @return array of QueueAclsInfo object for current user.
 * @throws IOException
 */
public QueueAclsInfo[] getQueueAclsForCurrentUser() 
throws IOException, InterruptedException;

/**
 * Gets the Queue ACLs for current user
 * @return array of QueueAclsInfo object for current user.
 * @throws IOException
 */
public QueueAclsInfo[] getQueueAclsForCurrentUser() 
throws IOException, InterruptedException;

/**
 * Gets the Queue ACLs for current user
 * @return array of QueueAclsInfo object for current user.
 * @throws IOException
 */
public QueueAclsInfo[] getQueueAclsForCurrentUser() 
throws IOException, InterruptedException;