From 6deb418c360c3286b08f5572f38749dfd4840ce1 Mon Sep 17 00:00:00 2001
From: hxy1991 <cannel_2020@163.com>
Date: Wed, 5 Dec 2018 21:41:03 +0800
Subject: [PATCH] Fixes #391

---
 console/pom.xml                               |  8 +-
 .../controller/HealthControllerTest.java      | 91 +++++++++++++++++++
 2 files changed, 97 insertions(+), 2 deletions(-)
 create mode 100644 console/src/test/java/com/alibaba/nacos/console/controller/HealthControllerTest.java

diff --git a/console/pom.xml b/console/pom.xml
index 56e333670..ee482ec26 100644
--- a/console/pom.xml
+++ b/console/pom.xml
@@ -59,8 +59,12 @@
             <groupId>org.slf4j</groupId>
             <artifactId>jul-to-slf4j</artifactId>
         </dependency>
-
-        <!-- log -->
+		<!-- log -->
+		<dependency>
+			<groupId>org.mockito</groupId>
+			<artifactId>mockito-core</artifactId>
+			<scope>test</scope>
+		</dependency>
 	</dependencies>
 	<build>
 		<finalName>nacos-server</finalName>
diff --git a/console/src/test/java/com/alibaba/nacos/console/controller/HealthControllerTest.java b/console/src/test/java/com/alibaba/nacos/console/controller/HealthControllerTest.java
new file mode 100644
index 000000000..62bb784cd
--- /dev/null
+++ b/console/src/test/java/com/alibaba/nacos/console/controller/HealthControllerTest.java
@@ -0,0 +1,91 @@
+package com.alibaba.nacos.console.controller;
+
+import com.alibaba.fastjson.JSONObject;
+import com.alibaba.nacos.config.server.service.PersistService;
+import com.alibaba.nacos.naming.web.ApiCommands;
+import org.junit.Assert;
+import org.junit.Before;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.mockito.InjectMocks;
+import org.mockito.Mock;
+import org.mockito.Mockito;
+import org.springframework.mock.web.MockHttpServletResponse;
+import org.springframework.mock.web.MockServletContext;
+import org.springframework.test.context.ContextConfiguration;
+import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
+import org.springframework.test.context.web.WebAppConfiguration;
+import org.springframework.test.web.servlet.MockMvc;
+import org.springframework.test.web.servlet.request.MockHttpServletRequestBuilder;
+import org.springframework.test.web.servlet.request.MockMvcRequestBuilders;
+import org.springframework.test.web.servlet.setup.MockMvcBuilders;
+
+import javax.servlet.http.HttpServletRequest;
+
+import static org.mockito.ArgumentMatchers.any;
+
+@RunWith(SpringJUnit4ClassRunner.class)
+@ContextConfiguration(classes = MockServletContext.class)
+@WebAppConfiguration
+public class HealthControllerTest {
+
+    @InjectMocks
+    private HealthController healthController;
+
+    @Mock
+    private PersistService persistService;
+
+    @Mock
+    private ApiCommands apiCommands;
+
+    private MockMvc mockmvc;
+
+    @Before
+    public void setUp() {
+        mockmvc = MockMvcBuilders.standaloneSetup(healthController).build();
+    }
+
+    @Test
+    public void testLiveness() throws Exception {
+        String url = "/v1/console/health/liveness";
+        MockHttpServletRequestBuilder builder = MockMvcRequestBuilders.get(url);
+        Assert.assertEquals(200, mockmvc.perform(builder).andReturn().getResponse().getStatus());
+    }
+
+    @Test
+    public void testReadiness() throws Exception {
+        String url = "/v1/console/health/readiness";
+
+        Mockito.when(persistService.configInfoCount(any(String.class))).thenReturn(0);
+        Mockito.when(apiCommands.hello(any(HttpServletRequest.class))).thenReturn(new JSONObject());
+        MockHttpServletRequestBuilder builder = MockMvcRequestBuilders.get(url);
+        Assert.assertEquals(200, mockmvc.perform(builder).andReturn().getResponse().getStatus());
+
+        // Config and Naming are not in readiness
+        Mockito.when(persistService.configInfoCount(any(String.class))).thenThrow(
+            new RuntimeException("HealthControllerTest.testReadiness"));
+        Mockito.when(apiCommands.hello(any(HttpServletRequest.class))).thenThrow(
+            new RuntimeException("HealthControllerTest.testReadiness"));
+        builder = MockMvcRequestBuilders.get(url);
+        MockHttpServletResponse response = mockmvc.perform(builder).andReturn().getResponse();
+        Assert.assertEquals(500, response.getStatus());
+        Assert.assertEquals("Config and Naming are not in readiness", response.getContentAsString());
+
+        // Config is not in readiness
+        Mockito.when(persistService.configInfoCount(any(String.class))).thenThrow(
+            new RuntimeException("HealthControllerTest.testReadiness"));
+        Mockito.when(apiCommands.hello(any(HttpServletRequest.class))).thenReturn(new JSONObject());
+        response = mockmvc.perform(builder).andReturn().getResponse();
+        Assert.assertEquals(500, response.getStatus());
+        Assert.assertEquals("Config is not in readiness", response.getContentAsString());
+
+        // Naming is not in readiness
+        Mockito.when(persistService.configInfoCount(any(String.class))).thenReturn(0);
+        Mockito.when(apiCommands.hello(any(HttpServletRequest.class))).thenThrow(
+            new RuntimeException("HealthControllerTest.testReadiness"));
+        builder = MockMvcRequestBuilders.get(url);
+        response = mockmvc.perform(builder).andReturn().getResponse();
+        Assert.assertEquals(500, response.getStatus());
+        Assert.assertEquals("Naming is not in readiness", response.getContentAsString());
+    }
+}
\ No newline at end of file