import org.apache.hadoop.yarn.server.nodemanager.NMAuditLogger.Keys; //导入依赖的package包/类
/**
 * Test the AuditLog format with key-val pair.
 */
@Test  
public void testKeyValLogFormat() throws Exception {
  StringBuilder actLog = new StringBuilder();
  StringBuilder expLog = new StringBuilder();
  // add the first k=v pair and check
  NMAuditLogger.start(Keys.USER, USER, actLog);
  expLog.append("USER=test");
  assertEquals(expLog.toString(), actLog.toString());

  // append another k1=v1 pair to already added k=v and test
  NMAuditLogger.add(Keys.OPERATION, OPERATION, actLog);
  expLog.append("\tOPERATION=oper");
  assertEquals(expLog.toString(), actLog.toString());

  // append another k1=null pair and test
  NMAuditLogger.add(Keys.APPID, (String)null, actLog);
  expLog.append("\tAPPID=null");
  assertEquals(expLog.toString(), actLog.toString());

  // now add the target and check of the final string
  NMAuditLogger.add(Keys.TARGET, TARGET, actLog);
  expLog.append("\tTARGET=tgt");
  assertEquals(expLog.toString(), actLog.toString());
}
 

import org.apache.hadoop.yarn.server.nodemanager.NMAuditLogger.Keys; //导入依赖的package包/类
/**
 * Test the AuditLog format for successful events.
 */
private void testSuccessLogFormatHelper(boolean checkIP, 
    ApplicationId appId, ContainerId containerId) {
  // check without the IP
  String sLog = NMAuditLogger.createSuccessLog(USER, OPERATION, TARGET,
      appId, containerId);
  StringBuilder expLog = new StringBuilder();
  expLog.append("USER=test\t");
  if (checkIP) {
    InetAddress ip = Server.getRemoteIp();
    expLog.append(Keys.IP.name() + "=" + ip.getHostAddress() + "\t");
  }
  expLog.append("OPERATION=oper\tTARGET=tgt\tRESULT=SUCCESS");
  if (appId != null) {
    expLog.append("\tAPPID=app_1");
  }
  if (containerId != null) {
    expLog.append("\tCONTAINERID=container_1");
  }
  assertEquals(expLog.toString(), sLog);
}
 

import org.apache.hadoop.yarn.server.nodemanager.NMAuditLogger.Keys; //导入依赖的package包/类
/**
 * Test the AuditLog format for failure events.
 */
private void testFailureLogFormatHelper(boolean checkIP, ApplicationId appId,
    ContainerId containerId) {
  String fLog =
    NMAuditLogger.createFailureLog(USER, OPERATION, TARGET, DESC, appId,
    containerId);
  StringBuilder expLog = new StringBuilder();
  expLog.append("USER=test\t");
  if (checkIP) {
    InetAddress ip = Server.getRemoteIp();
    expLog.append(Keys.IP.name() + "=" + ip.getHostAddress() + "\t");
  }
  expLog.append("OPERATION=oper\tTARGET=tgt\tRESULT=FAILURE\t");
  expLog.append("DESCRIPTION=description of an audit log");

  if (appId != null) {
    expLog.append("\tAPPID=app_1");
  }
  if (containerId != null) {
    expLog.append("\tCONTAINERID=container_1");
  }
  assertEquals(expLog.toString(), fLog);
}
 

import org.apache.hadoop.yarn.server.nodemanager.NMAuditLogger.Keys; //导入依赖的package包/类
/**
 * Test the AuditLog format for successful events passing nulls.
 */
private void testSuccessLogNulls(boolean checkIP) {
  String sLog = NMAuditLogger.createSuccessLog(null, null, null,
      null, null);
  StringBuilder expLog = new StringBuilder();
  expLog.append("USER=null\t");
  if (checkIP) {
    InetAddress ip = Server.getRemoteIp();
    expLog.append(Keys.IP.name() + "=" + ip.getHostAddress() + "\t");
  }
  expLog.append("OPERATION=null\tTARGET=null\tRESULT=SUCCESS");
  assertEquals(expLog.toString(), sLog);
}