Skip to content

Commit 380be82

Browse files
committed
refactor: remove StreamChecker, since it's only used in one spot
1 parent cc3f145 commit 380be82

File tree

2 files changed

+27
-37
lines changed

2 files changed

+27
-37
lines changed

server/src/test/java/net/laprun/sustainability/power/PowerResourceTest.java

Lines changed: 27 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,8 +5,17 @@
55

66
import java.net.URI;
77
import java.time.Duration;
8+
import java.util.ArrayList;
9+
import java.util.Collections;
10+
import java.util.List;
811
import java.util.Set;
12+
import java.util.concurrent.CompletableFuture;
13+
import java.util.concurrent.TimeUnit;
914

15+
import jakarta.ws.rs.client.ClientBuilder;
16+
import jakarta.ws.rs.sse.SseEventSource;
17+
18+
import org.assertj.core.api.Assertions;
1019
import org.junit.jupiter.api.Test;
1120
import org.junit.jupiter.api.condition.EnabledIfSystemProperty;
1221
import org.junit.jupiter.api.condition.EnabledOnOs;
@@ -24,7 +33,24 @@ public class PowerResourceTest {
2433
@Test
2534
public void testPowerEndpoint() throws Exception {
2635
final var pid = getPid();
27-
StreamChecker.checkPowerForPID(uri, pid);
36+
try (final var client = ClientBuilder.newClient();
37+
final var eventSource = SseEventSource
38+
.target(client.target(uri)
39+
.path("power/stream/{pid}")
40+
.resolveTemplate("pid", pid))
41+
.build()) {
42+
CompletableFuture<List<String>> res = new CompletableFuture<>();
43+
List<String> collect = Collections.synchronizedList(new ArrayList<>());
44+
eventSource.register(inboundSseEvent -> {
45+
collect.add(inboundSseEvent.readData());
46+
// stop after one event
47+
eventSource.close();
48+
},
49+
res::completeExceptionally,
50+
() -> res.complete(collect));
51+
eventSource.open();
52+
Assertions.assertThat(res.get(5, TimeUnit.SECONDS)).hasSize(1);
53+
}
2854
}
2955

3056
@Test

server/src/test/java/net/laprun/sustainability/power/StreamChecker.java

Lines changed: 0 additions & 36 deletions
This file was deleted.

0 commit comments

Comments
 (0)