냉장고 가계부 프로젝트 19

Fridge 프로젝트를 열고 수정, 삭제, 전체삭제 기능을 구현한다.
먼저 Repository 테스트 클래스부터 작성한다.

@RunWith(SpringRunner.class)
@DataJpaTest
public class JpaFridgeRepositoryTests {
    
    @Autowired
    JpaFridgeRepository jpaFridgeRepository;
    
    @Test
    public void createFridge() {
        String nickname = "myFridge";
        Fridge fridge = jpaFridgeRepository.save(new Fridge(nickname));
        assertThat(fridge).isNotNull();
        assertThat(fridge.getNickname()).isEqualTo(nickname);
        assertThat(fridge.getNickname()).isNotEqualTo("another nickname");
    }
    
    @Test
    public void update() {
        Fridge fridge = jpaFridgeRepository.save(new Fridge("myFridge"));
        assertThat(jpaFridgeRepository.findOne(fridge.getId())).isNotNull();
        
        Fridge changeNicknameFridge = new Fridge("otherFridge");
        changeNicknameFridge.setId(fridge.getId());
        jpaFridgeRepository.save(changeNicknameFridge);
        
        Fridge savedFridge = jpaFridgeRepository.findOne(fridge.getId());
        assertThat(savedFridge.getNickname()).isEqualTo(changeNicknameFridge.getNickname());
    }
    
    @Test
    public void remove() {
        Fridge fridge = jpaFridgeRepository.save(new Fridge("myFridge"));
        assertThat(jpaFridgeRepository.findOne(fridge.getId())).isNotNull();
        assertThat(jpaFridgeRepository.count()).isEqualTo(1L);
        
        jpaFridgeRepository.delete(fridge.getId());
        assertThat(jpaFridgeRepository.count()).isZero();
    }

}

JpaFridgeRepositoryTests 테스트 클래스에 update, remove 테스트 메서드를 추가하고 간단한 테스트를 수행한다. update 메서드는 myFridge 라는 닉네임을 가진 Fridge 객체를 저장한다. 그리고 otherFridge라는 별도의 Fridge 객체를 생성한 후 myFridge 객체의 id를 전달한다.
changeNicknameFridge 객체를 저장하면 기존 myFridge 닉네임이 변경되는 것을 검증한다.
remove 메서드는 fridge 객체를 저장하고 삭제한 후 전체 레코드수를 검증한다.

API 제공을 위한 FridgeController 클래스의 테스트 클래스를 작성한다.

    @Test
    public void put() throws Exception {
        given(fridgeRepository.findOne(anyInt())).willReturn(myFridge);
        given(fridgeService.save(any(Fridge.class))).willReturn(myFridge);
        URI uri = UriComponentsBuilder.fromUriString("/fridges/{id}").buildAndExpand(ID).toUri();
        mvc.perform(MockMvcRequestBuilders.put(uri)
                .content(mapper.writeValueAsString(myFridge))
                .contentType(MediaType.APPLICATION_JSON_UTF8))
            .andExpect(status().isNoContent())
            .andExpect(content().string(""));
    }
    
    @Test
    public void delete() throws Exception {
        given(fridgeRepository.findOne(anyInt())).willReturn(myFridge);
        URI uri = UriComponentsBuilder.fromUriString("/fridges/{id}").buildAndExpand(ID).toUri();
        mvc.perform(MockMvcRequestBuilders.delete(uri)
                .contentType(MediaType.APPLICATION_JSON_UTF8))
            .andExpect(status().isNoContent())
            .andExpect(content().string(""));
    }
    
    @Test
    public void deleteAll() throws Exception {
        doNothing().when(fridgeService).removeAll();
        URI uri = UriComponentsBuilder.fromUriString("/fridges").build().toUri();
        mvc.perform(MockMvcRequestBuilders.delete(uri))
            .andExpect(status().isNoContent())
            .andExpect(content().string(""));
        verify(fridgeService, times(1)).removeAll();
    }

