我已经使用返回 SseEmitter 的方法实现了一个控制器,现在我想对其进行测试。到目前为止,我能找到的唯一方法是:
@RunWith(SpringJUnit4ClassRunner.class)
@SpringApplicationConfiguration(classes = {SsePaymentReceivedController.class, AutomatBackendContextInitializer.class, EventBusImpl.class})
@WebAppConfiguration
public class SsePaymentReceivedControllerIntegrationTest {
@Inject
WebApplicationContext context;
@Inject
SsePaymentReceivedController sseCoinController;
@Inject
EventBusImpl eventBus;
MockMvc mockMvc;
@Before
public void setUpMockMvc() {
this.mockMvc = MockMvcBuilders.webAppContextSetup(this.context).build();
}
@Test
public void subscriptionToSseChannelIsFine() throws Exception {
MvcResult result = mockMvc.perform(get("/sse/payment"))
.andExpect(status().isOk())
.andReturn();
eventBus.fireNotification(new PaymentReceivedNotification("50", Currency.EURO));
LinkedHashSet<ResponseBodyEmitter.DataWithMediaType> emitters =
(LinkedHashSet<ResponseBodyEmitter.DataWithMediaType>)Whitebox.getInternalState(((SseEmitter)result.getModelAndView().getModel().get("sseEmitter")), "earlySendAttempts");
final Iterator<ResponseBodyEmitter.DataWithMediaType> iterator = emitters.iterator();
ResponseBodyEmitter.DataWithMediaType dataField = iterator.next();
assertEquals("data:", dataField.getData());
ResponseBodyEmitter.DataWithMediaType valueField = iterator.next();
assertEquals("{\"remainingAmount\":\"50\",\"currency\":\"EURO\"}", valueField.getData());
ResponseBodyEmitter.DataWithMediaType lastField = iterator.next();
assertEquals("\n\n", lastField.getData());
}
}
必须有一种比检查返回模型的内部更好的方法,我正在寻找它 - 有什么想法吗?