Rejestracja @ControllerAdvice adnotacją Controller w JUnitTest z MockMVC

Question

Moja @ControllerAdvice adnotacją Kontroler wygląda następująco:

@ControllerAdvice 
public class GlobalControllerExceptionHandler { 

    @ResponseStatus(value = HttpStatus.UNAUTHORIZED) 
    @ExceptionHandler(AuthenticationException.class) 
    public void authenticationExceptionHandler() { 
    } 
} 

Oczywiście mój rozwój Test Driven i chciałbym używać wyjątku Handler w testy JUnit. Mój przypadek testowy wygląda następująco:

public class ClientQueriesControllerTest { 

    private MockMvc mockMvc; 

    @InjectMocks 
    private ClientQueriesController controller; 

    @Mock 
    private AuthenticationService authenticationService; 

    @Before 
    public void setup() { 
     MockitoAnnotations.initMocks(this); 
     mockMvc = MockMvcBuilders.standaloneSetup(controller).build(); 
    } 

    @Test 
    public void findAllAccountRelatedClientsUnauthorized() throws Exception { 
     when(authenticationService.validateAuthorization(anyString())).thenThrow(AuthenticationException.class); 

     mockMvc.perform(get("/rest/clients").header("Authorization", UUID.randomUUID().toString())) 
       .andExpect(status().isUnauthorized()); 
    } 
} 

Prawdopodobnie muszę zarejestrować ControllerAdvice klasie. Jak to zrobić?

Answer 1

Aby aktywować pełną konfigurację Spring MVC, należy użyć MockMvcBuilders.webAppContextSetup zamiast MockMvcBuilders.standaloneSetup.

Aby uzyskać więcej informacji, zapoznaj się z częścią Dokumentacja sprężynowa pod numerem this.

Twój kod będzie wyglądać następująco:

@RunWith(SpringJUnit4ClassRunner.class) 
@WebAppConfiguration 
@ContextConfiguration("test-config.xml") 
public class ClientQueriesControllerTest { 

    private MockMvc mockMvc; 

    @Autowired 
    private WebApplicationContext webApplicationContext; 

    @Autowired 
    private AuthenticationService authenticationService; 

    @Before 
    public void setup() { 
     MockitoAnnotations.initMocks(this); 
     mockMvc = MockMvcBuilders.webAppContextSetup(webApplicationContext).build(); 
    } 

    @Test 
    public void findAllAccountRelatedClientsUnauthorized() throws Exception { 
     when(authenticationService.validateAuthorization(anyString())).thenThrow(AuthenticationException.class); 

     mockMvc.perform(get("/rest/clients").header("Authorization", UUID.randomUUID().toString())) 
       .andExpect(status().isUnauthorized()); 
    } 
} 

Następnie wewnątrz test-config.xml byś dodać Wiosna fasoli dla AuthenticationService że jest fałszywa.

<bean id="authenticationService" class="org.mockito.Mockito" factory-method="mock"> 
    <constructor-arg value="your.package.structure.AuthenticationService"/> 
</bean> 

---

Można profili ruchu oczywiście wstrzyknąć mock AuthenticationService w testach jeśli chcesz ponownie użyć zwykły plik konfiguracyjny Wiosna zamiast tworzyć test-config.xml.

---

UPDATE

Po wykopaniu się trochę, to okazało się, że StandaloneMockMvcBuilder zwrócony przez (MockMvcBuilders.standaloneSetup) jest w pełni konfigurowalny. Oznacza to, że możesz podłączyć dowolny preferowany wyjątek.

Ponieważ jednak używasz @ControllerAdvice, poniższy kod nie będzie działać. Jeśli wszystko trzeba by zmienić jednak metoda @ExceptionHandler był wewnątrz samego kontrolera kod jest następujący:

mockMvc = MockMvcBuilders.standaloneSetup(controller).setHandlerExceptionResolvers(new ExceptionHandlerExceptionResolver()).build(); 

---

UPDATE 2

Niektóre bardziej kopanie dał odpowiedź na to, jak można zarejestrować prawidłową procedurę obsługi wyjątku, gdy używasz także @ControllerAdvice.

Musisz zaktualizować kod instalacyjny w teście na następujące kwestie:

@Before 
    public void setUp() throws Exception { 
     final ExceptionHandlerExceptionResolver exceptionHandlerExceptionResolver = new ExceptionHandlerExceptionResolver(); 

     //here we need to setup a dummy application context that only registers the GlobalControllerExceptionHandler 
     final StaticApplicationContext applicationContext = new StaticApplicationContext(); 
     applicationContext.registerBeanDefinition("advice", new RootBeanDefinition(GlobalControllerExceptionHandler.class, null, null)); 

     //set the application context of the resolver to the dummy application context we just created 
     exceptionHandlerExceptionResolver.setApplicationContext(applicationContext); 

     //needed in order to force the exception resolver to update it's internal caches 
     exceptionHandlerExceptionResolver.afterPropertiesSet(); 

     mockMvc = MockMvcBuilders.standaloneSetup(controller).setHandlerExceptionResolvers(exceptionHandlerExceptionResolver).build(); 
    } 

Answer 2

Got piłkę NestedServletException z następującym rozwiązaniem ...

final StaticApplicationContext applicationContext = new StaticApplicationContext(); 
    applicationContext.registerSingleton("exceptionHandler", GlobalControllerExceptionHandler.class); 

    final WebMvcConfigurationSupport webMvcConfigurationSupport = new WebMvcConfigurationSupport(); 
    webMvcConfigurationSupport.setApplicationContext(applicationContext); 

    mockMvc = MockMvcBuilders.standaloneSetup(controller). 
     setHandlerExceptionResolvers(webMvcConfigurationSupport.handlerExceptionResolver()). 
     build(); 

Answer 3

Można dodać do swojej klasy testowej

@Autowired 
@Qualifier("handlerExceptionResolver") 
void setExceptionResolver(HandlerExceptionResolver resolver) 
{ 
    this.exceptionResolver = resolver; 
} 

a następnie dodaj exceptionResolver do MockMvc

@Before 
public void setup() { 
    MockitoAnnotations.initMocks(this); 
    mockMvc = MockMvcBuilders.standaloneSetup(controller) 
       .setHandlerExceptionResolvers(this.exceptionResolver).build(); 
} 

Answer 4

Od wiosny 4.2, można zarejestrować swój ControllerAdvice bezpośrednio do StandaloneMockMvcBuilder :

MockMvcBuilders 
    .standaloneSetup(myController) 
    .setControllerAdvice(new MyontrollerAdvice()) 
    .build();