@Test
public void shouldReturnTheRightMessageForTheGivenException() throws Exception {

  when(mockException.getMessage()).thenReturn("Error message!!");
  String errorMessage = infrastructureErrorBundle.getErrorMessage();
  verify(mockException, times(2)).getMessage();
  verifyNoMoreInteractions(mockException);
  assertThat(errorMessage, notNullValue());
  assertThat(errorMessage, is(instanceOf(String.class)));
  assertThat(errorMessage, is("Error message!!"));
}