RxJava 1.x: Как имитировать противодавление в модульном тесте
Учитывая код:
logItemPublisher
.buffer(
loggingProperties.getBufferTimeoutMillis(),
TimeUnit.MILLISECONDS,
loggingProperties.getBufferSize(),
logDispatchScheduler
)
.onBackpressureDrop(droppedLogsHandler)
// persist uses Spring RestOperations
.flatMap(logs -> persist(logs, url)
.subscribeOn(logDispatchScheduler)
)
.subscribe();
где,
logItemPublisher = ReplaySubject.<AbstractHttpLogItem>createWithSize(4 * loggingProperties.getBufferSize())
.toSerialized();
logDispatchScheduler = new TestScheduler()
У меня есть юнит-тест:
@Test
public void testLogServiceSlow() {
loggingProperties.setBufferSize(1);
// rx.ring-buffer.size property stores the size of any in-memory ring buffers that RxJava uses when an
// Observable cannot keep up with rate of event emissions.
// default value is 128 on the JVM; RxJava 2.x makes this configurable in flatMap
System.setProperty("rx.ring-buffer.size", "2");
// this is what persist does
when(restOperations.postForEntity(anyString(), any(HttpEntity.class), eq(Void.class)))
.thenAnswer(invocation -> {
Thread.sleep(500);
return ResponseEntity.ok().build();
});
logServiceClient.persistLogs(logs);
scheduler.advanceTimeBy(2L, TimeUnit.SECONDS);
System.clearProperty("rx.ring-buffer.size");
Mockito.verify(restOperations, times(2))
.postForEntity(Mockito.eq("http://log:9000/log/service"), logsCaptors.capture(), eq(Void.class));
}
Тест не проходит, потому что, несмотря на Thread.sleep
, обратное давление не генерируется. Я не понимаю, почему нет; внутренний кольцевой буфер должен заполняться через 2 элемента, а остальные элементы должны быть отброшены.
1 ответ
Решение
Отвечая на мой собственный вопрос:
int ringBufferSize = Optional.ofNullable(System.getProperty("rx.ring-buffer.size"))
.map(Integer::parseInt)
.orElse(128);
logItemPublisher
.buffer(
loggingProperties.getBufferTimeoutMillis(),
TimeUnit.MILLISECONDS,
loggingProperties.getBufferSize(),
logDispatchScheduler
)
.onBackpressureDrop(droppedLogsHandler)
.observeOn(logDispatchScheduler, ringBufferSize)
.flatMap(logs -> persist(logs, url))
.subscribe();
Затем в тесте просто установите ringBufferSize = 2
нет необходимости устанавливать системное свойство rx.ring-buffer.size
,