This is an automated email from the ASF dual-hosted git repository.
dinglei pushed a commit to branch develop
in repository https://gitbox.apache.org/repos/asf/rocketmq.git
The following commit(s) were added to refs/heads/develop by this push:
new e2885acc52 [ISSUE #8553] Add UnitTest of OffsetSerialize (#8554)
e2885acc52 is described below
commit e2885acc52d4eea720e7aedc18067c9aef5c3425
Author: syhleo <[email protected]>
AuthorDate: Wed Aug 21 10:57:59 2024 +0800
[ISSUE #8553] Add UnitTest of OffsetSerialize (#8554)
* [ISSUE #8553] Add UnitTest of ZoneRouteRPCHookćOffsetSerialize
---
.../offset/RocksDBOffsetSerializeWrapperTest.java | 53 +++++++
.../namesrv/route/ZoneRouteRPCHookMoreTest.java | 154 +++++++++++++++++++++
2 files changed, 207 insertions(+)
diff --git
a/broker/src/test/java/org/apache/rocketmq/broker/offset/RocksDBOffsetSerializeWrapperTest.java
b/broker/src/test/java/org/apache/rocketmq/broker/offset/RocksDBOffsetSerializeWrapperTest.java
new file mode 100644
index 0000000000..dde0401e8a
--- /dev/null
+++
b/broker/src/test/java/org/apache/rocketmq/broker/offset/RocksDBOffsetSerializeWrapperTest.java
@@ -0,0 +1,53 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements. See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.apache.rocketmq.broker.offset;
+
+
+import java.util.concurrent.ConcurrentHashMap;
+import java.util.concurrent.ConcurrentMap;
+
+import org.junit.Before;
+import org.junit.Test;
+
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertNotNull;
+
+public class RocksDBOffsetSerializeWrapperTest {
+
+ private RocksDBOffsetSerializeWrapper wrapper;
+
+ @Before
+ public void setUp() {
+ wrapper = new RocksDBOffsetSerializeWrapper();
+ }
+
+ @Test
+ public void testGetOffsetTable_ShouldReturnConcurrentHashMap() {
+ ConcurrentMap<Integer, Long> offsetTable = wrapper.getOffsetTable();
+ assertNotNull("The offsetTable should not be null", offsetTable);
+ }
+
+ @Test
+ public void testSetOffsetTable_ShouldSetTheOffsetTableCorrectly() {
+ ConcurrentMap<Integer, Long> newOffsetTable = new
ConcurrentHashMap<>();
+ wrapper.setOffsetTable(newOffsetTable);
+ ConcurrentMap<Integer, Long> offsetTable = wrapper.getOffsetTable();
+ assertEquals("The offsetTable should be the same as the one set",
newOffsetTable, offsetTable);
+ }
+
+}
diff --git
a/namesrv/src/test/java/org/apache/rocketmq/namesrv/route/ZoneRouteRPCHookMoreTest.java
b/namesrv/src/test/java/org/apache/rocketmq/namesrv/route/ZoneRouteRPCHookMoreTest.java
new file mode 100644
index 0000000000..ed42becdf5
--- /dev/null
+++
b/namesrv/src/test/java/org/apache/rocketmq/namesrv/route/ZoneRouteRPCHookMoreTest.java
@@ -0,0 +1,154 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements. See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.apache.rocketmq.namesrv.route;
+
+
+import org.apache.rocketmq.common.MixAll;
+import org.apache.rocketmq.remoting.protocol.RemotingCommand;
+import org.apache.rocketmq.remoting.protocol.RemotingSerializable;
+import org.apache.rocketmq.remoting.protocol.RequestCode;
+import org.apache.rocketmq.remoting.protocol.ResponseCode;
+import org.apache.rocketmq.remoting.protocol.route.BrokerData;
+import org.apache.rocketmq.remoting.protocol.route.QueueData;
+import org.apache.rocketmq.remoting.protocol.route.TopicRouteData;
+import org.junit.Before;
+import org.junit.Test;
+
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertNull;
+
+public class ZoneRouteRPCHookMoreTest {
+
+ private ZoneRouteRPCHook zoneRouteRPCHook;
+
+ @Before
+ public void setUp() {
+ zoneRouteRPCHook = new ZoneRouteRPCHook();
+ }
+
+ @Test
+ public void testFilterByZoneName_ValidInput_ShouldFilterCorrectly() {
+ // Arrange
+ TopicRouteData topicRouteData = new TopicRouteData();
+ topicRouteData.setBrokerDatas(generateBrokerDataList());
+ topicRouteData.setQueueDatas(generateQueueDataList());
+
+ RemotingCommand request =
RemotingCommand.createRequestCommand(RequestCode.GET_ROUTEINFO_BY_TOPIC,null);
+ request.setExtFields(createExtFields("true","ZoneA"));
+
+ RemotingCommand response =
RemotingCommand.createResponseCommand(ResponseCode.SUCCESS, "remark");
+
+ // Act
+ zoneRouteRPCHook.doAfterResponse("127.0.0.1", request, response);
+ TopicRouteData decodedResponse =
RemotingSerializable.decode(response.getBody(), TopicRouteData.class);
+
+ // Assert
+ assertNull(decodedResponse);
+ }
+
+ @Test
+ public void testFilterByZoneName_NoZoneName_ShouldNotFilter() {
+ // Arrange
+ TopicRouteData topicRouteData = new TopicRouteData();
+ topicRouteData.setBrokerDatas(generateBrokerDataList());
+ topicRouteData.setQueueDatas(generateQueueDataList());
+
+ RemotingCommand request =
RemotingCommand.createRequestCommand(RequestCode.GET_ROUTEINFO_BY_TOPIC,null);
+ HashMap<String, String> extFields = new HashMap<>();
+ extFields.put(MixAll.ZONE_MODE, "true");
+ request.setExtFields(extFields);
+
+ RemotingCommand response =
RemotingCommand.createResponseCommand(ResponseCode.SUCCESS, null);
+
+ // Act
+ zoneRouteRPCHook.doAfterResponse("127.0.0.1", request, response);
+ TopicRouteData decodedResponse =
RemotingSerializable.decode(response.getBody(), TopicRouteData.class);
+
+ // Assert
+ assertEquals(topicRouteData.getBrokerDatas().size(), 2);
+ assertEquals(topicRouteData.getQueueDatas().size(), 2);
+ }
+
+ @Test
+ public void testFilterByZoneName_ZoneModeFalse_ShouldNotFilter() {
+ // Arrange
+ TopicRouteData topicRouteData = new TopicRouteData();
+ topicRouteData.setBrokerDatas(generateBrokerDataList());
+ topicRouteData.setQueueDatas(generateQueueDataList());
+
+ RemotingCommand request =
RemotingCommand.createRequestCommand(RequestCode.GET_ROUTEINFO_BY_TOPIC,null);
+ request.setExtFields(createExtFields("false","ZoneA"));
+
+ RemotingCommand response =
RemotingCommand.createResponseCommand(ResponseCode.SUCCESS ,null);
+
+ // Act
+ zoneRouteRPCHook.doAfterResponse("127.0.0.1", request, response);
+ TopicRouteData decodedResponse =
RemotingSerializable.decode(response.getBody(), TopicRouteData.class);
+
+ // Assert
+ assertEquals(topicRouteData.getBrokerDatas().size(), 2);
+ assertEquals(topicRouteData.getQueueDatas().size(), 2);
+ }
+
+ private List<BrokerData> generateBrokerDataList() {
+ List<BrokerData> brokerDataList = new ArrayList<>();
+ BrokerData brokerData1 = new BrokerData();
+ brokerData1.setBrokerName("BrokerA");
+ brokerData1.setZoneName("ZoneA");
+ Map<Long, String> brokerAddrs = new HashMap<>();
+ brokerAddrs.put(MixAll.MASTER_ID, "127.0.0.1:10911");
+ brokerData1.setBrokerAddrs((HashMap<Long, String>) brokerAddrs);
+ brokerDataList.add(brokerData1);
+
+ BrokerData brokerData2 = new BrokerData();
+ brokerData2.setBrokerName("BrokerB");
+ brokerData2.setZoneName("ZoneB");
+ brokerAddrs = new HashMap<>();
+ brokerAddrs.put(MixAll.MASTER_ID, "127.0.0.1:10912");
+ brokerData2.setBrokerAddrs((HashMap<Long, String>) brokerAddrs);
+ brokerDataList.add(brokerData2);
+
+ return brokerDataList;
+ }
+
+ private List<QueueData> generateQueueDataList() {
+ List<QueueData> queueDataList = new ArrayList<>();
+ QueueData queueData1 = new QueueData();
+ queueData1.setBrokerName("BrokerA");
+ queueDataList.add(queueData1);
+
+ QueueData queueData2 = new QueueData();
+ queueData2.setBrokerName("BrokerB");
+ queueDataList.add(queueData2);
+
+ return queueDataList;
+ }
+
+ private HashMap<String, String> createExtFields(String zoneMode, String
zoneName) {
+ HashMap<String, String> extFields = new HashMap<>();
+ extFields.put(MixAll.ZONE_MODE, zoneMode);
+ extFields.put(MixAll.ZONE_NAME, zoneName);
+ return extFields;
+ }
+
+}
\ No newline at end of file
