1. package com.fosspowered.peist.controller;
2.  
3. import static org.assertj.core.api.Assertions.assertThat;
4. import static org.mockito.BDDMockito.given;
5. import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.get;
6. import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.post;
7. import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.content;
8. import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.status;
9.  
10. import com.fasterxml.jackson.databind.ObjectMapper;
11. import com.fosspowered.peist.model.exceptions.PeistInvalidRequestException;
12. import com.fosspowered.peist.model.json.PasteRequest;
13. import com.fosspowered.peist.model.json.PasteResponse;
14. import com.fosspowered.peist.service.CRUDService;
15. import java.util.List;
16. import java.util.stream.Collectors;
17. import java.util.stream.IntStream;
18. import org.junit.jupiter.api.Test;
19. import org.junit.jupiter.api.extension.ExtendWith;
20. import org.mockito.Mockito;
21. import org.springframework.beans.factory.annotation.Autowired;
22. import org.springframework.boot.test.autoconfigure.web.servlet.WebMvcTest;
23. import org.springframework.boot.test.mock.mockito.MockBean;
24. import org.springframework.http.MediaType;
25. import org.springframework.test.context.junit.jupiter.SpringExtension;
26. import org.springframework.test.web.servlet.MockMvc;
27. import org.springframework.test.web.servlet.MvcResult;
28.  
29. @ExtendWith(SpringExtension.class)
30. @WebMvcTest(PasteController.class)
31. class PasteControllerTest {
32.   @Autowired private MockMvc mvc;
33.  
34.   @Autowired private ObjectMapper objectMapper;
35.  
36.   @MockBean private CRUDService CRUDService;
37.  
38.   @Test
39.   void whenPing_Normal_ReturnPong() throws Exception {
40.     mvc.perform(get("/paste/ping")).andExpect(content().string("pong"));
41.   }
42.  
43.   @Test
44.   void whenAddPaste_ValidRequest_PersistAndReturnPaste() throws Exception {
45.     given(CRUDService.addPaste(Mockito.any(PasteRequest.class)))
46.         .willReturn(PasteResponse.builder().title("title").build());
47.  
48.     MvcResult mvcResult =
49.         mvc.perform(
50.                 post("/paste")
51.                     .contentType(MediaType.APPLICATION_JSON)
52.                     .content(objectMapper.writeValueAsString(PasteRequest.builder().build())))
53.             .andExpect(status().isCreated())
54.             .andReturn();
55.  
56.     PasteResponse pasteResponse =
57.         objectMapper.readValue(mvcResult.getResponse().getContentAsString(), PasteResponse.class);
58.     assertThat(pasteResponse).isNotNull();
59.     assertThat(pasteResponse.getTitle()).isEqualTo("title");
60.   }
61.  
62.   @Test
63.   void whenGetPaste_PasteExists_ReturnsPaste() throws Exception {
64.     given(CRUDService.fetchPaste(Mockito.eq("id")))
65.         .willReturn(PasteResponse.builder().title("title").build());
66.  
67.     MvcResult mvcResult = mvc.perform(get("/paste/id")).andExpect(status().isOk()).andReturn();
68.  
69.     PasteResponse pasteResponse =
70.         objectMapper.readValue(mvcResult.getResponse().getContentAsString(), PasteResponse.class);
71.     assertThat(pasteResponse).isNotNull();
72.     assertThat(pasteResponse.getTitle()).isEqualTo("title");
73.   }
74.  
75.   @Test
76.   void whenGetPaste_NoPasteExists_Show404() throws Exception {
77.     given(CRUDService.fetchPaste(Mockito.eq("id"))).willThrow(PeistInvalidRequestException.class);
78.  
79.     mvc.perform(get("/paste/id")).andExpect(status().isNotFound());
80.   }
81.  
82.   @Test
83.   void whenGetAllPaste_Normal_ReturnsAll() throws Exception {
84.     given(CRUDService.fetchRecentPastes())
85.         .willReturn(
86.             IntStream.of(1, 2, 3)
87.                 .mapToObj(i -> PasteResponse.builder().build())
88.                 .collect(Collectors.toList()));
89.  
90.     MvcResult mvcResult = mvc.perform(get("/paste")).andExpect(status().isOk()).andReturn();
91.     @SuppressWarnings("unchecked")
92.     List<PasteResponse> response =
93.         objectMapper.readValue(mvcResult.getResponse().getContentAsString(), List.class);
94.     assertThat(response).hasSize(3);
95.   }
96. }
97.