모든 Spring 사용자의 경우 비동기 동작이 관련된 현재 통합 테스트를 수행하는 방법은 다음과 같습니다.
비동기 작업 (예 : I / O 호출)이 완료되면 프로덕션 코드에서 응용 프로그램 이벤트를 시작합니다. 어쨌든이 이벤트는 프로덕션에서 비동기 작업의 응답을 처리하는 데 필요합니다.
이 이벤트가 완료되면 테스트 케이스에서 다음 전략을 사용할 수 있습니다.
- 테스트중인 시스템 실행
- 이벤트를 듣고 이벤트가 시작되었는지 확인하십시오
- 당신의 주장을하세요
이 문제를 해결하려면 먼저 일종의 도메인 이벤트가 필요합니다. 완료된 작업을 식별하기 위해 여기에 UUID를 사용하고 있지만 고유 한 경우 다른 것을 자유롭게 사용할 수 있습니다.
(다음 코드 스 니펫은 롬복 주석을 사용 하여 보일러 플레이트 코드를 제거합니다.)
@RequiredArgsConstructor
class TaskCompletedEvent() {
private final UUID taskId;
// add more fields containing the result of the task if required
}
생산 코드 자체는 일반적으로 다음과 같습니다.
@Component
@RequiredArgsConstructor
class Production {
private final ApplicationEventPublisher eventPublisher;
void doSomeTask(UUID taskId) {
// do something like calling a REST endpoint asynchronously
eventPublisher.publishEvent(new TaskCompletedEvent(taskId));
}
}
그런 다음 Spring @EventListener을 사용하여 테스트 코드에서 게시 된 이벤트를 잡을 수 있습니다 . 이벤트 리스너는 스레드 안전 방식으로 두 가지 경우를 처리해야하기 때문에 조금 더 복잡합니다.
- 프로덕션 코드가 테스트 케이스보다 빠르며 테스트 케이스가 이벤트를 확인하기 전에 이벤트가 이미 시작되었거나
- 테스트 케이스는 프로덕션 코드보다 빠르며 테스트 케이스는 이벤트를 기다려야합니다.
ㅏ CountDownLatch다른 답변에서 언급 한 것처럼 두 번째 경우에는 가 사용됩니다. 또한 @Order이벤트 핸들러 메소드 의 주석은이 이벤트 핸들러 메소드가 프로덕션에 사용 된 다른 이벤트 리스너 후에 호출되도록합니다.
@Component
class TaskCompletionEventListener {
private Map<UUID, CountDownLatch> waitLatches = new ConcurrentHashMap<>();
private List<UUID> eventsReceived = new ArrayList<>();
void waitForCompletion(UUID taskId) {
synchronized (this) {
if (eventAlreadyReceived(taskId)) {
return;
}
checkNobodyIsWaiting(taskId);
createLatch(taskId);
}
waitForEvent(taskId);
}
private void checkNobodyIsWaiting(UUID taskId) {
if (waitLatches.containsKey(taskId)) {
throw new IllegalArgumentException("Only one waiting test per task ID supported, but another test is already waiting for " + taskId + " to complete.");
}
}
private boolean eventAlreadyReceived(UUID taskId) {
return eventsReceived.remove(taskId);
}
private void createLatch(UUID taskId) {
waitLatches.put(taskId, new CountDownLatch(1));
}
@SneakyThrows
private void waitForEvent(UUID taskId) {
var latch = waitLatches.get(taskId);
latch.await();
}
@EventListener
@Order
void eventReceived(TaskCompletedEvent event) {
var taskId = event.getTaskId();
synchronized (this) {
if (isSomebodyWaiting(taskId)) {
notifyWaitingTest(taskId);
} else {
eventsReceived.add(taskId);
}
}
}
private boolean isSomebodyWaiting(UUID taskId) {
return waitLatches.containsKey(taskId);
}
private void notifyWaitingTest(UUID taskId) {
var latch = waitLatches.remove(taskId);
latch.countDown();
}
}
마지막 단계는 테스트중인 시스템을 테스트 사례에서 실행하는 것입니다. JUnit 5와 함께 SpringBoot 테스트를 사용하고 있지만 Spring 컨텍스트를 사용하는 모든 테스트에서 동일하게 작동합니다.
@SpringBootTest
class ProductionIntegrationTest {
@Autowired
private Production sut;
@Autowired
private TaskCompletionEventListener listener;
@Test
void thatTaskCompletesSuccessfully() {
var taskId = UUID.randomUUID();
sut.doSomeTask(taskId);
listener.waitForCompletion(taskId);
// do some assertions like looking into the DB if value was stored successfully
}
}
여기의 다른 답변과 달리이 솔루션은 테스트를 병렬로 실행하고 여러 스레드가 동시에 비동기 코드를 행사하는 경우에도 작동합니다.