YARN-10647. Fix TestRMNodeLabelsManager failed after YARN-10501. Contributed by

Qi Zhu.

(cherry picked from commit 47420ae3edb9076d3aac7775de8712a7685dc135)
This commit is contained in:
Eric Badger 2021-02-22 19:15:51 +00:00
parent d6797520b9
commit ea73b6ad5c

View File

@ -647,7 +647,9 @@ protected void internalUpdateLabelsOnNodes(
host.labels.addAll(labels);
for (Node node : host.nms.values()) {
replaceNodeForLabels(node.nodeId, node.labels, labels);
replaceLabelsForNode(node.nodeId, node.labels, labels);
if (node.labels != null && labels != null) {
replaceLabelsForNode(node.nodeId, node.labels, labels);
}
node.labels = null;
}
break;