AddressServiceTest.java

@Test(expected = AddressUpdateException.class)
public void shouldNotUpdateAddressAndDoNotSendNotificationWhenAddressIdIsWrong() {
    AddressUpdateRequest addressUpdateRequest = AddressUpdateRequestBuilder.anAddressUpdateRequestBuilder()
            .id(1L)
            .build();

    when(addressRepository.findById(1L)).thenReturn(Optional.empty());

    try {
        addressService.update(addressUpdateRequest);
    } catch (AddressUpdateException ex) {
        verify(addressRepository).findById(1L);

        throw ex;
    }
}