在單元測試期間注入模擬HttpServletResponse

Question

問題：我如何在CXF JAX-RS服務端點的單元測試期間注入模擬（或真實？）HttpServletResponse？我們的服務是如何設置的

例子（注意我管不着/無權改變非測試結構）：

package services.example; 

import org.springframework.http.MediaType; 
import javax.servlet.http.HttpServletResponse; 
import javax.ws.rs.GET; 
import javax.ws.rs.POST; 
import javax.ws.rs.PUT; 
import javax.ws.rs.Path; 
import javax.ws.rs.PathParam; 
import javax.ws.rs.Produces; 
import javax.ws.rs.core.Context; 

@RestController 
@Path("/base") 
@Produces(MediaType.APPLICATION_JSON_VALUE); 
public class ExampleService { 

    @Autowired 
    SomeInjectedService injectedService; 

    @POST 
    @Path("/endpoint") 
    public Object testEndpoint(Object exampleInput, @Context HttpServletResponse response) { 
    if (exampleInput.someMethod != null) { 
     // Exception thrown here during tests, response is null!!! 
     response.setStatus(301); 
     return exampleInput; 
    } 

    return new injectedService.returnOutputFromInput(exampleInput); 
    } 
} 

測試類：

package test.services.example 

// A gazillion imports. Mockito, JUnit, Hamcrest, Javax.ws.rs, org.apache.cxf, Jackson providers. 

public class testExampleService { 

    @Mock 
    private HttpServletResponse mockResponse; // Should I do this? 

    @Mock 
    SomeInjectedService mockInjectionService; 

    @InjectMocks 
    ExampleService service = new ExampleService(); 

    @Before 
    public void setUpStuff() { 
    MockitoAnnotations.initMocks(this); 

    JAXRSServerFactoryBean sf = new JAXRSServerFactoryBean(); 
    sf.setResourceClasses(ExampleService.class); 

    List<Object> providers = new ArrayList<Object>; 
    JacksonJsonProvider provider = new JacksonJsonProvider(); 
    providers.add(provider); 

    sf.setProviders(providers); 

    sf.setResourceProvider(ExampleService.class, new SingletonResourceProvider(service, true)); 
    sf.setAddress("local://someuri"); 

    // This is where I'm trying to set up my injected Response context. Help? 
    sf.setInvoker(new Invoker() { 
     Invoker jarsInvoker = new JAXRSInvoker(); 

     @Override 
     public Object invoke(Exchange exchange, Object o) { 
     // What goes in here? I've seen an example for setting a mock 
     // Request context, but no for a Response. Here's my current 
     // (broken) attempt to do this... 
     exchange.getInMessage().put(AbstractHTTPDestination.HTTP_RESPONSE, mockResponse); 
     return jarsInvoker.invoke(exchange, o); 
     } 
    }); 
    } 

    @Test 
    public void testEndpointShouldReturn301() { 
    List<Object> providers = new ArrayList<Object>(); 
    JacksonJaxbJsonProvider provider = new JacksonJaxbJsonProvider(); 
    providers.add(provider); 

    WebClient client = WebClient.create("local://someuri", providers); 
    WebClient.getConfig(client) 
      .getRequestContext() 
      .put(LocalConduit.DIRECT_DISPATCH, Boolean.TRUE); 

    // Exception thrown here, Response context doesn't get injected! 
    client.path("/base/endpoint"); 
    SomeTestObject obj = new SomeTestObject(); 
    Response response = client.type(MediaType.APPLICATION_JSON).post(obj); 

    // Never gets run because an exception was thrown :(
    assertThat(response.getStatus(), is(301)); 
    } 

} 

這可能是我在這裏做了幾件事情，但我希望我只需要在我的自定義調用器中進入正確的「魔術」，以便正確地將HttpServletResponse注入到我的端點。我會注意到，在沒有注入Context的端點上，測試正常工作（當我刪除自定義的調用者位時）。

Answer 1

您不必初始化構造示例服務，只要保持這樣的：

@InjectMocks 
ExampleService service; 

如果你想測試@Path（「/端點」），那麼它是正確的你做了什麼，但不怪異。 如果你想測試你的代碼，只是這樣做：

@Test 
    public void testEndpointShouldReturn301() { 
    service.testEndpoint(someObject, mockResponse); 
    assertThat(response.getStatus(), is(301)); 
    }