@Test public void shouldValidateIfStaffExists() { String staffId = "1234567"; when(staffService.getUserByEmailIdOrMotechId(staffId)).thenReturn(null); List<FormError> formErrors = formValidator.validateIfStaffExists(staffId); assertThat(formErrors, hasItem(new FormError(FormValidator.STAFF_ID, NOT_FOUND))); MRSUser staff = mock(MRSUser.class); when(staffService.getUserByEmailIdOrMotechId(staffId)).thenReturn(staff); formErrors = formValidator.validateIfStaffExists(staffId); assertThat(formErrors, not(hasItem(new FormError(FormValidator.STAFF_ID, NOT_FOUND)))); }