Addressed the TestContainerStateMachineStream changes

sarvekshayr · Feb 7, 2024 · 3804fdf · 3804fdf
1 parent 312b36c
commit 3804fdf
Showing 1 changed file with 11 additions and 8 deletions.
diff --git a/...est/src/test/java/org/apache/hadoop/ozone/client/rpc/TestContainerStateMachineStream.java b/...est/src/test/java/org/apache/hadoop/ozone/client/rpc/TestContainerStateMachineStream.java
@@ -17,14 +17,15 @@
 
 package org.apache.hadoop.ozone.client.rpc;
 
-import org.apache.hadoop.conf.StorageUnit;
 import org.apache.hadoop.hdds.client.ReplicationType;
 import org.apache.hadoop.hdds.conf.DatanodeRatisServerConfig;
 import org.apache.hadoop.hdds.conf.OzoneConfiguration;
+import org.apache.hadoop.hdds.conf.StorageUnit;
 import org.apache.hadoop.hdds.protocol.proto.HddsProtos;
 import org.apache.hadoop.hdds.ratis.conf.RatisClientConfig;
 import org.apache.hadoop.hdds.scm.OzoneClientConfig;
 import org.apache.hadoop.hdds.utils.IOUtils;
+import org.apache.hadoop.ozone.ClientConfigForTesting;
 import org.apache.hadoop.ozone.HddsDatanodeService;
 import org.apache.hadoop.ozone.MiniOzoneCluster;
 import org.apache.hadoop.ozone.OzoneConfigKeys;
@@ -112,18 +113,20 @@ public void setup() throws Exception {
     raftClientConfig.setRpcWatchRequestTimeout(Duration.ofSeconds(10));
     conf.setFromObject(raftClientConfig);
 
-    conf.setLong(OzoneConfigKeys.DFS_RATIS_SNAPSHOT_THRESHOLD_KEY, 1);
-    conf.setQuietMode(false);
-    cluster =
-        MiniOzoneCluster.newBuilder(conf)
-            .setNumDatanodes(3)
-            .setHbInterval(200)
+    ClientConfigForTesting.newBuilder(StorageUnit.BYTES)
             .setDataStreamMinPacketSize(1024)
             .setBlockSize(BLOCK_SIZE)
             .setChunkSize(CHUNK_SIZE)
             .setStreamBufferFlushSize(FLUSH_SIZE)
             .setStreamBufferMaxSize(MAX_FLUSH_SIZE)
-            .setStreamBufferSizeUnit(StorageUnit.BYTES)
+            .applyTo(conf);
+
+    conf.setLong(OzoneConfigKeys.OZONE_RATIS_SNAPSHOT_THRESHOLD_KEY, 1);
+    conf.setQuietMode(false);
+    cluster =
+        MiniOzoneCluster.newBuilder(conf)
+            .setNumDatanodes(3)
+            .setHbInterval(200)
             .build();
     cluster.waitForClusterToBeReady();
     cluster.waitForPipelineTobeReady(HddsProtos.ReplicationFactor.ONE, 60000);