import org.apache.hadoop.yarn.server.api.protocolrecords.impl.pb.RegisterNodeManagerResponsePBImpl; //导入依赖的package包/类
/**
 * Test RegisterNodeManagerResponsePBImpl. Test getters and setters. The
 * RegisterNodeManagerResponsePBImpl should generate a prototype and data
 * restore from prototype
 */
@Test
public void testRegisterNodeManagerResponsePBImpl() {
  RegisterNodeManagerResponsePBImpl original =
      new RegisterNodeManagerResponsePBImpl();
  original.setContainerTokenMasterKey(getMasterKey());
  original.setNMTokenMasterKey(getMasterKey());
  original.setNodeAction(NodeAction.NORMAL);
  original.setDiagnosticsMessage("testDiagnosticMessage");

  RegisterNodeManagerResponsePBImpl copy =
      new RegisterNodeManagerResponsePBImpl(
          original.getProto());
  assertEquals(1, copy.getContainerTokenMasterKey().getKeyId());
  assertEquals(1, copy.getNMTokenMasterKey().getKeyId());
  assertEquals(NodeAction.NORMAL, copy.getNodeAction());
  assertEquals("testDiagnosticMessage", copy.getDiagnosticsMessage());

}
 

import org.apache.hadoop.yarn.server.api.protocolrecords.impl.pb.RegisterNodeManagerResponsePBImpl; //导入依赖的package包/类
/**
 * Test RegisterNodeManagerResponsePBImpl. Test getters and setters. The
 * RegisterNodeManagerResponsePBImpl should generate a prototype and data
 * restore from prototype
 */
@Test
public void testRegisterNodeManagerResponsePBImpl() {
  RegisterNodeManagerResponsePBImpl original =
      new RegisterNodeManagerResponsePBImpl();
  original.setContainerTokenMasterKey(getMasterKey());
  original.setNMTokenMasterKey(getMasterKey());
  original.setNodeAction(NodeAction.NORMAL);
  original.setDiagnosticsMessage("testDiagnosticMessage");

  RegisterNodeManagerResponsePBImpl copy =
      new RegisterNodeManagerResponsePBImpl(
          original.getProto());
  assertEquals(1, copy.getContainerTokenMasterKey().getKeyId());
  assertEquals(1, copy.getNMTokenMasterKey().getKeyId());
  assertEquals(NodeAction.NORMAL, copy.getNodeAction());
  assertEquals("testDiagnosticMessage", copy.getDiagnosticsMessage());
  assertFalse(copy.getAreNodeLabelsAcceptedByRM());
}
 

import org.apache.hadoop.yarn.server.api.protocolrecords.impl.pb.RegisterNodeManagerResponsePBImpl; //导入依赖的package包/类
@Override
public RegisterNodeManagerResponse registerNodeManager(
    RegisterNodeManagerRequest request) throws YarnException,
    IOException {
  RegisterNodeManagerRequestProto requestProto = ((RegisterNodeManagerRequestPBImpl)request).getProto();
  try {
    return new RegisterNodeManagerResponsePBImpl(proxy.registerNodeManager(null, requestProto));
  } catch (ServiceException e) {
    RPCUtil.unwrapAndThrowException(e);
    return null;
  }
}
 

import org.apache.hadoop.yarn.server.api.protocolrecords.impl.pb.RegisterNodeManagerResponsePBImpl; //导入依赖的package包/类
public static RegisterNodeManagerResponse serDe(RegisterNodeManagerResponse orig) throws Exception {
  RegisterNodeManagerResponsePBImpl asPB = (RegisterNodeManagerResponsePBImpl)orig;
  RegisterNodeManagerResponseProto proto = asPB.getProto();
  ByteArrayOutputStream out = new ByteArrayOutputStream();
  proto.writeTo(out);
  ByteArrayInputStream in = new ByteArrayInputStream(out.toByteArray());
  RegisterNodeManagerResponseProto.Builder cp = RegisterNodeManagerResponseProto.newBuilder();
  cp.mergeFrom(in);
  return new RegisterNodeManagerResponsePBImpl(cp.build());
}
 

import org.apache.hadoop.yarn.server.api.protocolrecords.impl.pb.RegisterNodeManagerResponsePBImpl; //导入依赖的package包/类
@Override
public RegisterNodeManagerResponseProto registerNodeManager(
    RpcController controller, RegisterNodeManagerRequestProto proto)
    throws ServiceException {
  RegisterNodeManagerRequestPBImpl request = new RegisterNodeManagerRequestPBImpl(proto);
  try {
    RegisterNodeManagerResponse response = real.registerNodeManager(request);
    return ((RegisterNodeManagerResponsePBImpl)response).getProto();
  } catch (YarnException | IOException e) {
    throw new ServiceException(e);
  }
}
 

import org.apache.hadoop.yarn.server.api.protocolrecords.impl.pb.RegisterNodeManagerResponsePBImpl; //导入依赖的package包/类
@Test
public void testRegisterNodeManagerResponsePBImplWithRMAcceptLbls() {
  RegisterNodeManagerResponsePBImpl original =
      new RegisterNodeManagerResponsePBImpl();
  original.setAreNodeLabelsAcceptedByRM(true);
  RegisterNodeManagerResponsePBImpl copy =
      new RegisterNodeManagerResponsePBImpl(original.getProto());
  assertTrue(copy.getAreNodeLabelsAcceptedByRM());
}