[ISSUE #673] Add unit tests for ProducerController and NamesvrController. (#764)

src/test/java/org/apache/rocketmq/console/controller/NamesvrControllerTest.java

@@ -0,0 +1,56 @@
+/*
+ * 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.console.controller;
+
+import org.apache.rocketmq.console.service.impl.OpsServiceImpl;
+import org.junit.Assert;
+import org.junit.Test;
+import org.mockito.InjectMocks;
+import org.mockito.Spy;
+import org.springframework.test.web.servlet.ResultActions;
+import org.springframework.test.web.servlet.request.MockHttpServletRequestBuilder;
+import org.springframework.test.web.servlet.request.MockMvcRequestBuilders;
+
+public class NamesvrControllerTest extends BaseControllerTest {
+
+    @InjectMocks
+    private NamesvrController namesvrController;
+
+    private MockHttpServletRequestBuilder requestBuilder = null;
+
+    private ResultActions perform;
+
+    @Spy
+    private OpsServiceImpl opsService;
+
+    @Test
+    public void testNsaddr() throws Exception {
+        final String url = "/rocketmq/nsaddr";
+        {
+            super.mockRmqConfigure();
+        }
+        requestBuilder = MockMvcRequestBuilders.get(url);
+        perform = mockMvc.perform(requestBuilder);
+        String namesrvAddr = perform.andReturn().getResponse().getContentAsString();
+        Assert.assertEquals(namesrvAddr, "127.0.0.1:9876");
+    }
+
+    @Override protected Object getTestController() {
+        return namesvrController;
+    }
+}

src/test/java/org/apache/rocketmq/console/controller/ProducerControllerTest.java

@@ -0,0 +1,91 @@
+/*
+ * 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.console.controller;
+
+import java.util.HashSet;
+import org.apache.rocketmq.client.exception.MQClientException;
+import org.apache.rocketmq.common.protocol.body.Connection;
+import org.apache.rocketmq.common.protocol.body.ProducerConnection;
+import org.apache.rocketmq.console.service.impl.ProducerServiceImpl;
+import org.apache.rocketmq.remoting.protocol.LanguageCode;
+import org.junit.Test;
+import org.mockito.InjectMocks;
+import org.mockito.Spy;
+import org.springframework.test.web.servlet.ResultActions;
+import org.springframework.test.web.servlet.request.MockHttpServletRequestBuilder;
+import org.springframework.test.web.servlet.request.MockMvcRequestBuilders;
+
+import static org.hamcrest.Matchers.hasSize;
+import static org.mockito.ArgumentMatchers.anyString;
+import static org.mockito.Mockito.when;
+import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.jsonPath;
+import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.status;
+
+public class ProducerControllerTest extends BaseControllerTest {
+
+    @InjectMocks
+    private ProducerController producerController;
+
+    private MockHttpServletRequestBuilder requestBuilder = null;
+
+    private ResultActions perform;
+
+    @Spy
+    private ProducerServiceImpl producerService;
+
+    @Test
+    public void testProducerConnection() throws Exception {
+        final String url = "/producer/producerConnection.query";
+        {
+            ProducerConnection producerConnection = new ProducerConnection();
+            HashSet<Connection> connections = new HashSet<>();
+            Connection conn = new Connection();
+            conn.setClientAddr("127.0.0.1");
+            conn.setClientId("clientId");
+            conn.setVersion(LanguageCode.JAVA.getCode());
+            connections.add(conn);
+            producerConnection.setConnectionSet(connections);
+            when(mqAdminExt.examineProducerConnectionInfo(anyString(), anyString()))
+                .thenThrow(new MQClientException("Not found the producer group connection", null))
+                .thenReturn(producerConnection);
+        }
+        requestBuilder = MockMvcRequestBuilders.get(url);
+        requestBuilder.param("producerGroup", "producer_test")
+            .param("topic", "topic_test");
+        // 1、no connection
+        perform = mockMvc.perform(requestBuilder);
+        perform.andExpect(status().isOk())
+            .andExpect(jsonPath("$").exists())
+            .andExpect(jsonPath("$.data").doesNotExist())
+            .andExpect(jsonPath("$.status").value(-1))
+            .andExpect(jsonPath("$.errMsg").isNotEmpty());
+
+        // 2、producer connection exist
+        perform = mockMvc.perform(requestBuilder);
+        perform.andExpect(status().isOk())
+            .andExpect(jsonPath("$.data").isMap())
+            .andExpect(jsonPath("$.data.connectionSet").isArray())
+            .andExpect(jsonPath("$.data.connectionSet", hasSize(1)))
+            .andExpect(jsonPath("$.data.connectionSet[0].clientAddr").value("127.0.0.1"))
+            .andExpect(jsonPath("$.data.connectionSet[0].clientId").value("clientId"));
+    }
+
+    @Override protected Object getTestController() {
+        return producerController;
+    }
+}