BAEL-4331 (#10460)

* BAEL-4331 * Add an integration test
2021-02-08 21:11:25 -05:00
parent d89a51b555
commit 6c5c6fe317
8 changed files with 169 additions and 0 deletions
@@ -0,0 +1,12 @@
+package com.baeldung.heap;
+
+import org.springframework.boot.SpringApplication;
+import org.springframework.boot.autoconfigure.SpringBootApplication;
+
+@SpringBootApplication
+public class HeapSizeDemoApplication {
+
+    public static void main(String[] args) {
+        SpringApplication.run(HeapSizeDemoApplication.class, args);
+    }
+}
@@ -0,0 +1,31 @@
+package com.baeldung.heap;
+
+public class MemoryStats {
+    private long heapSize;
+    private long heapMaxSize;
+    private long heapFreeSize;
+    
+    public long getHeapSize() {
+        return heapSize;
+    }
+    
+    public void setHeapSize(long heapSize) {
+        this.heapSize = heapSize;
+    }
+    
+    public long getHeapMaxSize() {
+        return heapMaxSize;
+    }
+    
+    public void setHeapMaxSize(long heapMaxSize) {
+        this.heapMaxSize = heapMaxSize;
+    }
+    
+    public long getHeapFreeSize() {
+        return heapFreeSize;
+    }
+    
+    public void setHeapFreeSize(long heapFreeSize) {
+        this.heapFreeSize = heapFreeSize;
+    }
+}
@@ -0,0 +1,20 @@
+package com.baeldung.heap;
+
+import org.springframework.web.bind.annotation.GetMapping;
+import org.springframework.web.bind.annotation.RestController;
+
+@RestController
+public class MemoryStatusController {
+
+    @GetMapping("memory-status")
+    public MemoryStats getMemoryStatistics() {
+        MemoryStats stats = new MemoryStats();
+        stats.setHeapSize(Runtime.getRuntime()
+            .totalMemory());
+        stats.setHeapMaxSize(Runtime.getRuntime()
+            .maxMemory());
+        stats.setHeapFreeSize(Runtime.getRuntime()
+            .freeMemory());
+        return stats;
+    }
+}
@@ -0,0 +1 @@
+JAVA_OPTS="-Xms512m -Xmx1024m"
@@ -0,0 +1,32 @@
+package com.baeldung.heap;
+
+import static org.hamcrest.Matchers.notANumber;
+import static org.hamcrest.Matchers.not;
+import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.get;
+import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.status;
+import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.jsonPath;
+
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.boot.test.autoconfigure.web.servlet.WebMvcTest;
+import org.springframework.http.MediaType;
+import org.springframework.test.context.junit4.SpringRunner;
+import org.springframework.test.web.servlet.MockMvc;
+
+@RunWith(SpringRunner.class)
+@WebMvcTest(MemoryStatusController.class)
+public class MemoryStatusControllerIntegrationTest {
+
+    @Autowired
+    private MockMvc mvc;
+
+    @Test
+    public void whenGetMemoryStatistics_thenReturnJsonArray() throws Exception {
+        mvc.perform(get("/memory-status").contentType(MediaType.APPLICATION_JSON))
+            .andExpect(status().isOk())
+            .andExpect(jsonPath("heapSize", not(notANumber())))
+            .andExpect(jsonPath("heapMaxSize", not(notANumber())))
+            .andExpect(jsonPath("heapFreeSize", not(notANumber())));
+    }
+}