put, delete, deleteAll 메서드는 FoodControllerTests 메서드와 동일한 스타일이다. put메서드는 PUT방식으로 /fridges/{id} URL 을 호출한다. 요청 본문에는 수정 내용이 담긴 Fridge 클래스의 JSON 데이터가 전달된다. 응답상태는 NO CONTENT, 응답본문은 비어있다.
delete 메서드는 DELETE 방식으로 /fridges/{id} URL을 호출한다. 요청본문은 없으며, 응답상태는 NO_CONTENT, 응답내용은 비어있다.
deleteAll 메서드는 DELETE방식으로 /fridges URL을 호출한다. 응답상태는 NO Content, 응답본문은 비어있다. Mock 객체 fridgeService 의 removeAll 메서드가 1번 호출됨을 검증한다.
이제 FridgeController 클래스를 구현한다.

    @PutMapping("/{id}")
    public ResponseEntity<?> updateFridge(@PathVariable final int id, @RequestBody final Fridge fridge) {
        if(jpaFridgeRepository.findOne(id) != null) {
            fridgeService.save(fridge);
        }
        return ResponseEntity.noContent().build();
    }
    
    @DeleteMapping("/{id}")
    public ResponseEntity<?> deleteFridgeById(@PathVariable final int id) {
        if(jpaFridgeRepository.findOne(id) != null) {
            fridgeService.remove(id);
        }
        return ResponseEntity.noContent().build();
    }
    
    @DeleteMapping
    public ResponseEntity<?> deleteAllFridge() {
        fridgeService.removeAll();
        return ResponseEntity.noContent().build();
    }

updateFridge 메서드는 Http방식이 Put이고, URL은 /fridges/{id} URI변수 방식으로 매핑된다. 요청본문에 Fridge 객체를 전달받고, uri 변수값 id 변수를 이용해서 repository에서 객체를 호출한다. 데이터가 존재하면 요청본문의 Fridge객체를 persist(save)한다. 리턴값은 ResponseEntitty이다.
deleteFridgeById 메서드는 Http방식이 Delete이고, URL은 /fridges/{id} URI변수 방식으로 매핑된다. uri 변수값 id 변수를 이용해서 repository에서 객체를 호출한다. 데이터가 존재하면 해당 id를 remove메서드에 파라미터로 전달해서 데이터를 삭제한다.
deleteAllFridge 메서드는 Http방식이 Delete이고, URL은 /fridges 로 매핑된다. removeAll 메서드가 호출해서 전체 데이터를 삭제한다.

이제 web 프로젝트를 열고 Fridge 수정/삭제 UI를 구현한다. 먼저, FridgeControllerTests 테스트부터 작성한다.

    @Override
    protected void setUp() {
        restTemplate.delete(FRIDGE_API_URL + "/fridges", Collections.emptyMap());
    }

    @Test
    public void changeNickNameAndSubmit() {
        FridgeCommand fridge = new FridgeCommand();
        fridge.setNickname("myFridge");
        fridge.setId(createFridge(fridge));
        
        browser.get(BASE_URL + "/fridges/" + fridge.getId());
        
        String changeNickname = "otherFridge";
        WebElement nicknameElement = browser.findElement(By.name("nickname"));
        assertThat(nicknameElement.getAttribute("value")).isEqualTo(fridge.getNickname());
        
        nicknameElement.clear();
        nicknameElement.sendKeys(changeNickname);
        browser.findElementByTagName("form").submit();
        
        WebDriverWait wait = new WebDriverWait(browser, 10);
        wait.until(ExpectedConditions.alertIsPresent());
        
        Alert alert = browser.switchTo().alert();
        assertThat(alert.getText()).isEqualTo(changeNickname + "을 수정했습니다.");
        alert.accept();
        
        browser.get(BASE_URL + "/fridges/" + fridge.getId());
        assertThat(browser.findElement(By.name("nickname")).getAttribute("value")).isEqualTo(changeNickname);
    }
    
    @Test
    public void clickDeleteFridgeButton() {
        FridgeCommand fridge = new FridgeCommand();
        fridge.setNickname("myFridge");
        fridge.setId(createFridge(fridge));
        
        browser.get(BASE_URL + "/fridges/" + fridge.getId());
        
        WebElement deleteBtn = browser.findElement(By.linkText("삭제"));
        assertThat(deleteBtn.getAttribute("href")).isEqualTo(BASE_URL + "/fridges/delete/" + fridge.getId());
        deleteBtn.click();
        
        WebDriverWait wait = new WebDriverWait(browser, 10);
        wait.until(ExpectedConditions.alertIsPresent());
        Alert alert = browser.switchTo().alert();
        assertThat(alert.getText()).isEqualTo("삭제했습니다.");
        alert.accept();
        
        ResponseEntity<FridgeCommand> response = restTemplate.getForEntity(FRIDGE_API_URL + "/fridges/" + fridge.getId(), FridgeCommand.class);
        assertThat(response.getStatusCode()).isEqualTo(HttpStatus.NO_CONTENT);
    }

