From 9659f28a39fba3529cf202482f5ef442ad730d7e Mon Sep 17 00:00:00 2001
From: teamconfx <team.fuzzing@gmail.com>
Date: Fri, 15 Sep 2023 01:11:54 +0800
Subject: [PATCH] set the protocol engine in testWrappedStopProxy to avoid
 flakiness

---
 .../src/test/java/org/apache/hadoop/ipc/TestRPC.java           | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/hadoop-common-project/hadoop-common/src/test/java/org/apache/hadoop/ipc/TestRPC.java b/hadoop-common-project/hadoop-common/src/test/java/org/apache/hadoop/ipc/TestRPC.java
index 88d9204f69078..295748be2f8ec 100644
--- a/hadoop-common-project/hadoop-common/src/test/java/org/apache/hadoop/ipc/TestRPC.java
+++ b/hadoop-common-project/hadoop-common/src/test/java/org/apache/hadoop/ipc/TestRPC.java
@@ -764,6 +764,9 @@ public void testStopProxy() throws IOException {
 
   @Test
   public void testWrappedStopProxy() throws IOException {
+    RPC.setProtocolEngine(conf,
+        StoppedProtocol.class, StoppedRpcEngine.class);
+
     StoppedProtocol wrappedProxy = RPC.getProxy(StoppedProtocol.class,
         StoppedProtocol.versionID, null, conf);
     StoppedInvocationHandler invocationHandler = (StoppedInvocationHandler)