test - login test

Author: robocikd

src/main/java/pl/robocikd/restapi/controller/LoginController.java

@@ -1,5 +1,6 @@
 package pl.robocikd.restapi.controller;
 
+import org.springframework.web.bind.annotation.GetMapping;
 import org.springframework.web.bind.annotation.PostMapping;
 import org.springframework.web.bind.annotation.RequestBody;
 import org.springframework.web.bind.annotation.RestController;
@@ -10,6 +11,10 @@ public class LoginController {
 
     @PostMapping("/login")
     public void login(@RequestBody LoginCredentials credentials) {
+    }
 
+    @GetMapping("/secured")
+    public String secured() {
+        return "secured";
     }
 }

src/test/java/pl/robocikd/restapi/controller/LoginControllerTest.java

@@ -0,0 +1,42 @@
+package pl.robocikd.restapi.controller;
+
+import org.junit.jupiter.api.Test;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.boot.test.autoconfigure.web.servlet.AutoConfigureMockMvc;
+import org.springframework.boot.test.context.SpringBootTest;
+import org.springframework.test.web.servlet.MockMvc;
+import org.springframework.test.web.servlet.MvcResult;
+
+import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.get;
+import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.post;
+import static org.springframework.test.web.servlet.result.MockMvcResultHandlers.print;
+import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.content;
+import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.status;
+
+@SpringBootTest
+@AutoConfigureMockMvc
+class LoginControllerTest {
+
+    @Autowired
+    private MockMvc mockMvc;
+
+    @Test
+    void shouldLoginAndGetContent() throws Exception {
+        MvcResult login = mockMvc.perform(post("/login")
+                        .content("{\"username\": \"test\", \"password\": \"test\"}"))
+                .andDo(print())
+                .andExpect(status().is(200))
+                .andReturn();
+        String token = login.getResponse().getHeader("Authorization");
+
+        mockMvc.perform(get("/secured")
+                        .header("Authorization", token))
+                .andDo(print())
+                .andExpect(status().is(200))
+                .andExpect(content().string("secured"));
+
+        mockMvc.perform(get("/secured"))
+                .andDo(print())
+                .andExpect(status().is(401));
+    }
+}

src/test/java/pl/robocikd/restapi/controller/PostControllerTest.java

@@ -10,6 +10,10 @@
 import org.springframework.test.web.servlet.MockMvc;
 import org.springframework.test.web.servlet.MvcResult;
 import pl.robocikd.restapi.model.Post;
+import pl.robocikd.restapi.repository.PostRepository;
+
+import javax.transaction.Transactional;
+import java.time.LocalDateTime;
 
 import static org.assertj.core.api.AssertionsForClassTypes.assertThat;
 import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.get;
@@ -26,20 +30,28 @@ class PostControllerTest {
     private MockMvc mockMvc;
     @Autowired
     private ObjectMapper objectMapper;
+    @Autowired
+    private PostRepository postRepository;
 
     @Test
+    @Transactional
     void shouldGetSinglePost() throws Exception {
         // given
+        Post newPost = new Post();
+        newPost.setTitle("Test");
+        newPost.setContent("Test content");
+        newPost.setCreated(LocalDateTime.now());
+        postRepository.save(newPost);
         // when
-        MvcResult mvcResult = mockMvc.perform(get("/posts/1"))
+        MvcResult mvcResult = mockMvc.perform(get("/posts/" + newPost.getId()))
                 .andDo(print())
                 .andExpect(status().is(200))
                 .andReturn();
         // then
         Post post = objectMapper.readValue(mvcResult.getResponse().getContentAsString(), Post.class);
         assertThat(post).isNotNull();
-        assertThat(post.getId()).isEqualTo(1L);
-        assertThat(post.getTitle()).isEqualTo("Test post 1");
-        assertThat(post.getComment().size()).isEqualTo(9);
+        assertThat(post.getId()).isEqualTo(newPost.getId());
+        assertThat(post.getTitle()).isEqualTo("Test");
+        assertThat(post.getContent()).isEqualTo("Test content");
     }
 }