|
13 | 13 | */
|
14 | 14 | package io.trino.plugin.openlineage;
|
15 | 15 |
|
16 |
| -import io.airlift.log.Logger; |
17 | 16 | import io.trino.SessionRepresentation;
|
18 | 17 | import io.trino.testing.QueryRunner;
|
19 | 18 |
|
|
24 | 23 | import java.net.http.HttpResponse;
|
25 | 24 |
|
26 | 25 | import static java.nio.charset.StandardCharsets.UTF_8;
|
27 |
| -import static java.util.Locale.ENGLISH; |
28 | 26 | import static org.assertj.core.api.Assertions.assertThat;
|
29 | 27 |
|
30 | 28 | final class TestOpenLineageEventListenerMarquezIntegration
|
31 | 29 | extends BaseTestOpenLineageQueries
|
32 | 30 | {
|
33 |
| - private static final Logger logger = Logger.get(TestOpenLineageEventListenerMarquezIntegration.class); |
34 |
| - |
35 | 31 | private static MarquezServer server;
|
36 | 32 | private static String marquezURI;
|
37 | 33 | private static final HttpClient client = HttpClient.newHttpClient();
|
@@ -191,10 +187,13 @@ private void checkJobRegistration(HttpClient client, String expectedJobName)
|
191 | 187 |
|
192 | 188 | HttpResponse<String> responseJob = client.send(requestJob, HttpResponse.BodyHandlers.ofString());
|
193 | 189 |
|
194 |
| - logger.info(responseJob.body()); |
| 190 | + assertThat(responseJob.statusCode()) |
| 191 | + .withFailMessage("Marquez responded with status %s: %s", responseJob.statusCode(), responseJob.body()) |
| 192 | + .isEqualTo(200); |
195 | 193 |
|
196 |
| - assertThat(responseJob.statusCode()).isEqualTo(200); |
197 |
| - assertThat(responseJob.body().toLowerCase(ENGLISH)).contains("complete"); |
| 194 | + assertThat(responseJob.body()) |
| 195 | + .withFailMessage("Expected job %s to be completed: %s", expectedJobName, responseJob.body()) |
| 196 | + .containsIgnoringCase("complete"); |
198 | 197 | }
|
199 | 198 | catch (Exception e) {
|
200 | 199 | throw new RuntimeException(e);
|
|
0 commit comments