setUp 메서드에 전체 삭제기능을 호출해서 테스트 메서드가 실행전에 데이터를 초기화해준다.
changeNicknameAndSubmit 메서드는 새로 생성한 fridge 상세페이지로 이동해서 nickname 필드 값을 변경하고 submit해서 alert 상태메시지를 검증한 후 다시 상세페이지로 이동해서 nickname필드를 다시 검증한다.

clickDeleteFridgeButton 메서드는 새로 생성한 fridge 상세페이지로 이동해서 삭제 앵커태그를 찾고 anchor 태그의 href 속성값이 삭제URL과 동일한지 검증한다. 검증을 통과하면 click해서 서버를 호출하고 alert 상태메시지를 검증한다. HTTP 통신으로 fridge 데이터가 존재하는지 한번 더 검증한다.

다음은 상세페이지 updateFridgeForm.html 파일이다.

<!DOCTYPE html>
<html xmlns:th="http://www.thymeleaf.org">
<head>
  <title>Fridge</title>
</head>
<body>
  <h2>냉장고 상세</h2>
  <form th:action="@{/fridges/{id}(id=${fridgeCommand.id})}" th:object="${fridgeCommand}" th:method="put">
 <div th:if="${#fields.hasErrors('*')}">
     <ul>
         <li th:each="err : ${#fields.errors('*')}" th:text="${err}">Input is incorrect</li>
     </ul>
 </div>
    <label for="nickname">이름: 
     <input type="text" th:field="*{nickname}" />
    </label><br/>
    <button type="submit">저장</button>
    <a href="/fridges/delete/1" th:href="@{/fridges/delete/{id}(id=*{id})}">삭제</a>
  </form>
</body>
</html>

변경된 부분은 삭제 앵커태그가 submit버튼 옆에 생겼다. 다른 곳은 수정된 곳이 없다.
다음은 FridgeController 구현이다.

    @PutMapping("/{id}")
    public String processUpdateFridge(@PathVariable int id, @Valid FridgeCommand fridgeCommand, Errors errors, RedirectAttributes ra) {
        if(errors.hasErrors()) {
            return "fridges/updateFridgeForm";
        }
        
        restTemplate.put("/fridges/{id}", fridgeCommand, id);
        ra.addFlashAttribute("registerMessage", fridgeCommand.getNickname() + "을 수정했습니다.");
        return "redirect:/fridges";
    }
    
    @GetMapping("/delete/{id}")
    public String deleteFridge(@PathVariable int id, RedirectAttributes ra) {
        restTemplate.delete("/fridges/{id}", id);
        ra.addFlashAttribute("registerMessage", "삭제했습니다.");
        return "redirect:/fridges";
    }

processUpdateFridge 메서드는 Put방식, /fridges/{id} URL로 매핑하며 FridgeCommand 가 요청본문에 전달된다. Validation을 통과하면 restTemplate으로 API서버와 HTTP 통신을 한다. flashAttribute에는 registerMessage 라는 키값과 메시지 Value를 fridges URL로 리다이렉트한다.
deleteFridge 메서드는 GET 방식, /fridges/delete/{id} URL로 매핑한다. restTemplate으로 API서버에 DELETE방식으로 /fridges/{id} URL을 호출한다. 상태메시지를 flashAttribute에 담아서 /fridges URL로 리다이렉트한다.

테스트를 수행해서 문제가 없는지 확인한다.

이전글: 냉장고 가계부 프로젝트 18
다음글: 냉장고 가계부 프로젝트 20