Skip to content

Commit

Permalink
Add more @multipart tests (#2152)
Browse files Browse the repository at this point in the history
  • Loading branch information
reta authored Dec 23, 2024
1 parent 3e2b391 commit 477ee89
Show file tree
Hide file tree
Showing 4 changed files with 283 additions and 0 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -39,6 +39,7 @@

import jakarta.activation.DataHandler;
import jakarta.mail.util.ByteArrayDataSource;
import jakarta.ws.rs.client.Client;
import jakarta.ws.rs.client.ClientBuilder;
import jakarta.ws.rs.client.Entity;
import jakarta.ws.rs.client.WebTarget;
Expand Down Expand Up @@ -421,6 +422,49 @@ public void testAddBookAsJAXB2() throws Exception {
doAddBook(address, "attachmentData", 200);
}

@Test
public void testAddBookWithDetailsAsMultipart() throws Exception {
String address = "http://localhost:" + PORT + "/bookstore/books/details";

final Client client = ClientBuilder.newClient();
try (InputStream is = getClass()
.getResourceAsStream("/org/apache/cxf/systest/jaxrs/resources/attachmentData")) {
final MultipartBody builder = new MultipartBody(Arrays.asList(
new AttachmentBuilder()
.mediaType("application/xml")
.id("book")
.object(new Book())
.build(),
new AttachmentBuilder()
.id("upfile1Detail")
.object(is)
.contentDisposition(new ContentDisposition("form-data; name=\"field1\";"))
.build(),
new AttachmentBuilder()
.id("upfile2Detail")
.dataHandler(new DataHandler(
new InputStreamDataSource(new ByteArrayInputStream(new byte[0]), "text/xml")))
.contentDisposition(new ContentDisposition("form-data; name=\"field2\";"))
.build(),
new AttachmentBuilder()
.id("upfile3Detail")
.dataHandler(new DataHandler(new InputStreamDataSource(
new ByteArrayInputStream(new byte[0]), "text/xml")))
.contentDisposition(new ContentDisposition("form-data; name=\"field3\";"))
.build()));

final Response response = client
.target(address)
.request("text/xml")
.post(Entity.entity(builder, "multipart/form-data"));

final Book book = response.readEntity(Book.class);
assertThat("Unexpected status code for response:" + response,
response.getStatus(), equalTo(200));
assertThat(book.getName(), equalTo("upfile1Detail,upfile2Detail,upfile3Detail"));
}
}

@Test
public void testAddBookAsJAXBBody() throws Exception {
String address = "http://localhost:" + PORT + "/bookstore/books/jaxb-body";
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -405,6 +405,24 @@ public Response addBookParts(@Multipart("rootPart") Book b1,
return Response.ok(b1).build();
}

@POST
@Path("/books/details")
@Consumes("multipart/form-data")
@Produces("text/xml")
public Response addBookWithDetails(@Multipart(value = "book", type = "application/xml") Book book,
@Multipart("upfile1Detail") Attachment a1,
@Multipart("upfile2Detail") Attachment a2,
@Multipart("upfile3Detail") Attachment a3)
throws Exception {

if (a1.equals(a2) || a1.equals(a3) || a2.equals(a3)) {
throw new WebApplicationException();
}

book.setName(a1.getContentId() + "," + a2.getContentId() + "," + a3.getContentId());
return Response.ok(book).build();
}

@POST
@Path("/books/jaxb-body")
@Consumes("multipart/related;type=\"text/xml\"")
Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1,168 @@
/**
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
*/

package org.apache.cxf.systest.jaxrs.validation;

import java.io.ByteArrayInputStream;
import java.io.InputStream;
import java.util.Arrays;
import java.util.List;

import jakarta.activation.DataHandler;
import jakarta.ws.rs.client.Client;
import jakarta.ws.rs.client.ClientBuilder;
import jakarta.ws.rs.client.Entity;
import jakarta.ws.rs.core.Response;
import org.apache.cxf.Bus;
import org.apache.cxf.jaxrs.JAXRSServerFactoryBean;
import org.apache.cxf.jaxrs.ext.multipart.AttachmentBuilder;
import org.apache.cxf.jaxrs.ext.multipart.ContentDisposition;
import org.apache.cxf.jaxrs.ext.multipart.InputStreamDataSource;
import org.apache.cxf.jaxrs.ext.multipart.MultipartBody;
import org.apache.cxf.jaxrs.lifecycle.SingletonResourceProvider;
import org.apache.cxf.jaxrs.model.AbstractResourceInfo;
import org.apache.cxf.jaxrs.validation.ValidationExceptionMapper;
import org.apache.cxf.testutil.common.AbstractServerTestServerBase;
import org.apache.cxf.validation.BeanValidationFeature;

import org.junit.BeforeClass;
import org.junit.Test;

import static org.hamcrest.CoreMatchers.equalTo;
import static org.hamcrest.MatcherAssert.assertThat;
import static org.junit.Assert.assertTrue;

public class JAXRSMultipartValidationTest extends AbstractJAXRSValidationTest {
public static final String PORT = allocatePort(JAXRSMultipartValidationTest.class);

public static class Server extends AbstractServerTestServerBase {
@Override
protected org.apache.cxf.endpoint.Server createServer(Bus bus) throws Exception {
final JAXRSServerFactoryBean sf = new JAXRSServerFactoryBean();
sf.setResourceClasses(MultipartBookStoreWithValidation.class);
sf.setFeatures(List.of(new BeanValidationFeature()));
sf.setProvider(new ValidationExceptionMapper());
sf.setResourceProvider(new SingletonResourceProvider(new MultipartBookStoreWithValidation()));
sf.setAddress("http://localhost:" + PORT + "/");
return sf.create();
}

public static void main(String[] args) throws Exception {
new Server().start();
}
}

@BeforeClass
public static void startServers() throws Exception {
AbstractResourceInfo.clearAllMaps();
//keep out of process due to stack traces testing failures
assertTrue("server did not launch correctly", launchServer(Server.class, true));
}

@Test
public void testAddBookWithDetailsAsMultipartBadRequest() throws Exception {
String address = "http://localhost:" + PORT + "/bookstore/books/details";

final Client client = ClientBuilder.newClient();
try (InputStream is = getClass()
.getResourceAsStream("/org/apache/cxf/systest/jaxrs/resources/attachmentData")) {
final MultipartBody builder = new MultipartBody(Arrays.asList(
new AttachmentBuilder()
.mediaType("application/xml")
.id("book")
.object(new BookWithValidation())
.build(),
new AttachmentBuilder()
.id("upfile1Detail")
.object(is)
.contentDisposition(new ContentDisposition("form-data; name=\"field1\";"))
.build(),
new AttachmentBuilder()
.id("upfile2Detail")
.dataHandler(new DataHandler(
new InputStreamDataSource(new ByteArrayInputStream(new byte[0]), "text/xml")))
.contentDisposition(new ContentDisposition("form-data; name=\"field2\";"))
.build(),
new AttachmentBuilder()
.id("upfile3Detail")
.dataHandler(new DataHandler(new InputStreamDataSource(
new ByteArrayInputStream(new byte[0]), "text/xml")))
.contentDisposition(new ContentDisposition("form-data; name=\"field3\";"))
.build()));

final Response response = client
.target(address)
.request("text/xml")
.post(Entity.entity(builder, "multipart/form-data"));

// Book's name is 'null', validation should fail with 400
assertThat("Unexpected status code for response:" + response,
response.getStatus(), equalTo(400));
}
}

@Test
public void testAddBookWithDetailsAsMultipart() throws Exception {
String address = "http://localhost:" + PORT + "/bookstore/books/details";

final Client client = ClientBuilder.newClient();
try (InputStream is = getClass()
.getResourceAsStream("/org/apache/cxf/systest/jaxrs/resources/attachmentData")) {
final MultipartBody builder = new MultipartBody(Arrays.asList(
new AttachmentBuilder()
.mediaType("application/xml")
.id("book")
.object(new BookWithValidation("Book", "book1"))
.build(),
new AttachmentBuilder()
.id("upfile1Detail")
.object(is)
.contentDisposition(new ContentDisposition("form-data; name=\"field1\";"))
.build(),
new AttachmentBuilder()
.id("upfile2Detail")
.dataHandler(new DataHandler(
new InputStreamDataSource(new ByteArrayInputStream(new byte[0]), "text/xml")))
.contentDisposition(new ContentDisposition("form-data; name=\"field2\";"))
.build(),
new AttachmentBuilder()
.id("upfile3Detail")
.dataHandler(new DataHandler(new InputStreamDataSource(
new ByteArrayInputStream(new byte[0]), "text/xml")))
.contentDisposition(new ContentDisposition("form-data; name=\"field3\";"))
.build()));

final Response response = client
.target(address)
.request("text/xml")
.post(Entity.entity(builder, "multipart/form-data"));

final BookWithValidation book = response.readEntity(BookWithValidation.class);
assertThat("Unexpected status code for response:" + response,
response.getStatus(), equalTo(200));
assertThat(book.getName(), equalTo("upfile1Detail,upfile2Detail,upfile3Detail"));
}
}

@Override
protected String getPort() {
return PORT;
}
}

Original file line number Diff line number Diff line change
@@ -0,0 +1,53 @@
/**
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
*/

package org.apache.cxf.systest.jaxrs.validation;

import jakarta.validation.Valid;
import jakarta.ws.rs.Consumes;
import jakarta.ws.rs.POST;
import jakarta.ws.rs.Path;
import jakarta.ws.rs.Produces;
import jakarta.ws.rs.WebApplicationException;
import jakarta.ws.rs.core.MediaType;
import jakarta.ws.rs.core.Response;
import org.apache.cxf.jaxrs.ext.multipart.Attachment;
import org.apache.cxf.jaxrs.ext.multipart.Multipart;

@Path("/bookstore")
public class MultipartBookStoreWithValidation {
@POST
@Path("/books/details")
@Consumes("multipart/form-data")
@Produces("text/xml")
public Response addBookWithDetails(
@Valid @Multipart(value = "book", type = "application/xml") BookWithValidation book,
@Multipart("upfile1Detail") Attachment a1,
@Multipart("upfile2Detail") Attachment a2,
@Multipart("upfile3Detail") Attachment a3)
throws Exception {

if (a1.equals(a2) || a1.equals(a3) || a2.equals(a3)) {
throw new WebApplicationException();
}

book.setName(a1.getContentId() + "," + a2.getContentId() + "," + a3.getContentId());
return Response.ok(book, MediaType.APPLICATION_XML_TYPE).build();
}
}

0 comments on commit 477ee89

Please sign in to comment.