Apache
/
hadoop
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

HADOOP-18511. Fix flaky test TestDecayRpcScheduler.testPriority

This commit is contained in:
Kerr Zhang 2022-10-27 14:21:50 -05:00
parent 8396caa484
commit dbde211e28
1 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
hadoop-common-project/hadoop-common/src/test/java/org/apache/hadoop/ipc/TestDecayRpcScheduler.java
View File

@ -39,11 +39,14 @@ import org.apache.hadoop.conf.Configuration;
import javax.management.MBeanServer; import javax.management.MBeanServer;
import javax.management.ObjectName; import javax.management.ObjectName;
import com.fasterxml.jackson.databind.ObjectMapper;
import java.io.ByteArrayOutputStream; import java.io.ByteArrayOutputStream;
import java.io.PrintStream; import java.io.PrintStream;
import java.lang.management.ManagementFactory; import java.lang.management.ManagementFactory;
import java.util.concurrent.TimeUnit; import java.util.concurrent.TimeUnit;
import java.util.TreeMap;
import java.util.SortedMap;
public class TestDecayRpcScheduler { public class TestDecayRpcScheduler {
private Schedulable mockCall(String id) { private Schedulable mockCall(String id) {
@ -291,13 +294,18 @@ public class TestDecayRpcScheduler {
ObjectName mxbeanName = new ObjectName( ObjectName mxbeanName = new ObjectName(
"Hadoop:service="+ namespace + ",name=DecayRpcScheduler"); "Hadoop:service="+ namespace + ",name=DecayRpcScheduler");
String cvs1 = (String) mbs.getAttribute(mxbeanName, "CallVolumeSummary"); ObjectMapper mapper = new ObjectMapper();
String json1 = (String) mbs.getAttribute(mxbeanName, "CallVolumeSummary");
SortedMap<String, Integer> map1 = mapper.readValue(json1, TreeMap.class);
String cvs1 = mapper.writeValueAsString(map1);
assertTrue("Get expected JMX of CallVolumeSummary before decay", assertTrue("Get expected JMX of CallVolumeSummary before decay",
cvs1.equals("{\"A\":6,\"B\":2,\"C\":2}")); cvs1.equals("{\"A\":6,\"B\":2,\"C\":2}"));
scheduler.forceDecay(); scheduler.forceDecay();
String cvs2 = (String) mbs.getAttribute(mxbeanName, "CallVolumeSummary"); String json2 = (String) mbs.getAttribute(mxbeanName, "CallVolumeSummary");
SortedMap<String, Integer> map2 = mapper.readValue(json2, TreeMap.class);
String cvs2 = mapper.writeValueAsString(map2);
assertTrue("Get expected JMX for CallVolumeSummary after decay", assertTrue("Get expected JMX for CallVolumeSummary after decay",
cvs2.equals("{\"A\":3,\"B\":1,\"C\":1}")); cvs2.equals("{\"A\":3,\"B\":1,\"C\":1}"));
} }