MyResourceTest.java

package org.javaee7.jaxrs.fileupload;

import static org.assertj.core.api.Assertions.assertThat;

import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.net.MalformedURLException;
import java.net.URI;
import java.net.URL;

import javax.ws.rs.client.Client;
import javax.ws.rs.client.ClientBuilder;
import javax.ws.rs.client.Entity;
import javax.ws.rs.client.WebTarget;
import javax.ws.rs.core.MediaType;
import javax.ws.rs.core.Response;
import javax.ws.rs.core.Response.Status;

import org.assertj.core.api.Condition;
import org.jboss.arquillian.container.test.api.Deployment;
import org.jboss.arquillian.junit.Arquillian;
import org.jboss.arquillian.test.api.ArquillianResource;
import org.jboss.shrinkwrap.api.ShrinkWrap;
import org.jboss.shrinkwrap.api.spec.WebArchive;
import org.junit.Before;
import org.junit.BeforeClass;
import org.junit.Test;
import org.junit.runner.RunWith;

/**
 * @author Arun Gupta
 * @author Xavier Coulon
 */
@RunWith(Arquillian.class)
public class MyResourceTest {

    @Deployment(testable = false)
    public static WebArchive createDeployment() {
        return ShrinkWrap.create(WebArchive.class).addClasses(MyApplication.class, MyResource.class);
    }

    private static WebTarget target;

    private static File tempFile;
    @ArquillianResource
    private URL base;

    @BeforeClass
    public static void generateSampleFile() throws IOException {
        tempFile = File.createTempFile("javaee7samples", ".png");
        // fill the file with 1KB of content
        try (FileOutputStream outputStream = new FileOutputStream(tempFile)) {
            for (int i = 0; i < 1000; i++) {
                outputStream.write(0);
            }
        }
        assertThat(tempFile).canRead().has(new Condition<File>() {

            @Override
            public boolean matches(File tempFile) {
                return tempFile.length() == 1000;
            }
        });
    }

    @Before
    public void setUpClass() throws MalformedURLException {
        Client client = ClientBuilder.newClient();
        target = client.target(URI.create(new URL(base, "webresources/endpoint").toExternalForm()));
    }

    @Test
    public void shouldPostOctetStreamContentAsInputStream() {
        // when
        Long uploadedFileSize = target.path("/upload").request()
            .post(Entity.entity(tempFile, MediaType.APPLICATION_OCTET_STREAM), Long.class);
        // then
        assertThat(uploadedFileSize).isEqualTo(1000);
    }

    @Test
    public void shouldNotPostImagePngContentAsInputStream() {
        // when
        final Response response = target.path("/upload").request().post(Entity.entity(tempFile, "image/png"));
        // then
        assertThat(response.getStatus()).isEqualTo(Status.UNSUPPORTED_MEDIA_TYPE.getStatusCode());
    }

    @Test
    public void shouldPostOctetStreamContentAsFile() {
        // when
        Long uploadedFileSize = target.path("/upload2").request()
            .post(Entity.entity(tempFile, MediaType.APPLICATION_OCTET_STREAM), Long.class);
        // then
        assertThat(uploadedFileSize).isEqualTo(1000);
    }

    @Test
    public void shouldPostImagePngContentAsFile() {
        // when
        Long uploadedFileSize = target.path("/upload2").request()
            .post(Entity.entity(tempFile, "image/png"), Long.class);
        // then
        assertThat(uploadedFileSize).isEqualTo(1000);
    }

}