QuestionServiceTests.java 文件源码

@Test
public void testAddAnswerToQuestion_notFirstAnswer_shouldNotMarkItAsCorrect() {
    when(answerService.countAnswersInQuestion(question)).thenReturn(1);
    question.setIsValid(true);
    question.setCorrectAnswer(null);
    Answer answer = new Answer();
    answer.setId(1l);

    when(answerService.save(any(Answer.class))).thenAnswer(new org.mockito.stubbing.Answer<Answer>() {
        @Override
        public Answer answer(InvocationOnMock invocation) throws Throwable {
            Object[] args = invocation.getArguments();
            return (Answer) args[0];
        }
    });

    service.addAnswerToQuestion(answer, question);

    assertTrue(question.getIsValid());
    verify(answerService, times(1)).save(answer);
    verify(questionRepository, never()).save(question);
}