[ROCKETMQ-77] [TEST] org.apache.rocketmq.tools.* have NPEs
Project: http://git-wip-us.apache.org/repos/asf/incubator-rocketmq/repo Commit: http://git-wip-us.apache.org/repos/asf/incubator-rocketmq/commit/671e9a55 Tree: http://git-wip-us.apache.org/repos/asf/incubator-rocketmq/tree/671e9a55 Diff: http://git-wip-us.apache.org/repos/asf/incubator-rocketmq/diff/671e9a55 Branch: refs/heads/master Commit: 671e9a5537d941721e76efb18114e8054c9cc936 Parents: ef71e62 Author: stevenschew <[email protected]> Authored: Mon Feb 6 16:15:09 2017 +0800 Committer: stevenschew <[email protected]> Committed: Mon Feb 6 16:15:09 2017 +0800 ---------------------------------------------------------------------- .../broker/BrokerConsumeStatsSubCommadTest.java | 92 ------------- .../broker/BrokerStatusSubCommandTest.java | 88 ------------- .../broker/CleanExpiredCQSubCommandTest.java | 84 ------------ .../broker/CleanUnusedTopicCommandTest.java | 84 ------------ .../broker/GetBrokerConfigCommandTest.java | 89 ------------- .../broker/SendMsgStatusCommandTest.java | 78 ----------- .../UpdateBrokerConfigSubCommandTest.java | 79 ----------- .../ConsumerConnectionSubCommandTest.java | 99 -------------- .../ProducerConnectionSubCommandTest.java | 93 ------------- .../ConsumerProgressSubCommandTest.java | 111 ---------------- .../consumer/ConsumerStatusSubCommandTest.java | 131 ------------------- .../namesrv/GetNamesrvConfigCommandTest.java | 91 ------------- .../namesrv/WipeWritePermSubCommandTest.java | 90 ------------- .../offset/GetConsumerStatusCommandTest.java | 85 ------------ .../offset/ResetOffsetByTimeCommandTest.java | 105 --------------- .../offset/ResetOffsetByTimeOldCommandTest.java | 39 ------ .../command/topic/AllocateMQSubCommandTest.java | 38 ------ .../topic/DeleteTopicSubCommandTest.java | 38 ------ .../topic/TopicClusterSubCommandTest.java | 37 ------ .../command/topic/TopicRouteSubCommandTest.java | 37 ------ .../topic/TopicStatusSubCommandTest.java | 37 ------ .../topic/UpdateOrderConfCommandTest.java | 39 ------ .../topic/UpdateTopicPermSubCommandTest.java | 41 ------ .../topic/UpdateTopicSubCommandTest.java | 54 -------- 24 files changed, 1759 deletions(-) ---------------------------------------------------------------------- http://git-wip-us.apache.org/repos/asf/incubator-rocketmq/blob/671e9a55/tools/src/test/java/org/apache/rocketmq/tools/command/broker/BrokerConsumeStatsSubCommadTest.java ---------------------------------------------------------------------- diff --git a/tools/src/test/java/org/apache/rocketmq/tools/command/broker/BrokerConsumeStatsSubCommadTest.java b/tools/src/test/java/org/apache/rocketmq/tools/command/broker/BrokerConsumeStatsSubCommadTest.java deleted file mode 100644 index 3523175..0000000 --- a/tools/src/test/java/org/apache/rocketmq/tools/command/broker/BrokerConsumeStatsSubCommadTest.java +++ /dev/null @@ -1,92 +0,0 @@ -/* - * 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.tools.command.broker; - -import java.lang.reflect.Field; -import java.util.ArrayList; -import java.util.List; -import java.util.Map; -import org.apache.commons.cli.CommandLine; -import org.apache.commons.cli.Options; -import org.apache.commons.cli.PosixParser; -import org.apache.rocketmq.client.ClientConfig; -import org.apache.rocketmq.client.exception.MQClientException; -import org.apache.rocketmq.client.impl.MQClientAPIImpl; -import org.apache.rocketmq.client.impl.MQClientManager; -import org.apache.rocketmq.client.impl.factory.MQClientInstance; -import org.apache.rocketmq.common.admin.ConsumeStats; -import org.apache.rocketmq.common.protocol.body.ConsumeStatsList; -import org.apache.rocketmq.remoting.exception.RemotingConnectException; -import org.apache.rocketmq.remoting.exception.RemotingSendRequestException; -import org.apache.rocketmq.remoting.exception.RemotingTimeoutException; -import org.apache.rocketmq.srvutil.ServerUtil; -import org.apache.rocketmq.tools.admin.DefaultMQAdminExt; -import org.apache.rocketmq.tools.admin.DefaultMQAdminExtImpl; -import org.junit.AfterClass; -import org.junit.BeforeClass; -import org.junit.Test; - -import static org.mockito.ArgumentMatchers.anyBoolean; -import static org.mockito.ArgumentMatchers.anyLong; -import static org.mockito.ArgumentMatchers.anyString; -import static org.mockito.Mockito.mock; -import static org.mockito.Mockito.when; - -public class BrokerConsumeStatsSubCommadTest { - private static DefaultMQAdminExt defaultMQAdminExt; - private static DefaultMQAdminExtImpl defaultMQAdminExtImpl; - private static MQClientInstance mqClientInstance = MQClientManager.getInstance().getAndCreateMQClientInstance(new ClientConfig()); - private static MQClientAPIImpl mQClientAPIImpl; - - @BeforeClass - public static void init() throws NoSuchFieldException, IllegalAccessException, InterruptedException, RemotingTimeoutException, MQClientException, RemotingSendRequestException, RemotingConnectException { - mQClientAPIImpl = mock(MQClientAPIImpl.class); - defaultMQAdminExt = new DefaultMQAdminExt(); - defaultMQAdminExtImpl = new DefaultMQAdminExtImpl(defaultMQAdminExt, 1000); - - Field field = DefaultMQAdminExtImpl.class.getDeclaredField("mqClientInstance"); - field.setAccessible(true); - field.set(defaultMQAdminExtImpl, mqClientInstance); - field = MQClientInstance.class.getDeclaredField("mQClientAPIImpl"); - field.setAccessible(true); - field.set(mqClientInstance, mQClientAPIImpl); - field = DefaultMQAdminExt.class.getDeclaredField("defaultMQAdminExtImpl"); - field.setAccessible(true); - field.set(defaultMQAdminExt, defaultMQAdminExtImpl); - - ConsumeStatsList consumeStatsList = new ConsumeStatsList(); - consumeStatsList.setBrokerAddr("127.0l.0.1:10911"); - consumeStatsList.setConsumeStatsList(new ArrayList<Map<String, List<ConsumeStats>>>()); - consumeStatsList.setTotalDiff(123); - when(mQClientAPIImpl.fetchConsumeStatsInBroker(anyString(), anyBoolean(), anyLong())).thenReturn(consumeStatsList); - } - - @AfterClass - public static void terminate() { - } - - @Test - public void testExecute() { - BrokerConsumeStatsSubCommad cmd = new BrokerConsumeStatsSubCommad(); - Options options = ServerUtil.buildCommandlineOptions(new Options()); - String[] subargs = new String[] {"-b 127.0.0.1:10911", "-t 3000", "-l 5", "-o true"}; - final CommandLine commandLine = - ServerUtil.parseCmdLine("mqadmin " + cmd.commandName(), subargs, cmd.buildCommandlineOptions(options), new PosixParser()); - cmd.execute(commandLine, options, null); - } - -} \ No newline at end of file http://git-wip-us.apache.org/repos/asf/incubator-rocketmq/blob/671e9a55/tools/src/test/java/org/apache/rocketmq/tools/command/broker/BrokerStatusSubCommandTest.java ---------------------------------------------------------------------- diff --git a/tools/src/test/java/org/apache/rocketmq/tools/command/broker/BrokerStatusSubCommandTest.java b/tools/src/test/java/org/apache/rocketmq/tools/command/broker/BrokerStatusSubCommandTest.java deleted file mode 100644 index 1b08735..0000000 --- a/tools/src/test/java/org/apache/rocketmq/tools/command/broker/BrokerStatusSubCommandTest.java +++ /dev/null @@ -1,88 +0,0 @@ -/* - * 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.tools.command.broker; - -import java.lang.reflect.Field; -import java.util.HashMap; -import org.apache.commons.cli.CommandLine; -import org.apache.commons.cli.Options; -import org.apache.commons.cli.PosixParser; -import org.apache.rocketmq.client.ClientConfig; -import org.apache.rocketmq.client.exception.MQBrokerException; -import org.apache.rocketmq.client.exception.MQClientException; -import org.apache.rocketmq.client.impl.MQClientAPIImpl; -import org.apache.rocketmq.client.impl.MQClientManager; -import org.apache.rocketmq.client.impl.factory.MQClientInstance; -import org.apache.rocketmq.common.protocol.body.KVTable; -import org.apache.rocketmq.remoting.exception.RemotingConnectException; -import org.apache.rocketmq.remoting.exception.RemotingSendRequestException; -import org.apache.rocketmq.remoting.exception.RemotingTimeoutException; -import org.apache.rocketmq.srvutil.ServerUtil; -import org.apache.rocketmq.tools.admin.DefaultMQAdminExt; -import org.apache.rocketmq.tools.admin.DefaultMQAdminExtImpl; -import org.junit.AfterClass; -import org.junit.BeforeClass; -import org.junit.Test; - -import static org.mockito.ArgumentMatchers.anyLong; -import static org.mockito.ArgumentMatchers.anyString; -import static org.mockito.Mockito.mock; -import static org.mockito.Mockito.when; - -public class BrokerStatusSubCommandTest { - private static DefaultMQAdminExt defaultMQAdminExt; - private static DefaultMQAdminExtImpl defaultMQAdminExtImpl; - private static MQClientInstance mqClientInstance = MQClientManager.getInstance().getAndCreateMQClientInstance(new ClientConfig()); - private static MQClientAPIImpl mQClientAPIImpl; - - @BeforeClass - public static void init() throws NoSuchFieldException, IllegalAccessException, InterruptedException, RemotingTimeoutException, MQClientException, RemotingSendRequestException, RemotingConnectException, MQBrokerException { - mQClientAPIImpl = mock(MQClientAPIImpl.class); - defaultMQAdminExt = new DefaultMQAdminExt(); - defaultMQAdminExtImpl = new DefaultMQAdminExtImpl(defaultMQAdminExt, 1000); - - Field field = DefaultMQAdminExtImpl.class.getDeclaredField("mqClientInstance"); - field.setAccessible(true); - field.set(defaultMQAdminExtImpl, mqClientInstance); - field = MQClientInstance.class.getDeclaredField("mQClientAPIImpl"); - field.setAccessible(true); - field.set(mqClientInstance, mQClientAPIImpl); - field = DefaultMQAdminExt.class.getDeclaredField("defaultMQAdminExtImpl"); - field.setAccessible(true); - field.set(defaultMQAdminExt, defaultMQAdminExtImpl); - - KVTable kvTable = new KVTable(); - kvTable.setTable(new HashMap<String, String>()); - when(mQClientAPIImpl.getBrokerRuntimeInfo(anyString(), anyLong())).thenReturn(kvTable); - } - - @AfterClass - public static void terminate() { - defaultMQAdminExt.shutdown(); - } - - @Test - public void testExecute() { - BrokerStatusSubCommand cmd = new BrokerStatusSubCommand(); - Options options = ServerUtil.buildCommandlineOptions(new Options()); - String[] subargs = new String[] {"-b 127.0.0.1:10911", "-c default-cluster"}; - final CommandLine commandLine = - ServerUtil.parseCmdLine("mqadmin " + cmd.commandName(), subargs, cmd.buildCommandlineOptions(options), new PosixParser()); - cmd.execute(commandLine, options, null); - } - -} \ No newline at end of file http://git-wip-us.apache.org/repos/asf/incubator-rocketmq/blob/671e9a55/tools/src/test/java/org/apache/rocketmq/tools/command/broker/CleanExpiredCQSubCommandTest.java ---------------------------------------------------------------------- diff --git a/tools/src/test/java/org/apache/rocketmq/tools/command/broker/CleanExpiredCQSubCommandTest.java b/tools/src/test/java/org/apache/rocketmq/tools/command/broker/CleanExpiredCQSubCommandTest.java deleted file mode 100644 index 6fcf044..0000000 --- a/tools/src/test/java/org/apache/rocketmq/tools/command/broker/CleanExpiredCQSubCommandTest.java +++ /dev/null @@ -1,84 +0,0 @@ -/* - * 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.tools.command.broker; - -import java.lang.reflect.Field; -import org.apache.commons.cli.CommandLine; -import org.apache.commons.cli.Options; -import org.apache.commons.cli.PosixParser; -import org.apache.rocketmq.client.ClientConfig; -import org.apache.rocketmq.client.exception.MQBrokerException; -import org.apache.rocketmq.client.exception.MQClientException; -import org.apache.rocketmq.client.impl.MQClientAPIImpl; -import org.apache.rocketmq.client.impl.MQClientManager; -import org.apache.rocketmq.client.impl.factory.MQClientInstance; -import org.apache.rocketmq.remoting.exception.RemotingConnectException; -import org.apache.rocketmq.remoting.exception.RemotingSendRequestException; -import org.apache.rocketmq.remoting.exception.RemotingTimeoutException; -import org.apache.rocketmq.srvutil.ServerUtil; -import org.apache.rocketmq.tools.admin.DefaultMQAdminExt; -import org.apache.rocketmq.tools.admin.DefaultMQAdminExtImpl; -import org.junit.AfterClass; -import org.junit.BeforeClass; -import org.junit.Test; - -import static org.mockito.ArgumentMatchers.anyLong; -import static org.mockito.ArgumentMatchers.anyString; -import static org.mockito.Mockito.mock; -import static org.mockito.Mockito.when; - -public class CleanExpiredCQSubCommandTest { - private static DefaultMQAdminExt defaultMQAdminExt; - private static DefaultMQAdminExtImpl defaultMQAdminExtImpl; - private static MQClientInstance mqClientInstance = MQClientManager.getInstance().getAndCreateMQClientInstance(new ClientConfig()); - private static MQClientAPIImpl mQClientAPIImpl; - - @BeforeClass - public static void init() throws NoSuchFieldException, IllegalAccessException, InterruptedException, RemotingTimeoutException, MQClientException, RemotingSendRequestException, RemotingConnectException, MQBrokerException { - mQClientAPIImpl = mock(MQClientAPIImpl.class); - defaultMQAdminExt = new DefaultMQAdminExt(); - defaultMQAdminExtImpl = new DefaultMQAdminExtImpl(defaultMQAdminExt, 1000); - - Field field = DefaultMQAdminExtImpl.class.getDeclaredField("mqClientInstance"); - field.setAccessible(true); - field.set(defaultMQAdminExtImpl, mqClientInstance); - field = MQClientInstance.class.getDeclaredField("mQClientAPIImpl"); - field.setAccessible(true); - field.set(mqClientInstance, mQClientAPIImpl); - field = DefaultMQAdminExt.class.getDeclaredField("defaultMQAdminExtImpl"); - field.setAccessible(true); - field.set(defaultMQAdminExt, defaultMQAdminExtImpl); - - when(mQClientAPIImpl.cleanExpiredConsumeQueue(anyString(), anyLong())).thenReturn(true); - } - - @AfterClass - public static void terminate() { - defaultMQAdminExt.shutdown(); - } - - @Test - public void testExecute() { - CleanExpiredCQSubCommand cmd = new CleanExpiredCQSubCommand(); - Options options = ServerUtil.buildCommandlineOptions(new Options()); - String[] subargs = new String[] {"-b 127.0.0.1:10911", "-c default-cluster"}; - final CommandLine commandLine = - ServerUtil.parseCmdLine("mqadmin " + cmd.commandName(), subargs, cmd.buildCommandlineOptions(options), new PosixParser()); - cmd.execute(commandLine, options, null); - } - -} \ No newline at end of file http://git-wip-us.apache.org/repos/asf/incubator-rocketmq/blob/671e9a55/tools/src/test/java/org/apache/rocketmq/tools/command/broker/CleanUnusedTopicCommandTest.java ---------------------------------------------------------------------- diff --git a/tools/src/test/java/org/apache/rocketmq/tools/command/broker/CleanUnusedTopicCommandTest.java b/tools/src/test/java/org/apache/rocketmq/tools/command/broker/CleanUnusedTopicCommandTest.java deleted file mode 100644 index 3ae2c48..0000000 --- a/tools/src/test/java/org/apache/rocketmq/tools/command/broker/CleanUnusedTopicCommandTest.java +++ /dev/null @@ -1,84 +0,0 @@ -/* - * 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.tools.command.broker; - -import java.lang.reflect.Field; -import org.apache.commons.cli.CommandLine; -import org.apache.commons.cli.Options; -import org.apache.commons.cli.PosixParser; -import org.apache.rocketmq.client.ClientConfig; -import org.apache.rocketmq.client.exception.MQBrokerException; -import org.apache.rocketmq.client.exception.MQClientException; -import org.apache.rocketmq.client.impl.MQClientAPIImpl; -import org.apache.rocketmq.client.impl.MQClientManager; -import org.apache.rocketmq.client.impl.factory.MQClientInstance; -import org.apache.rocketmq.remoting.exception.RemotingConnectException; -import org.apache.rocketmq.remoting.exception.RemotingSendRequestException; -import org.apache.rocketmq.remoting.exception.RemotingTimeoutException; -import org.apache.rocketmq.srvutil.ServerUtil; -import org.apache.rocketmq.tools.admin.DefaultMQAdminExt; -import org.apache.rocketmq.tools.admin.DefaultMQAdminExtImpl; -import org.junit.AfterClass; -import org.junit.BeforeClass; -import org.junit.Test; - -import static org.mockito.ArgumentMatchers.anyLong; -import static org.mockito.ArgumentMatchers.anyString; -import static org.mockito.Mockito.mock; -import static org.mockito.Mockito.when; - -public class CleanUnusedTopicCommandTest { - private static DefaultMQAdminExt defaultMQAdminExt; - private static DefaultMQAdminExtImpl defaultMQAdminExtImpl; - private static MQClientInstance mqClientInstance = MQClientManager.getInstance().getAndCreateMQClientInstance(new ClientConfig()); - private static MQClientAPIImpl mQClientAPIImpl; - - @BeforeClass - public static void init() throws NoSuchFieldException, IllegalAccessException, InterruptedException, RemotingTimeoutException, MQClientException, RemotingSendRequestException, RemotingConnectException, MQBrokerException { - mQClientAPIImpl = mock(MQClientAPIImpl.class); - defaultMQAdminExt = new DefaultMQAdminExt(); - defaultMQAdminExtImpl = new DefaultMQAdminExtImpl(defaultMQAdminExt, 1000); - - Field field = DefaultMQAdminExtImpl.class.getDeclaredField("mqClientInstance"); - field.setAccessible(true); - field.set(defaultMQAdminExtImpl, mqClientInstance); - field = MQClientInstance.class.getDeclaredField("mQClientAPIImpl"); - field.setAccessible(true); - field.set(mqClientInstance, mQClientAPIImpl); - field = DefaultMQAdminExt.class.getDeclaredField("defaultMQAdminExtImpl"); - field.setAccessible(true); - field.set(defaultMQAdminExt, defaultMQAdminExtImpl); - - when(mQClientAPIImpl.cleanUnusedTopicByAddr(anyString(), anyLong())).thenReturn(true); - } - - @AfterClass - public static void terminate() { - defaultMQAdminExt.shutdown(); - } - - @Test - public void testExecute() { - CleanUnusedTopicCommand cmd = new CleanUnusedTopicCommand(); - Options options = ServerUtil.buildCommandlineOptions(new Options()); - String[] subargs = new String[] {"-b 127.0.0.1:10911", "-c default-cluster"}; - final CommandLine commandLine = - ServerUtil.parseCmdLine("mqadmin " + cmd.commandName(), subargs, cmd.buildCommandlineOptions(options), new PosixParser()); - cmd.execute(commandLine, options, null); - } - -} \ No newline at end of file http://git-wip-us.apache.org/repos/asf/incubator-rocketmq/blob/671e9a55/tools/src/test/java/org/apache/rocketmq/tools/command/broker/GetBrokerConfigCommandTest.java ---------------------------------------------------------------------- diff --git a/tools/src/test/java/org/apache/rocketmq/tools/command/broker/GetBrokerConfigCommandTest.java b/tools/src/test/java/org/apache/rocketmq/tools/command/broker/GetBrokerConfigCommandTest.java deleted file mode 100644 index 88a8ea8..0000000 --- a/tools/src/test/java/org/apache/rocketmq/tools/command/broker/GetBrokerConfigCommandTest.java +++ /dev/null @@ -1,89 +0,0 @@ -/* - * 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.tools.command.broker; - -import java.io.UnsupportedEncodingException; -import java.lang.reflect.Field; -import java.util.Properties; -import org.apache.commons.cli.CommandLine; -import org.apache.commons.cli.Options; -import org.apache.commons.cli.PosixParser; -import org.apache.rocketmq.client.ClientConfig; -import org.apache.rocketmq.client.exception.MQBrokerException; -import org.apache.rocketmq.client.exception.MQClientException; -import org.apache.rocketmq.client.impl.MQClientAPIImpl; -import org.apache.rocketmq.client.impl.MQClientManager; -import org.apache.rocketmq.client.impl.factory.MQClientInstance; -import org.apache.rocketmq.remoting.exception.RemotingConnectException; -import org.apache.rocketmq.remoting.exception.RemotingSendRequestException; -import org.apache.rocketmq.remoting.exception.RemotingTimeoutException; -import org.apache.rocketmq.srvutil.ServerUtil; -import org.apache.rocketmq.tools.admin.DefaultMQAdminExt; -import org.apache.rocketmq.tools.admin.DefaultMQAdminExtImpl; -import org.junit.AfterClass; -import org.junit.BeforeClass; -import org.junit.Test; - -import static org.mockito.ArgumentMatchers.anyLong; -import static org.mockito.ArgumentMatchers.anyString; -import static org.mockito.Mockito.mock; -import static org.mockito.Mockito.when; - -public class GetBrokerConfigCommandTest { - private static DefaultMQAdminExt defaultMQAdminExt; - private static DefaultMQAdminExtImpl defaultMQAdminExtImpl; - private static MQClientInstance mqClientInstance = MQClientManager.getInstance().getAndCreateMQClientInstance(new ClientConfig()); - private static MQClientAPIImpl mQClientAPIImpl; - - @BeforeClass - public static void init() throws NoSuchFieldException, IllegalAccessException, InterruptedException, RemotingTimeoutException, MQClientException, RemotingSendRequestException, RemotingConnectException, MQBrokerException, UnsupportedEncodingException { - mQClientAPIImpl = mock(MQClientAPIImpl.class); - defaultMQAdminExt = new DefaultMQAdminExt(); - defaultMQAdminExtImpl = new DefaultMQAdminExtImpl(defaultMQAdminExt, 1000); - - Field field = DefaultMQAdminExtImpl.class.getDeclaredField("mqClientInstance"); - field.setAccessible(true); - field.set(defaultMQAdminExtImpl, mqClientInstance); - field = MQClientInstance.class.getDeclaredField("mQClientAPIImpl"); - field.setAccessible(true); - field.set(mqClientInstance, mQClientAPIImpl); - field = DefaultMQAdminExt.class.getDeclaredField("defaultMQAdminExtImpl"); - field.setAccessible(true); - field.set(defaultMQAdminExt, defaultMQAdminExtImpl); - - Properties properties = new Properties(); - properties.setProperty("maxMessageSize", "5000000"); - properties.setProperty("flushDelayOffsetInterval", "15000"); - properties.setProperty("serverSocketRcvBufSize", "655350"); - when(mQClientAPIImpl.getBrokerConfig(anyString(), anyLong())).thenReturn(properties); - } - - @AfterClass - public static void terminate() { - defaultMQAdminExt.shutdown(); - } - - @Test - public void testExecute() { - GetBrokerConfigCommand cmd = new GetBrokerConfigCommand(); - Options options = ServerUtil.buildCommandlineOptions(new Options()); - String[] subargs = new String[] {"-b 127.0.0.1:10911", "-c default-cluster"}; - final CommandLine commandLine = - ServerUtil.parseCmdLine("mqadmin " + cmd.commandName(), subargs, cmd.buildCommandlineOptions(options), new PosixParser()); - cmd.execute(commandLine, options, null); - } -} \ No newline at end of file http://git-wip-us.apache.org/repos/asf/incubator-rocketmq/blob/671e9a55/tools/src/test/java/org/apache/rocketmq/tools/command/broker/SendMsgStatusCommandTest.java ---------------------------------------------------------------------- diff --git a/tools/src/test/java/org/apache/rocketmq/tools/command/broker/SendMsgStatusCommandTest.java b/tools/src/test/java/org/apache/rocketmq/tools/command/broker/SendMsgStatusCommandTest.java deleted file mode 100644 index 9089a39..0000000 --- a/tools/src/test/java/org/apache/rocketmq/tools/command/broker/SendMsgStatusCommandTest.java +++ /dev/null @@ -1,78 +0,0 @@ -/* - * 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.tools.command.broker; - -import java.lang.reflect.Field; -import org.apache.commons.cli.CommandLine; -import org.apache.commons.cli.Options; -import org.apache.commons.cli.PosixParser; -import org.apache.rocketmq.client.ClientConfig; -import org.apache.rocketmq.client.exception.MQBrokerException; -import org.apache.rocketmq.client.exception.MQClientException; -import org.apache.rocketmq.client.impl.MQClientAPIImpl; -import org.apache.rocketmq.client.impl.MQClientManager; -import org.apache.rocketmq.client.impl.factory.MQClientInstance; -import org.apache.rocketmq.remoting.exception.RemotingConnectException; -import org.apache.rocketmq.remoting.exception.RemotingSendRequestException; -import org.apache.rocketmq.remoting.exception.RemotingTimeoutException; -import org.apache.rocketmq.srvutil.ServerUtil; -import org.apache.rocketmq.tools.admin.DefaultMQAdminExt; -import org.apache.rocketmq.tools.admin.DefaultMQAdminExtImpl; -import org.junit.AfterClass; -import org.junit.BeforeClass; -import org.junit.Test; - -import static org.mockito.Mockito.mock; - -public class SendMsgStatusCommandTest { - private static DefaultMQAdminExt defaultMQAdminExt; - private static DefaultMQAdminExtImpl defaultMQAdminExtImpl; - private static MQClientInstance mqClientInstance = MQClientManager.getInstance().getAndCreateMQClientInstance(new ClientConfig()); - private static MQClientAPIImpl mQClientAPIImpl; - - @BeforeClass - public static void init() throws NoSuchFieldException, IllegalAccessException, InterruptedException, RemotingTimeoutException, MQClientException, RemotingSendRequestException, RemotingConnectException, MQBrokerException { - mQClientAPIImpl = mock(MQClientAPIImpl.class); - defaultMQAdminExt = new DefaultMQAdminExt(); - defaultMQAdminExtImpl = new DefaultMQAdminExtImpl(defaultMQAdminExt, 1000); - - Field field = DefaultMQAdminExtImpl.class.getDeclaredField("mqClientInstance"); - field.setAccessible(true); - field.set(defaultMQAdminExtImpl, mqClientInstance); - field = MQClientInstance.class.getDeclaredField("mQClientAPIImpl"); - field.setAccessible(true); - field.set(mqClientInstance, mQClientAPIImpl); - field = DefaultMQAdminExt.class.getDeclaredField("defaultMQAdminExtImpl"); - field.setAccessible(true); - field.set(defaultMQAdminExt, defaultMQAdminExtImpl); - } - - @AfterClass - public static void terminate() { - defaultMQAdminExt.shutdown(); - } - - @Test - public void testExecute() { - SendMsgStatusCommand cmd = new SendMsgStatusCommand(); - Options options = ServerUtil.buildCommandlineOptions(new Options()); - String[] subargs = new String[] {"-b 127.0.0.1:10911", "-s 1024 -c 10"}; - final CommandLine commandLine = - ServerUtil.parseCmdLine("mqadmin " + cmd.commandName(), subargs, cmd.buildCommandlineOptions(options), new PosixParser()); - //cmd.execute(commandLine, options, null); - } -} \ No newline at end of file http://git-wip-us.apache.org/repos/asf/incubator-rocketmq/blob/671e9a55/tools/src/test/java/org/apache/rocketmq/tools/command/broker/UpdateBrokerConfigSubCommandTest.java ---------------------------------------------------------------------- diff --git a/tools/src/test/java/org/apache/rocketmq/tools/command/broker/UpdateBrokerConfigSubCommandTest.java b/tools/src/test/java/org/apache/rocketmq/tools/command/broker/UpdateBrokerConfigSubCommandTest.java deleted file mode 100644 index cc459ba..0000000 --- a/tools/src/test/java/org/apache/rocketmq/tools/command/broker/UpdateBrokerConfigSubCommandTest.java +++ /dev/null @@ -1,79 +0,0 @@ -/* - * 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.tools.command.broker; - -import java.io.UnsupportedEncodingException; -import java.lang.reflect.Field; -import org.apache.commons.cli.CommandLine; -import org.apache.commons.cli.Options; -import org.apache.commons.cli.PosixParser; -import org.apache.rocketmq.client.ClientConfig; -import org.apache.rocketmq.client.exception.MQBrokerException; -import org.apache.rocketmq.client.exception.MQClientException; -import org.apache.rocketmq.client.impl.MQClientAPIImpl; -import org.apache.rocketmq.client.impl.MQClientManager; -import org.apache.rocketmq.client.impl.factory.MQClientInstance; -import org.apache.rocketmq.remoting.exception.RemotingConnectException; -import org.apache.rocketmq.remoting.exception.RemotingSendRequestException; -import org.apache.rocketmq.remoting.exception.RemotingTimeoutException; -import org.apache.rocketmq.srvutil.ServerUtil; -import org.apache.rocketmq.tools.admin.DefaultMQAdminExt; -import org.apache.rocketmq.tools.admin.DefaultMQAdminExtImpl; -import org.junit.AfterClass; -import org.junit.BeforeClass; -import org.junit.Test; - -import static org.mockito.Mockito.mock; - -public class UpdateBrokerConfigSubCommandTest { - private static DefaultMQAdminExt defaultMQAdminExt; - private static DefaultMQAdminExtImpl defaultMQAdminExtImpl; - private static MQClientInstance mqClientInstance = MQClientManager.getInstance().getAndCreateMQClientInstance(new ClientConfig()); - private static MQClientAPIImpl mQClientAPIImpl; - - @BeforeClass - public static void init() throws NoSuchFieldException, IllegalAccessException, InterruptedException, RemotingTimeoutException, MQClientException, RemotingSendRequestException, RemotingConnectException, MQBrokerException, UnsupportedEncodingException { - mQClientAPIImpl = mock(MQClientAPIImpl.class); - defaultMQAdminExt = new DefaultMQAdminExt(); - defaultMQAdminExtImpl = new DefaultMQAdminExtImpl(defaultMQAdminExt, 1000); - - Field field = DefaultMQAdminExtImpl.class.getDeclaredField("mqClientInstance"); - field.setAccessible(true); - field.set(defaultMQAdminExtImpl, mqClientInstance); - field = MQClientInstance.class.getDeclaredField("mQClientAPIImpl"); - field.setAccessible(true); - field.set(mqClientInstance, mQClientAPIImpl); - field = DefaultMQAdminExt.class.getDeclaredField("defaultMQAdminExtImpl"); - field.setAccessible(true); - field.set(defaultMQAdminExt, defaultMQAdminExtImpl); - } - - @AfterClass - public static void terminate() { - defaultMQAdminExt.shutdown(); - } - - @Test - public void testExecute() { - UpdateBrokerConfigSubCommand cmd = new UpdateBrokerConfigSubCommand(); - Options options = ServerUtil.buildCommandlineOptions(new Options()); - String[] subargs = new String[] {"-b 127.0.0.1:10911", "-c default-cluster", "-k topicname", "-v unit_test"}; - final CommandLine commandLine = - ServerUtil.parseCmdLine("mqadmin " + cmd.commandName(), subargs, cmd.buildCommandlineOptions(options), new PosixParser()); - cmd.execute(commandLine, options, null); - } -} \ No newline at end of file http://git-wip-us.apache.org/repos/asf/incubator-rocketmq/blob/671e9a55/tools/src/test/java/org/apache/rocketmq/tools/command/connection/ConsumerConnectionSubCommandTest.java ---------------------------------------------------------------------- diff --git a/tools/src/test/java/org/apache/rocketmq/tools/command/connection/ConsumerConnectionSubCommandTest.java b/tools/src/test/java/org/apache/rocketmq/tools/command/connection/ConsumerConnectionSubCommandTest.java deleted file mode 100644 index 88530e2..0000000 --- a/tools/src/test/java/org/apache/rocketmq/tools/command/connection/ConsumerConnectionSubCommandTest.java +++ /dev/null @@ -1,99 +0,0 @@ -/* - * 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.tools.command.connection; - -import java.lang.reflect.Field; -import java.util.HashSet; -import java.util.concurrent.ConcurrentHashMap; -import org.apache.commons.cli.CommandLine; -import org.apache.commons.cli.Options; -import org.apache.commons.cli.PosixParser; -import org.apache.rocketmq.client.ClientConfig; -import org.apache.rocketmq.client.exception.MQBrokerException; -import org.apache.rocketmq.client.exception.MQClientException; -import org.apache.rocketmq.client.impl.MQClientAPIImpl; -import org.apache.rocketmq.client.impl.MQClientManager; -import org.apache.rocketmq.client.impl.factory.MQClientInstance; -import org.apache.rocketmq.common.consumer.ConsumeFromWhere; -import org.apache.rocketmq.common.protocol.body.Connection; -import org.apache.rocketmq.common.protocol.body.ConsumerConnection; -import org.apache.rocketmq.common.protocol.heartbeat.ConsumeType; -import org.apache.rocketmq.common.protocol.heartbeat.MessageModel; -import org.apache.rocketmq.common.protocol.heartbeat.SubscriptionData; -import org.apache.rocketmq.remoting.exception.RemotingConnectException; -import org.apache.rocketmq.remoting.exception.RemotingSendRequestException; -import org.apache.rocketmq.remoting.exception.RemotingTimeoutException; -import org.apache.rocketmq.srvutil.ServerUtil; -import org.apache.rocketmq.tools.admin.DefaultMQAdminExt; -import org.apache.rocketmq.tools.admin.DefaultMQAdminExtImpl; -import org.junit.AfterClass; -import org.junit.BeforeClass; -import org.junit.Test; - -import static org.mockito.ArgumentMatchers.anyLong; -import static org.mockito.ArgumentMatchers.anyString; -import static org.mockito.Mockito.mock; -import static org.mockito.Mockito.when; - -public class ConsumerConnectionSubCommandTest { - private static DefaultMQAdminExt defaultMQAdminExt; - private static DefaultMQAdminExtImpl defaultMQAdminExtImpl; - private static MQClientInstance mqClientInstance = MQClientManager.getInstance().getAndCreateMQClientInstance(new ClientConfig()); - private static MQClientAPIImpl mQClientAPIImpl; - - @BeforeClass - public static void init() throws NoSuchFieldException, IllegalAccessException, InterruptedException, RemotingTimeoutException, MQClientException, RemotingSendRequestException, RemotingConnectException, MQBrokerException { - mQClientAPIImpl = mock(MQClientAPIImpl.class); - defaultMQAdminExt = new DefaultMQAdminExt(); - defaultMQAdminExtImpl = new DefaultMQAdminExtImpl(defaultMQAdminExt, 1000); - - Field field = DefaultMQAdminExtImpl.class.getDeclaredField("mqClientInstance"); - field.setAccessible(true); - field.set(defaultMQAdminExtImpl, mqClientInstance); - field = MQClientInstance.class.getDeclaredField("mQClientAPIImpl"); - field.setAccessible(true); - field.set(mqClientInstance, mQClientAPIImpl); - field = DefaultMQAdminExt.class.getDeclaredField("defaultMQAdminExtImpl"); - field.setAccessible(true); - field.set(defaultMQAdminExt, defaultMQAdminExtImpl); - - ConsumerConnection consumerConnection = new ConsumerConnection(); - consumerConnection.setConsumeType(ConsumeType.CONSUME_PASSIVELY); - consumerConnection.setMessageModel(MessageModel.CLUSTERING); - HashSet<Connection> connections = new HashSet<>(); - connections.add(new Connection()); - consumerConnection.setConnectionSet(connections); - consumerConnection.setSubscriptionTable(new ConcurrentHashMap<String, SubscriptionData>()); - consumerConnection.setConsumeFromWhere(ConsumeFromWhere.CONSUME_FROM_FIRST_OFFSET); - when(mQClientAPIImpl.getConsumerConnectionList(anyString(), anyString(), anyLong())).thenReturn(consumerConnection); - } - - @AfterClass - public static void terminate() { - defaultMQAdminExt.shutdown(); - } - - @Test - public void testExecute() { - ConsumerConnectionSubCommand cmd = new ConsumerConnectionSubCommand(); - Options options = ServerUtil.buildCommandlineOptions(new Options()); - String[] subargs = new String[] {"-g default-consumer-group"}; - final CommandLine commandLine = - ServerUtil.parseCmdLine("mqadmin " + cmd.commandName(), subargs, cmd.buildCommandlineOptions(options), new PosixParser()); - cmd.execute(commandLine, options, null); - } -} \ No newline at end of file http://git-wip-us.apache.org/repos/asf/incubator-rocketmq/blob/671e9a55/tools/src/test/java/org/apache/rocketmq/tools/command/connection/ProducerConnectionSubCommandTest.java ---------------------------------------------------------------------- diff --git a/tools/src/test/java/org/apache/rocketmq/tools/command/connection/ProducerConnectionSubCommandTest.java b/tools/src/test/java/org/apache/rocketmq/tools/command/connection/ProducerConnectionSubCommandTest.java deleted file mode 100644 index 8df66fb..0000000 --- a/tools/src/test/java/org/apache/rocketmq/tools/command/connection/ProducerConnectionSubCommandTest.java +++ /dev/null @@ -1,93 +0,0 @@ -/* - * 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.tools.command.connection; - -import java.lang.reflect.Field; -import java.util.HashSet; -import org.apache.commons.cli.CommandLine; -import org.apache.commons.cli.Options; -import org.apache.commons.cli.PosixParser; -import org.apache.rocketmq.client.ClientConfig; -import org.apache.rocketmq.client.exception.MQBrokerException; -import org.apache.rocketmq.client.exception.MQClientException; -import org.apache.rocketmq.client.impl.MQClientAPIImpl; -import org.apache.rocketmq.client.impl.MQClientManager; -import org.apache.rocketmq.client.impl.factory.MQClientInstance; -import org.apache.rocketmq.common.protocol.body.Connection; -import org.apache.rocketmq.common.protocol.body.ProducerConnection; -import org.apache.rocketmq.remoting.exception.RemotingConnectException; -import org.apache.rocketmq.remoting.exception.RemotingSendRequestException; -import org.apache.rocketmq.remoting.exception.RemotingTimeoutException; -import org.apache.rocketmq.srvutil.ServerUtil; -import org.apache.rocketmq.tools.admin.DefaultMQAdminExt; -import org.apache.rocketmq.tools.admin.DefaultMQAdminExtImpl; -import org.junit.AfterClass; -import org.junit.BeforeClass; -import org.junit.Test; - -import static org.mockito.ArgumentMatchers.anyLong; -import static org.mockito.ArgumentMatchers.anyString; -import static org.mockito.Mockito.mock; -import static org.mockito.Mockito.when; - -public class ProducerConnectionSubCommandTest { - private static DefaultMQAdminExt defaultMQAdminExt; - private static DefaultMQAdminExtImpl defaultMQAdminExtImpl; - private static MQClientInstance mqClientInstance = MQClientManager.getInstance().getAndCreateMQClientInstance(new ClientConfig()); - private static MQClientAPIImpl mQClientAPIImpl; - - @BeforeClass - public static void init() throws NoSuchFieldException, IllegalAccessException, InterruptedException, RemotingTimeoutException, MQClientException, RemotingSendRequestException, RemotingConnectException, MQBrokerException { - mQClientAPIImpl = mock(MQClientAPIImpl.class); - defaultMQAdminExt = new DefaultMQAdminExt(); - defaultMQAdminExtImpl = new DefaultMQAdminExtImpl(defaultMQAdminExt, 1000); - - Field field = DefaultMQAdminExtImpl.class.getDeclaredField("mqClientInstance"); - field.setAccessible(true); - field.set(defaultMQAdminExtImpl, mqClientInstance); - field = MQClientInstance.class.getDeclaredField("mQClientAPIImpl"); - field.setAccessible(true); - field.set(mqClientInstance, mQClientAPIImpl); - field = DefaultMQAdminExt.class.getDeclaredField("defaultMQAdminExtImpl"); - field.setAccessible(true); - field.set(defaultMQAdminExt, defaultMQAdminExtImpl); - - ProducerConnection producerConnection = new ProducerConnection(); - Connection connection = new Connection(); - connection.setClientAddr("127.0.0.1:9898"); - connection.setClientId("PID_12345"); - HashSet<Connection> connectionSet = new HashSet<>(); - connectionSet.add(connection); - producerConnection.setConnectionSet(connectionSet); - when(mQClientAPIImpl.getProducerConnectionList(anyString(), anyString(), anyLong())).thenReturn(producerConnection); - } - - @AfterClass - public static void terminate() { - defaultMQAdminExt.shutdown(); - } - - @Test - public void testExecute() { - ProducerConnectionSubCommand cmd = new ProducerConnectionSubCommand(); - Options options = ServerUtil.buildCommandlineOptions(new Options()); - String[] subargs = new String[] {"-g default-producer-group", "-t unit-test"}; - final CommandLine commandLine = - ServerUtil.parseCmdLine("mqadmin " + cmd.commandName(), subargs, cmd.buildCommandlineOptions(options), new PosixParser()); - cmd.execute(commandLine, options, null); - } -} \ No newline at end of file http://git-wip-us.apache.org/repos/asf/incubator-rocketmq/blob/671e9a55/tools/src/test/java/org/apache/rocketmq/tools/command/consumer/ConsumerProgressSubCommandTest.java ---------------------------------------------------------------------- diff --git a/tools/src/test/java/org/apache/rocketmq/tools/command/consumer/ConsumerProgressSubCommandTest.java b/tools/src/test/java/org/apache/rocketmq/tools/command/consumer/ConsumerProgressSubCommandTest.java deleted file mode 100644 index a5af04a..0000000 --- a/tools/src/test/java/org/apache/rocketmq/tools/command/consumer/ConsumerProgressSubCommandTest.java +++ /dev/null @@ -1,111 +0,0 @@ -/* - * 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.tools.command.consumer; - -import java.lang.reflect.Field; -import java.util.ArrayList; -import java.util.HashMap; -import java.util.List; -import org.apache.commons.cli.CommandLine; -import org.apache.commons.cli.Options; -import org.apache.commons.cli.PosixParser; -import org.apache.rocketmq.client.ClientConfig; -import org.apache.rocketmq.client.exception.MQBrokerException; -import org.apache.rocketmq.client.exception.MQClientException; -import org.apache.rocketmq.client.impl.MQClientAPIImpl; -import org.apache.rocketmq.client.impl.MQClientManager; -import org.apache.rocketmq.client.impl.factory.MQClientInstance; -import org.apache.rocketmq.common.admin.ConsumeStats; -import org.apache.rocketmq.common.admin.OffsetWrapper; -import org.apache.rocketmq.common.message.MessageQueue; -import org.apache.rocketmq.common.protocol.route.BrokerData; -import org.apache.rocketmq.common.protocol.route.QueueData; -import org.apache.rocketmq.common.protocol.route.TopicRouteData; -import org.apache.rocketmq.remoting.exception.RemotingException; -import org.apache.rocketmq.srvutil.ServerUtil; -import org.apache.rocketmq.tools.admin.DefaultMQAdminExt; -import org.apache.rocketmq.tools.admin.DefaultMQAdminExtImpl; -import org.junit.AfterClass; -import org.junit.BeforeClass; -import org.junit.Test; - -import static org.mockito.ArgumentMatchers.anyLong; -import static org.mockito.ArgumentMatchers.anyString; -import static org.mockito.Mockito.mock; -import static org.mockito.Mockito.when; - -public class ConsumerProgressSubCommandTest { - private static DefaultMQAdminExt defaultMQAdminExt; - private static DefaultMQAdminExtImpl defaultMQAdminExtImpl; - private static MQClientInstance mqClientInstance = MQClientManager.getInstance().getAndCreateMQClientInstance(new ClientConfig()); - private static MQClientAPIImpl mQClientAPIImpl; - - @BeforeClass - public static void init() throws NoSuchFieldException, IllegalAccessException, InterruptedException, RemotingException, MQClientException, MQBrokerException { - mQClientAPIImpl = mock(MQClientAPIImpl.class); - defaultMQAdminExt = new DefaultMQAdminExt(); - defaultMQAdminExtImpl = new DefaultMQAdminExtImpl(defaultMQAdminExt, 1000); - - Field field = DefaultMQAdminExtImpl.class.getDeclaredField("mqClientInstance"); - field.setAccessible(true); - field.set(defaultMQAdminExtImpl, mqClientInstance); - field = MQClientInstance.class.getDeclaredField("mQClientAPIImpl"); - field.setAccessible(true); - field.set(mqClientInstance, mQClientAPIImpl); - field = DefaultMQAdminExt.class.getDeclaredField("defaultMQAdminExtImpl"); - field.setAccessible(true); - field.set(defaultMQAdminExt, defaultMQAdminExtImpl); - - TopicRouteData topicRouteData = new TopicRouteData(); - List<BrokerData> brokerDatas = new ArrayList<>(); - HashMap<Long, String> brokerAddrs = new HashMap<>(); - brokerAddrs.put(1234l, "127.0.0.1:10911"); - BrokerData brokerData = new BrokerData(); - brokerData.setCluster("default-cluster"); - brokerData.setBrokerName("default-broker"); - brokerData.setBrokerAddrs(brokerAddrs); - brokerDatas.add(brokerData); - topicRouteData.setBrokerDatas(brokerDatas); - topicRouteData.setQueueDatas(new ArrayList<QueueData>()); - topicRouteData.setFilterServerTable(new HashMap<String, List<String>>()); - when(mQClientAPIImpl.getTopicRouteInfoFromNameServer(anyString(), anyLong())).thenReturn(topicRouteData); - - ConsumeStats consumeStats = new ConsumeStats(); - consumeStats.setConsumeTps(1234); - MessageQueue messageQueue = new MessageQueue(); - OffsetWrapper offsetWrapper = new OffsetWrapper(); - HashMap<MessageQueue, OffsetWrapper> stats = new HashMap<>(); - stats.put(messageQueue, offsetWrapper); - consumeStats.setOffsetTable(stats); - when(mQClientAPIImpl.getConsumeStats(anyString(), anyString(), anyString(), anyLong())).thenReturn(consumeStats); - } - - @AfterClass - public static void terminate() { - defaultMQAdminExt.shutdown(); - } - - @Test - public void testExecute() { - ConsumerProgressSubCommand cmd = new ConsumerProgressSubCommand(); - Options options = ServerUtil.buildCommandlineOptions(new Options()); - String[] subargs = new String[] {"-g default-group"}; - final CommandLine commandLine = - ServerUtil.parseCmdLine("mqadmin " + cmd.commandName(), subargs, cmd.buildCommandlineOptions(options), new PosixParser()); - cmd.execute(commandLine, options, null); - } -} \ No newline at end of file http://git-wip-us.apache.org/repos/asf/incubator-rocketmq/blob/671e9a55/tools/src/test/java/org/apache/rocketmq/tools/command/consumer/ConsumerStatusSubCommandTest.java ---------------------------------------------------------------------- diff --git a/tools/src/test/java/org/apache/rocketmq/tools/command/consumer/ConsumerStatusSubCommandTest.java b/tools/src/test/java/org/apache/rocketmq/tools/command/consumer/ConsumerStatusSubCommandTest.java deleted file mode 100644 index 8e846bc..0000000 --- a/tools/src/test/java/org/apache/rocketmq/tools/command/consumer/ConsumerStatusSubCommandTest.java +++ /dev/null @@ -1,131 +0,0 @@ -/* - * 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.tools.command.consumer; - -import java.lang.reflect.Field; -import java.util.ArrayList; -import java.util.HashMap; -import java.util.HashSet; -import java.util.List; -import java.util.TreeMap; -import java.util.TreeSet; -import java.util.concurrent.ConcurrentHashMap; -import org.apache.commons.cli.CommandLine; -import org.apache.commons.cli.Options; -import org.apache.commons.cli.PosixParser; -import org.apache.rocketmq.client.ClientConfig; -import org.apache.rocketmq.client.exception.MQBrokerException; -import org.apache.rocketmq.client.exception.MQClientException; -import org.apache.rocketmq.client.impl.MQClientAPIImpl; -import org.apache.rocketmq.client.impl.MQClientManager; -import org.apache.rocketmq.client.impl.factory.MQClientInstance; -import org.apache.rocketmq.common.consumer.ConsumeFromWhere; -import org.apache.rocketmq.common.message.MessageQueue; -import org.apache.rocketmq.common.protocol.body.Connection; -import org.apache.rocketmq.common.protocol.body.ConsumeStatus; -import org.apache.rocketmq.common.protocol.body.ConsumerConnection; -import org.apache.rocketmq.common.protocol.body.ConsumerRunningInfo; -import org.apache.rocketmq.common.protocol.body.ProcessQueueInfo; -import org.apache.rocketmq.common.protocol.heartbeat.ConsumeType; -import org.apache.rocketmq.common.protocol.heartbeat.MessageModel; -import org.apache.rocketmq.common.protocol.heartbeat.SubscriptionData; -import org.apache.rocketmq.common.protocol.route.BrokerData; -import org.apache.rocketmq.common.protocol.route.QueueData; -import org.apache.rocketmq.common.protocol.route.TopicRouteData; -import org.apache.rocketmq.remoting.exception.RemotingException; -import org.apache.rocketmq.srvutil.ServerUtil; -import org.apache.rocketmq.tools.admin.DefaultMQAdminExt; -import org.apache.rocketmq.tools.admin.DefaultMQAdminExtImpl; -import org.junit.AfterClass; -import org.junit.BeforeClass; -import org.junit.Test; - -import static org.mockito.ArgumentMatchers.anyBoolean; -import static org.mockito.ArgumentMatchers.anyLong; -import static org.mockito.ArgumentMatchers.anyString; -import static org.mockito.Mockito.mock; -import static org.mockito.Mockito.when; - -public class ConsumerStatusSubCommandTest { - private static DefaultMQAdminExt defaultMQAdminExt; - private static DefaultMQAdminExtImpl defaultMQAdminExtImpl; - private static MQClientInstance mqClientInstance = MQClientManager.getInstance().getAndCreateMQClientInstance(new ClientConfig()); - private static MQClientAPIImpl mQClientAPIImpl; - - @BeforeClass - public static void init() throws NoSuchFieldException, IllegalAccessException, InterruptedException, RemotingException, MQClientException, MQBrokerException { - mQClientAPIImpl = mock(MQClientAPIImpl.class); - defaultMQAdminExt = new DefaultMQAdminExt(); - defaultMQAdminExtImpl = new DefaultMQAdminExtImpl(defaultMQAdminExt, 1000); - - Field field = DefaultMQAdminExtImpl.class.getDeclaredField("mqClientInstance"); - field.setAccessible(true); - field.set(defaultMQAdminExtImpl, mqClientInstance); - field = MQClientInstance.class.getDeclaredField("mQClientAPIImpl"); - field.setAccessible(true); - field.set(mqClientInstance, mQClientAPIImpl); - field = DefaultMQAdminExt.class.getDeclaredField("defaultMQAdminExtImpl"); - field.setAccessible(true); - field.set(defaultMQAdminExt, defaultMQAdminExtImpl); - - TopicRouteData topicRouteData = new TopicRouteData(); - List<BrokerData> brokerDatas = new ArrayList<>(); - HashMap<Long, String> brokerAddrs = new HashMap<>(); - brokerAddrs.put(1234l, "127.0.0.1:10911"); - BrokerData brokerData = new BrokerData(); - brokerData.setCluster("default-cluster"); - brokerData.setBrokerName("default-broker"); - brokerData.setBrokerAddrs(brokerAddrs); - brokerDatas.add(brokerData); - topicRouteData.setBrokerDatas(brokerDatas); - topicRouteData.setQueueDatas(new ArrayList<QueueData>()); - topicRouteData.setFilterServerTable(new HashMap<String, List<String>>()); - when(mQClientAPIImpl.getTopicRouteInfoFromNameServer(anyString(), anyLong())).thenReturn(topicRouteData); - - ConsumerConnection consumerConnection = new ConsumerConnection(); - consumerConnection.setConsumeType(ConsumeType.CONSUME_PASSIVELY); - consumerConnection.setMessageModel(MessageModel.CLUSTERING); - HashSet<Connection> connections = new HashSet<>(); - connections.add(new Connection()); - consumerConnection.setConnectionSet(connections); - consumerConnection.setSubscriptionTable(new ConcurrentHashMap<String, SubscriptionData>()); - consumerConnection.setConsumeFromWhere(ConsumeFromWhere.CONSUME_FROM_FIRST_OFFSET); - when(mQClientAPIImpl.getConsumerConnectionList(anyString(), anyString(), anyLong())).thenReturn(consumerConnection); - - ConsumerRunningInfo consumerRunningInfo = new ConsumerRunningInfo(); - consumerRunningInfo.setJstack("test"); - consumerRunningInfo.setMqTable(new TreeMap<MessageQueue, ProcessQueueInfo>()); - consumerRunningInfo.setStatusTable(new TreeMap<String, ConsumeStatus>()); - consumerRunningInfo.setSubscriptionSet(new TreeSet<SubscriptionData>()); - when(mQClientAPIImpl.getConsumerRunningInfo(anyString(), anyString(), anyString(), anyBoolean(), anyLong())).thenReturn(consumerRunningInfo); - } - - @AfterClass - public static void terminate() { - defaultMQAdminExt.shutdown(); - } - - @Test - public void testExecute() { - ConsumerStatusSubCommand cmd = new ConsumerStatusSubCommand(); - Options options = ServerUtil.buildCommandlineOptions(new Options()); - String[] subargs = new String[] {"-g default-group", "-i cid_one"}; - final CommandLine commandLine = - ServerUtil.parseCmdLine("mqadmin " + cmd.commandName(), subargs, cmd.buildCommandlineOptions(options), new PosixParser()); - cmd.execute(commandLine, options, null); - } -} \ No newline at end of file http://git-wip-us.apache.org/repos/asf/incubator-rocketmq/blob/671e9a55/tools/src/test/java/org/apache/rocketmq/tools/command/namesrv/GetNamesrvConfigCommandTest.java ---------------------------------------------------------------------- diff --git a/tools/src/test/java/org/apache/rocketmq/tools/command/namesrv/GetNamesrvConfigCommandTest.java b/tools/src/test/java/org/apache/rocketmq/tools/command/namesrv/GetNamesrvConfigCommandTest.java deleted file mode 100644 index 49802b9..0000000 --- a/tools/src/test/java/org/apache/rocketmq/tools/command/namesrv/GetNamesrvConfigCommandTest.java +++ /dev/null @@ -1,91 +0,0 @@ -/* - * 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.tools.command.namesrv; - -import java.io.UnsupportedEncodingException; -import java.lang.reflect.Field; -import java.util.ArrayList; -import java.util.HashMap; -import java.util.List; -import java.util.Map; -import java.util.Properties; -import org.apache.commons.cli.CommandLine; -import org.apache.commons.cli.Options; -import org.apache.commons.cli.PosixParser; -import org.apache.rocketmq.client.ClientConfig; -import org.apache.rocketmq.client.exception.MQBrokerException; -import org.apache.rocketmq.client.exception.MQClientException; -import org.apache.rocketmq.client.impl.MQClientAPIImpl; -import org.apache.rocketmq.client.impl.MQClientManager; -import org.apache.rocketmq.client.impl.factory.MQClientInstance; -import org.apache.rocketmq.remoting.exception.RemotingConnectException; -import org.apache.rocketmq.remoting.exception.RemotingSendRequestException; -import org.apache.rocketmq.remoting.exception.RemotingTimeoutException; -import org.apache.rocketmq.srvutil.ServerUtil; -import org.apache.rocketmq.tools.admin.DefaultMQAdminExt; -import org.apache.rocketmq.tools.admin.DefaultMQAdminExtImpl; -import org.junit.AfterClass; -import org.junit.BeforeClass; -import org.junit.Test; -import org.mockito.ArgumentMatchers; - -import static org.mockito.ArgumentMatchers.anyLong; -import static org.mockito.Mockito.mock; -import static org.mockito.Mockito.when; - -public class GetNamesrvConfigCommandTest { - private static DefaultMQAdminExt defaultMQAdminExt; - private static DefaultMQAdminExtImpl defaultMQAdminExtImpl; - private static MQClientInstance mqClientInstance = MQClientManager.getInstance().getAndCreateMQClientInstance(new ClientConfig()); - private static MQClientAPIImpl mQClientAPIImpl; - - @BeforeClass - public static void init() throws NoSuchFieldException, IllegalAccessException, InterruptedException, RemotingTimeoutException, MQClientException, RemotingSendRequestException, RemotingConnectException, MQBrokerException, UnsupportedEncodingException { - mQClientAPIImpl = mock(MQClientAPIImpl.class); - defaultMQAdminExt = new DefaultMQAdminExt(); - defaultMQAdminExtImpl = new DefaultMQAdminExtImpl(defaultMQAdminExt, 1000); - - Field field = DefaultMQAdminExtImpl.class.getDeclaredField("mqClientInstance"); - field.setAccessible(true); - field.set(defaultMQAdminExtImpl, mqClientInstance); - field = MQClientInstance.class.getDeclaredField("mQClientAPIImpl"); - field.setAccessible(true); - field.set(mqClientInstance, mQClientAPIImpl); - field = DefaultMQAdminExt.class.getDeclaredField("defaultMQAdminExtImpl"); - field.setAccessible(true); - field.set(defaultMQAdminExt, defaultMQAdminExtImpl); - - Map<String, Properties> propertiesMap = new HashMap<>(); - List<String> nameServers = new ArrayList<>(); - when(mQClientAPIImpl.getNameServerConfig(ArgumentMatchers.<String>anyList(), anyLong())).thenReturn(propertiesMap); - } - - @AfterClass - public static void terminate() { - defaultMQAdminExt.shutdown(); - } - - @Test - public void testExecute() { - GetNamesrvConfigCommand cmd = new GetNamesrvConfigCommand(); - Options options = ServerUtil.buildCommandlineOptions(new Options()); - String[] subargs = new String[] {}; - final CommandLine commandLine = - ServerUtil.parseCmdLine("mqadmin " + cmd.commandName(), subargs, cmd.buildCommandlineOptions(options), new PosixParser()); - cmd.execute(commandLine, options, null); - } -} \ No newline at end of file http://git-wip-us.apache.org/repos/asf/incubator-rocketmq/blob/671e9a55/tools/src/test/java/org/apache/rocketmq/tools/command/namesrv/WipeWritePermSubCommandTest.java ---------------------------------------------------------------------- diff --git a/tools/src/test/java/org/apache/rocketmq/tools/command/namesrv/WipeWritePermSubCommandTest.java b/tools/src/test/java/org/apache/rocketmq/tools/command/namesrv/WipeWritePermSubCommandTest.java deleted file mode 100644 index 5d2781a..0000000 --- a/tools/src/test/java/org/apache/rocketmq/tools/command/namesrv/WipeWritePermSubCommandTest.java +++ /dev/null @@ -1,90 +0,0 @@ -/* - * 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.tools.command.namesrv; - -import java.lang.reflect.Field; -import java.util.ArrayList; -import java.util.List; -import org.apache.commons.cli.CommandLine; -import org.apache.commons.cli.Options; -import org.apache.commons.cli.PosixParser; -import org.apache.rocketmq.client.ClientConfig; -import org.apache.rocketmq.client.exception.MQBrokerException; -import org.apache.rocketmq.client.exception.MQClientException; -import org.apache.rocketmq.client.impl.MQClientAPIImpl; -import org.apache.rocketmq.client.impl.MQClientManager; -import org.apache.rocketmq.client.impl.factory.MQClientInstance; -import org.apache.rocketmq.remoting.exception.RemotingCommandException; -import org.apache.rocketmq.remoting.exception.RemotingConnectException; -import org.apache.rocketmq.remoting.exception.RemotingSendRequestException; -import org.apache.rocketmq.remoting.exception.RemotingTimeoutException; -import org.apache.rocketmq.srvutil.ServerUtil; -import org.apache.rocketmq.tools.admin.DefaultMQAdminExt; -import org.apache.rocketmq.tools.admin.DefaultMQAdminExtImpl; -import org.junit.AfterClass; -import org.junit.BeforeClass; -import org.junit.Test; - -import static org.mockito.ArgumentMatchers.anyLong; -import static org.mockito.ArgumentMatchers.anyString; -import static org.mockito.Mockito.mock; -import static org.mockito.Mockito.when; - -public class WipeWritePermSubCommandTest { - private static DefaultMQAdminExt defaultMQAdminExt; - private static DefaultMQAdminExtImpl defaultMQAdminExtImpl; - private static MQClientInstance mqClientInstance = MQClientManager.getInstance().getAndCreateMQClientInstance(new ClientConfig()); - private static MQClientAPIImpl mQClientAPIImpl; - - @BeforeClass - public static void init() throws NoSuchFieldException, IllegalAccessException, InterruptedException, RemotingTimeoutException, MQClientException, RemotingSendRequestException, RemotingConnectException, MQBrokerException, RemotingCommandException { - mQClientAPIImpl = mock(MQClientAPIImpl.class); - defaultMQAdminExt = new DefaultMQAdminExt(); - defaultMQAdminExtImpl = new DefaultMQAdminExtImpl(defaultMQAdminExt, 1000); - - Field field = DefaultMQAdminExtImpl.class.getDeclaredField("mqClientInstance"); - field.setAccessible(true); - field.set(defaultMQAdminExtImpl, mqClientInstance); - field = MQClientInstance.class.getDeclaredField("mQClientAPIImpl"); - field.setAccessible(true); - field.set(mqClientInstance, mQClientAPIImpl); - field = DefaultMQAdminExt.class.getDeclaredField("defaultMQAdminExtImpl"); - field.setAccessible(true); - field.set(defaultMQAdminExt, defaultMQAdminExtImpl); - - List<String> result = new ArrayList<>(); - result.add("default-name-one"); - result.add("default-name-two"); - when(mqClientInstance.getMQClientAPIImpl().getNameServerAddressList()).thenReturn(result); - when(mQClientAPIImpl.wipeWritePermOfBroker(anyString(), anyString(), anyLong())).thenReturn(6); - } - - @AfterClass - public static void terminate() { - defaultMQAdminExt.shutdown(); - } - - @Test - public void testExecute() { - WipeWritePermSubCommand cmd = new WipeWritePermSubCommand(); - Options options = ServerUtil.buildCommandlineOptions(new Options()); - String[] subargs = new String[] {"-b default-broker"}; - final CommandLine commandLine = - ServerUtil.parseCmdLine("mqadmin " + cmd.commandName(), subargs, cmd.buildCommandlineOptions(options), new PosixParser()); - cmd.execute(commandLine, options, null); - } -} \ No newline at end of file http://git-wip-us.apache.org/repos/asf/incubator-rocketmq/blob/671e9a55/tools/src/test/java/org/apache/rocketmq/tools/command/offset/GetConsumerStatusCommandTest.java ---------------------------------------------------------------------- diff --git a/tools/src/test/java/org/apache/rocketmq/tools/command/offset/GetConsumerStatusCommandTest.java b/tools/src/test/java/org/apache/rocketmq/tools/command/offset/GetConsumerStatusCommandTest.java deleted file mode 100644 index 07dda80..0000000 --- a/tools/src/test/java/org/apache/rocketmq/tools/command/offset/GetConsumerStatusCommandTest.java +++ /dev/null @@ -1,85 +0,0 @@ -/* - * 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.tools.command.offset; - -import java.lang.reflect.Field; -import java.util.HashMap; -import java.util.Map; -import org.apache.commons.cli.CommandLine; -import org.apache.commons.cli.Options; -import org.apache.commons.cli.PosixParser; -import org.apache.rocketmq.client.ClientConfig; -import org.apache.rocketmq.client.exception.MQBrokerException; -import org.apache.rocketmq.client.exception.MQClientException; -import org.apache.rocketmq.client.impl.MQClientAPIImpl; -import org.apache.rocketmq.client.impl.MQClientManager; -import org.apache.rocketmq.client.impl.factory.MQClientInstance; -import org.apache.rocketmq.common.message.MessageQueue; -import org.apache.rocketmq.remoting.exception.RemotingException; -import org.apache.rocketmq.srvutil.ServerUtil; -import org.apache.rocketmq.tools.admin.DefaultMQAdminExt; -import org.apache.rocketmq.tools.admin.DefaultMQAdminExtImpl; -import org.junit.AfterClass; -import org.junit.BeforeClass; -import org.junit.Test; - -import static org.mockito.ArgumentMatchers.anyLong; -import static org.mockito.ArgumentMatchers.anyString; -import static org.mockito.Mockito.mock; -import static org.mockito.Mockito.when; - -public class GetConsumerStatusCommandTest { - private static DefaultMQAdminExt defaultMQAdminExt; - private static DefaultMQAdminExtImpl defaultMQAdminExtImpl; - private static MQClientInstance mqClientInstance = MQClientManager.getInstance().getAndCreateMQClientInstance(new ClientConfig()); - private static MQClientAPIImpl mQClientAPIImpl; - - @BeforeClass - public static void init() throws NoSuchFieldException, IllegalAccessException, InterruptedException, RemotingException, MQClientException, MQBrokerException { - mQClientAPIImpl = mock(MQClientAPIImpl.class); - defaultMQAdminExt = new DefaultMQAdminExt(); - defaultMQAdminExtImpl = new DefaultMQAdminExtImpl(defaultMQAdminExt, 1000); - - Field field = DefaultMQAdminExtImpl.class.getDeclaredField("mqClientInstance"); - field.setAccessible(true); - field.set(defaultMQAdminExtImpl, mqClientInstance); - field = MQClientInstance.class.getDeclaredField("mQClientAPIImpl"); - field.setAccessible(true); - field.set(mqClientInstance, mQClientAPIImpl); - field = DefaultMQAdminExt.class.getDeclaredField("defaultMQAdminExtImpl"); - field.setAccessible(true); - field.set(defaultMQAdminExt, defaultMQAdminExtImpl); - - Map<String, Map<MessageQueue, Long>> consumerStatus = new HashMap<>(); - when(mQClientAPIImpl.invokeBrokerToGetConsumerStatus(anyString(), anyString(), anyString(), anyString(), anyLong())).thenReturn(consumerStatus); - } - - @AfterClass - public static void terminate() { - defaultMQAdminExt.shutdown(); - } - - @Test - public void testExecute() { - GetConsumerStatusCommand cmd = new GetConsumerStatusCommand(); - Options options = ServerUtil.buildCommandlineOptions(new Options()); - String[] subargs = new String[] {"-g default-group", "-t unit-test", "-i clientid"}; - final CommandLine commandLine = - ServerUtil.parseCmdLine("mqadmin " + cmd.commandName(), subargs, cmd.buildCommandlineOptions(options), new PosixParser()); - cmd.execute(commandLine, options, null); - } -} \ No newline at end of file http://git-wip-us.apache.org/repos/asf/incubator-rocketmq/blob/671e9a55/tools/src/test/java/org/apache/rocketmq/tools/command/offset/ResetOffsetByTimeCommandTest.java ---------------------------------------------------------------------- diff --git a/tools/src/test/java/org/apache/rocketmq/tools/command/offset/ResetOffsetByTimeCommandTest.java b/tools/src/test/java/org/apache/rocketmq/tools/command/offset/ResetOffsetByTimeCommandTest.java deleted file mode 100644 index de32660..0000000 --- a/tools/src/test/java/org/apache/rocketmq/tools/command/offset/ResetOffsetByTimeCommandTest.java +++ /dev/null @@ -1,105 +0,0 @@ -/* - * 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.tools.command.offset; - -import java.lang.reflect.Field; -import java.util.ArrayList; -import java.util.HashMap; -import java.util.List; -import java.util.Map; -import org.apache.commons.cli.CommandLine; -import org.apache.commons.cli.Options; -import org.apache.commons.cli.PosixParser; -import org.apache.rocketmq.client.ClientConfig; -import org.apache.rocketmq.client.exception.MQBrokerException; -import org.apache.rocketmq.client.exception.MQClientException; -import org.apache.rocketmq.client.impl.MQClientAPIImpl; -import org.apache.rocketmq.client.impl.MQClientManager; -import org.apache.rocketmq.client.impl.factory.MQClientInstance; -import org.apache.rocketmq.common.message.MessageQueue; -import org.apache.rocketmq.common.protocol.route.BrokerData; -import org.apache.rocketmq.common.protocol.route.QueueData; -import org.apache.rocketmq.common.protocol.route.TopicRouteData; -import org.apache.rocketmq.remoting.exception.RemotingException; -import org.apache.rocketmq.srvutil.ServerUtil; -import org.apache.rocketmq.tools.admin.DefaultMQAdminExt; -import org.apache.rocketmq.tools.admin.DefaultMQAdminExtImpl; -import org.junit.AfterClass; -import org.junit.BeforeClass; -import org.junit.Test; - -import static org.mockito.ArgumentMatchers.anyBoolean; -import static org.mockito.ArgumentMatchers.anyLong; -import static org.mockito.ArgumentMatchers.anyString; -import static org.mockito.Mockito.mock; -import static org.mockito.Mockito.when; - -public class ResetOffsetByTimeCommandTest { - private static DefaultMQAdminExt defaultMQAdminExt; - private static DefaultMQAdminExtImpl defaultMQAdminExtImpl; - private static MQClientInstance mqClientInstance = MQClientManager.getInstance().getAndCreateMQClientInstance(new ClientConfig()); - private static MQClientAPIImpl mQClientAPIImpl; - - @BeforeClass - public static void init() throws NoSuchFieldException, IllegalAccessException, InterruptedException, RemotingException, MQClientException, MQBrokerException { - mQClientAPIImpl = mock(MQClientAPIImpl.class); - defaultMQAdminExt = new DefaultMQAdminExt(); - defaultMQAdminExtImpl = new DefaultMQAdminExtImpl(defaultMQAdminExt, 1000); - - Field field = DefaultMQAdminExtImpl.class.getDeclaredField("mqClientInstance"); - field.setAccessible(true); - field.set(defaultMQAdminExtImpl, mqClientInstance); - field = MQClientInstance.class.getDeclaredField("mQClientAPIImpl"); - field.setAccessible(true); - field.set(mqClientInstance, mQClientAPIImpl); - field = DefaultMQAdminExt.class.getDeclaredField("defaultMQAdminExtImpl"); - field.setAccessible(true); - field.set(defaultMQAdminExt, defaultMQAdminExtImpl); - - TopicRouteData topicRouteData = new TopicRouteData(); - List<BrokerData> brokerDatas = new ArrayList<>(); - HashMap<Long, String> brokerAddrs = new HashMap<>(); - brokerAddrs.put(1234l, "127.0.0.1:10911"); - BrokerData brokerData = new BrokerData(); - brokerData.setCluster("default-cluster"); - brokerData.setBrokerName("default-broker"); - brokerData.setBrokerAddrs(brokerAddrs); - brokerDatas.add(brokerData); - topicRouteData.setBrokerDatas(brokerDatas); - topicRouteData.setQueueDatas(new ArrayList<QueueData>()); - topicRouteData.setFilterServerTable(new HashMap<String, List<String>>()); - when(mQClientAPIImpl.getTopicRouteInfoFromNameServer(anyString(), anyLong())).thenReturn(topicRouteData); - - Map<MessageQueue, Long> messageQueueLongMap = new HashMap<>(); - when(mQClientAPIImpl.invokeBrokerToResetOffset(anyString(), anyString(), anyString(), anyLong(), anyBoolean(), anyLong())).thenReturn(messageQueueLongMap); - } - - @AfterClass - public static void terminate() { - defaultMQAdminExt.shutdown(); - } - - @Test - public void testExecute() { - ResetOffsetByTimeCommand cmd = new ResetOffsetByTimeCommand(); - Options options = ServerUtil.buildCommandlineOptions(new Options()); - String[] subargs = new String[] {"-g default-group", "-t unit-test", "-s 1412131213231", "-f false"}; - final CommandLine commandLine = - ServerUtil.parseCmdLine("mqadmin " + cmd.commandName(), subargs, cmd.buildCommandlineOptions(options), new PosixParser()); - cmd.execute(commandLine, options, null); - } -} \ No newline at end of file http://git-wip-us.apache.org/repos/asf/incubator-rocketmq/blob/671e9a55/tools/src/test/java/org/apache/rocketmq/tools/command/offset/ResetOffsetByTimeOldCommandTest.java ---------------------------------------------------------------------- diff --git a/tools/src/test/java/org/apache/rocketmq/tools/command/offset/ResetOffsetByTimeOldCommandTest.java b/tools/src/test/java/org/apache/rocketmq/tools/command/offset/ResetOffsetByTimeOldCommandTest.java deleted file mode 100644 index c172c7e..0000000 --- a/tools/src/test/java/org/apache/rocketmq/tools/command/offset/ResetOffsetByTimeOldCommandTest.java +++ /dev/null @@ -1,39 +0,0 @@ -/* - * 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.tools.command.offset; - -import org.apache.commons.cli.CommandLine; -import org.apache.commons.cli.Options; -import org.apache.commons.cli.PosixParser; -import org.apache.rocketmq.srvutil.ServerUtil; -import org.junit.Test; - -import static org.assertj.core.api.Assertions.assertThat; - -public class ResetOffsetByTimeOldCommandTest { - @Test - public void testExecute() { - ResetOffsetByTimeOldCommand cmd = new ResetOffsetByTimeOldCommand(); - Options options = ServerUtil.buildCommandlineOptions(new Options()); - String[] subargs = new String[] {"-g default-group", "-t unit-test", "-s 1412131213231", "-f false"}; - final CommandLine commandLine = - ServerUtil.parseCmdLine("mqadmin " + cmd.commandName(), subargs, cmd.buildCommandlineOptions(options), new PosixParser()); - assertThat(commandLine.getOptionValue('g').trim()).isEqualTo("default-group"); - assertThat(commandLine.getOptionValue('t').trim()).isEqualTo("unit-test"); - assertThat(commandLine.getOptionValue('s').trim()).isEqualTo("1412131213231"); - } -} \ No newline at end of file http://git-wip-us.apache.org/repos/asf/incubator-rocketmq/blob/671e9a55/tools/src/test/java/org/apache/rocketmq/tools/command/topic/AllocateMQSubCommandTest.java ---------------------------------------------------------------------- diff --git a/tools/src/test/java/org/apache/rocketmq/tools/command/topic/AllocateMQSubCommandTest.java b/tools/src/test/java/org/apache/rocketmq/tools/command/topic/AllocateMQSubCommandTest.java deleted file mode 100644 index f309141..0000000 --- a/tools/src/test/java/org/apache/rocketmq/tools/command/topic/AllocateMQSubCommandTest.java +++ /dev/null @@ -1,38 +0,0 @@ -/* - * 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.tools.command.topic; - -import org.apache.commons.cli.CommandLine; -import org.apache.commons.cli.Options; -import org.apache.commons.cli.PosixParser; -import org.apache.rocketmq.srvutil.ServerUtil; -import org.junit.Test; - -import static org.assertj.core.api.Assertions.assertThat; - -public class AllocateMQSubCommandTest { - @Test - public void testExecute() { - AllocateMQSubCommand cmd = new AllocateMQSubCommand(); - Options options = ServerUtil.buildCommandlineOptions(new Options()); - String[] subargs = new String[] {"-t unit-test", "-i 127.0.0.1:10911"}; - final CommandLine commandLine = - ServerUtil.parseCmdLine("mqadmin " + cmd.commandName(), subargs, cmd.buildCommandlineOptions(options), new PosixParser()); - assertThat(commandLine.getOptionValue('t').trim()).isEqualTo("unit-test"); - assertThat(commandLine.getOptionValue("i").trim()).isEqualTo("127.0.0.1:10911"); - } -} \ No newline at end of file http://git-wip-us.apache.org/repos/asf/incubator-rocketmq/blob/671e9a55/tools/src/test/java/org/apache/rocketmq/tools/command/topic/DeleteTopicSubCommandTest.java ---------------------------------------------------------------------- diff --git a/tools/src/test/java/org/apache/rocketmq/tools/command/topic/DeleteTopicSubCommandTest.java b/tools/src/test/java/org/apache/rocketmq/tools/command/topic/DeleteTopicSubCommandTest.java deleted file mode 100644 index 4539c0a..0000000 --- a/tools/src/test/java/org/apache/rocketmq/tools/command/topic/DeleteTopicSubCommandTest.java +++ /dev/null @@ -1,38 +0,0 @@ -/* - * 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.tools.command.topic; - -import org.apache.commons.cli.CommandLine; -import org.apache.commons.cli.Options; -import org.apache.commons.cli.PosixParser; -import org.apache.rocketmq.srvutil.ServerUtil; -import org.junit.Test; - -import static org.assertj.core.api.Assertions.assertThat; - -public class DeleteTopicSubCommandTest { - @Test - public void testExecute() { - DeleteTopicSubCommand cmd = new DeleteTopicSubCommand(); - Options options = ServerUtil.buildCommandlineOptions(new Options()); - String[] subargs = new String[] {"-t unit-test", "-c default-cluster"}; - final CommandLine commandLine = - ServerUtil.parseCmdLine("mqadmin " + cmd.commandName(), subargs, cmd.buildCommandlineOptions(options), new PosixParser()); - assertThat(commandLine.getOptionValue('t').trim()).isEqualTo("unit-test"); - assertThat(commandLine.getOptionValue("c").trim()).isEqualTo("default-cluster"); - } -} \ No newline at end of file
