在单元测试期间注入模拟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)); 
    }