Skip to content

Commit 68c3542

Browse files
committed
Test: SSE 통합 테스트에 연결 대기 로직 추가
1 parent 89bad26 commit 68c3542

1 file changed

Lines changed: 5 additions & 0 deletions

File tree

src/test/java/kr/co/amateurs/server/controller/alarm/SseIntegrationTest.java

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,10 +15,12 @@
1515
import org.springframework.test.util.ReflectionTestUtils;
1616
import org.springframework.web.servlet.mvc.method.annotation.SseEmitter;
1717

18+
import java.time.Duration;
1819
import java.util.concurrent.ConcurrentHashMap;
1920

2021
import static io.restassured.RestAssured.given;
2122
import static org.assertj.core.api.Assertions.assertThat;
23+
import static org.awaitility.Awaitility.await;
2224

2325
@Sql(scripts = "/reset.sql", executionPhase = Sql.ExecutionPhase.BEFORE_TEST_CLASS)
2426
@Sql(scripts = "/reset.sql", executionPhase = Sql.ExecutionPhase.AFTER_TEST_METHOD)
@@ -104,6 +106,9 @@ void setUp() {
104106
.then()
105107
.statusCode(HttpStatus.OK.value());
106108

109+
await().atMost(Duration.ofSeconds(2))
110+
.until(() -> getConnections().size() == 1);
111+
107112
given().header("Authorization", "Bearer " + accessToken)
108113
.accept(MediaType.TEXT_EVENT_STREAM_VALUE)
109114
.when()

0 commit comments

Comments
 (0)