Skip to content

Commit

Permalink
Merge pull request quarkusio#45406 from quarkusio/revert-44903-test/s…
Browse files Browse the repository at this point in the history
…se-response-header-test

Revert "Take RestMulti headers and status into account when using SSE resource method"
  • Loading branch information
geoand authored Jan 7, 2025
2 parents 0bc79ad + 70078e0 commit e48292f
Show file tree
Hide file tree
Showing 3 changed files with 52 additions and 264 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -225,105 +225,6 @@ public void testReturnRestMulti5() {
"header2", "h2"));
}

@Test
public void testReturnRestSse() {
Map<String, String> expectedHeaders = Map.of(
"Access-Control-Allow-Origin", "foo",
"Keep-Alive", "bar");
RestAssured
.given()
.get("/test/rest-sse")
.then()
.statusCode(200)
.headers(expectedHeaders);
}

@Test
public void testReturnRestSse2() {
RestAssured
.given()
.get("/test/rest-sse2")
.then()
.statusCode(200)
.headers(Map.of(
"Access-Control-Allow-Origin", "foo",
"Keep-Alive", "bar"));

RestAssured
.given()
.get("/test/rest-sse2?keepAlive=dummy")
.then()
.statusCode(200)
.headers(Map.of(
"Access-Control-Allow-Origin", "foo",
"Keep-Alive", "dummy"));
}

@Test
public void testReturnRestSse3() {
RestAssured
.given()
.get("/test/rest-sse3")
.then()
.statusCode(200)
.headers(Map.of(
"header1", "foo",
"header2", "bar"));

RestAssured
.given()
.get("/test/rest-sse3?h1=h1&h2=h2")
.then()
.statusCode(200)
.headers(Map.of(
"header1", "h1",
"header2", "h2"));
}

@Test
public void testReturnRestSse4() {
RestAssured
.given()
.get("/test/rest-sse2")
.then()
.statusCode(200)
.contentType(MediaType.SERVER_SENT_EVENTS)
.headers(Map.of(
"Access-Control-Allow-Origin", "foo",
"Keep-Alive", "bar"));

RestAssured
.given()
.get("/test/rest-sse2?keepAlive=dummy")
.then()
.statusCode(200)
.contentType(MediaType.SERVER_SENT_EVENTS)
.headers(Map.of(
"Access-Control-Allow-Origin", "foo",
"Keep-Alive", "dummy"));
}

@Test
public void testReturnRestSse5() {
RestAssured
.given()
.get("/test/rest-sse3")
.then()
.statusCode(200)
.headers(Map.of(
"header1", "foo",
"header2", "bar"));

RestAssured
.given()
.get("/test/rest-sse3?h1=h1&h2=h2")
.then()
.statusCode(200)
.headers(Map.of(
"header1", "h1",
"header2", "h2"));
}

@Path("/test")
public static class TestResource {

Expand Down Expand Up @@ -400,80 +301,37 @@ public String throwExceptionPlain() {
@ResponseHeader(name = "Keep-Alive", value = "timeout=5, max=997")
@GET
@Path("/rest-multi")
public Multi<String> getTestRestMulti() {
public RestMulti<String> getTestRestMulti() {
return RestMulti.fromMultiData(Multi.createFrom().item("test")).header("Access-Control-Allow-Origin", "foo")
.header("Keep-Alive", "bar").build();
}

@GET
@Path("/rest-multi2")
public Multi<String> getTestRestMulti2(@DefaultValue("bar") @RestQuery String keepAlive) {
public RestMulti<String> getTestRestMulti2(@DefaultValue("bar") @RestQuery String keepAlive) {
return RestMulti.fromMultiData(Multi.createFrom().item("test")).header("Access-Control-Allow-Origin", "foo")
.header("Keep-Alive", keepAlive).build();
}

@GET
@Path("/rest-multi3")
@Produces("application/octet-stream")
public Multi<byte[]> getTestRestMulti3(@DefaultValue("foo") @RestQuery("h1") String header1,
public RestMulti<byte[]> getTestRestMulti3(@DefaultValue("foo") @RestQuery("h1") String header1,
@DefaultValue("bar") @RestQuery("h2") String header2) {
return RestMulti.fromUniResponse(getWrapper(header1, header2), Wrapper::getData, Wrapper::getHeaders);
}

@GET
@Path("/rest-multi4")
public Multi<byte[]> getTestRestMulti4(@DefaultValue("bar") @RestQuery String keepAlive) {
public RestMulti<byte[]> getTestRestMulti4(@DefaultValue("bar") @RestQuery String keepAlive) {
return RestMulti.fromMultiData(Multi.createFrom().item("test".getBytes(StandardCharsets.UTF_8)))
.header("Access-Control-Allow-Origin", "foo")
.header("Keep-Alive", keepAlive).header("Content-Type", MediaType.TEXT_PLAIN).build();
}

@GET
@Path("/rest-multi5")
public Multi<byte[]> getTestRestMulti5(@DefaultValue("foo") @RestQuery("h1") String header1,
@DefaultValue("bar") @RestQuery("h2") String header2) {
return RestMulti.fromUniResponse(getWrapper(header1, header2), Wrapper::getData, Wrapper::getHeaders);
}

@ResponseHeader(name = "Access-Control-Allow-Origin", value = "*")
@ResponseHeader(name = "Keep-Alive", value = "timeout=5, max=997")
@GET
@Path("/rest-sse")
@Produces(MediaType.SERVER_SENT_EVENTS)
public Multi<String> getTestRestSse() {
return RestMulti.fromMultiData(Multi.createFrom().item("test")).header("Access-Control-Allow-Origin", "foo")
.header("Keep-Alive", "bar").build();
}

@GET
@Path("/rest-sse2")
@Produces(MediaType.SERVER_SENT_EVENTS)
public Multi<String> getTestRestSse2(@DefaultValue("bar") @RestQuery String keepAlive) {
return RestMulti.fromMultiData(Multi.createFrom().item("test")).header("Access-Control-Allow-Origin", "foo")
.header("Keep-Alive", keepAlive).build();
}

@GET
@Path("/rest-sse3")
@Produces(MediaType.SERVER_SENT_EVENTS)
public Multi<byte[]> getTestRestSse3(@DefaultValue("foo") @RestQuery("h1") String header1,
@DefaultValue("bar") @RestQuery("h2") String header2) {
return RestMulti.fromUniResponse(getWrapper(header1, header2), Wrapper::getData, Wrapper::getHeaders);
}

@GET
@Path("/rest-sse4")
@Produces(MediaType.SERVER_SENT_EVENTS)
public Multi<byte[]> getTestRestSse4(@DefaultValue("bar") @RestQuery String keepAlive) {
return RestMulti.fromMultiData(Multi.createFrom().item("test".getBytes(StandardCharsets.UTF_8)))
.header("Access-Control-Allow-Origin", "foo")
.header("Keep-Alive", keepAlive).header("Content-Type", MediaType.TEXT_PLAIN).build();
}

@GET
@Path("/rest-sse5")
@Produces(MediaType.SERVER_SENT_EVENTS)
public Multi<byte[]> getTestRestSse5(@DefaultValue("foo") @RestQuery("h1") String header1,
public RestMulti<byte[]> getTestRestMulti5(@DefaultValue("foo") @RestQuery("h1") String header1,
@DefaultValue("bar") @RestQuery("h2") String header2) {
return RestMulti.fromUniResponse(getWrapper(header1, header2), Wrapper::getData, Wrapper::getHeaders);
}
Expand Down
Loading

0 comments on commit e48292f

Please sign in to comment.