3rd version of ReservationServiceTest - with ReservationRepository returning null

ReservationServiceTest3.java

@RunWith(MockitoJUnitRunner.class)
public class ReservationServiceTest {
	ReservationService reservationService;
	
	@Mock
	ReservationRepository reservationRepository;
	
	@Before
	public void setUp() throws Exception {
		reservationService = new ReservationService(reservationRepository);
	}
	
	@Test
	public void getReservationDetails_returnsReservationInfo() {
		BDDMockito.given(reservationRepository.findByFirstName("Dhaval"))
					.willReturn(new Reservation(Long.valueOf(1), "Dhaval", "Shah"));
		
		Reservation aReservation = reservationService.getReservationDetails("Dhaval");
		
		assertThat(aReservation.getFirstName()).isEqualTo("Dhaval");
		assertThat(aReservation.getLastName()).isEqualTo("Shah");
	}
	
	@Test(expected = ReservationNotFoundException.class)
	public void getReservationDetails_whenNotFound() {
		BDDMockito.given(reservationRepository.findByFirstName("Dhaval")).willReturn(null);
		Reservation aReservation = reservationService.getReservationDetails("Dhaval");
	}
}