Cộng đồng Hỏi & Đáp của Inflearn

안녕하세요 강사님! 테스트 코드를 작성하던 중 문제가 발생하여 문의드립니다.

returnBookTest 상에서 UserLoanHistory가 DB에만 저장되고(90번째 줄), savedUser.userLoanHistories에는 추가가 안되어 user.returnBook을 호출하는 과정에서 NoSuchElementException이 발생하는 것 같습니다.

    @Test
    fun returnBookTest() {
        // given
        bookRepository.save(Book("이상한 나라의 엘리스"))
        val savedUser = userRepository.save(User("김", null))
        userLoanHistoryRepository.save(UserLoanHistory(savedUser, "이상한 나라의 엘리스", false)) // 생성한 UserLoanHistory 객체가 DB에만 저장됨
        val request = BookReturnRequest("김", "이상한 나라의 엘리스")

        // when
        bookService.returnBook(request) // 최종적으로 user.returnBook 호출

        // then
        val results = userLoanHistoryRepository.findAll()
        assertThat(results).hasSize(1)
        assertThat(results[0].isReturn).isTrue
    }

public void returnBook(String bookName) {
  UserLoanHistory targetHistory = this.userLoanHistories.stream() // userLoanHistories.size() == 0
      .filter(history -> history.getBookName().equals(bookName))
      .findFirst()
      .orElseThrow(); // 여기서 예외 발생
  targetHistory.doReturn();
}

따라서 다음과 같이 수정해야 하지 않을까 싶은데 이 부분 확인 부탁드립니다!!

    @Test
    fun returnBookTest() {
        // given
        bookRepository.save(Book("이상한 나라의 엘리스"))
        userRepository.save(User("김", null))

        val loanRequest = BookLoanRequest("김", "이상한 나라의 엘리스")
        bookService.loanBook(loanRequest)

        val returnRequest = BookReturnRequest("김", "이상한 나라의 엘리스")

        // when
        bookService.returnBook(returnRequest)

        // then
        val results = userLoanHistoryRepository.findAll()
        assertThat(results).hasSize(1)
        assertThat(results[0].isReturn).isTrue
    }