TestWebb_UpDownload.java

package com.goebl.david;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.InputStream;
import java.util.Random;
import org.json.JSONArray;

public class TestWebb_UpDownload extends AbstractTestWebb {
    private static File TEST_FILE;
    private boolean testFileCreated;

    // @BeforeClass
    public static void createTestFile() throws Exception {
        if (TEST_FILE == null) {
            File dir = new File(
                    isAndroid()
                            ? System.getProperty("CACHE_DIR")
                            : System.getProperty("java.io.tmpdir"));
            TEST_FILE = new File(dir, "test-upload.tmp");
        }
        FileOutputStream fos = new FileOutputStream(TEST_FILE);
        byte[] bytes = new byte[1000];
        for (int i = 0; i < bytes.length; ++i) {
            bytes[i] = (byte) 64;
        }
        for (int i = 0; i < 5; ++i) {
            fos.write(bytes);
        }
        fos.flush();
        fos.close();
    }

    // @AfterClass
    public static void unlinkTestFile() {
        TEST_FILE.delete();
    }

    @Override
    protected void tearDown() throws Exception {
        if (testFileCreated) {
            unlinkTestFile();
        }

        super.tearDown();
    }

    public void testUploadFile() throws Exception {
        createTestFile();
        testFileCreated = true;

        Response<Void> response = webb
                .post("/upload?file")
                .body(TEST_FILE)
                .asVoid();

        assertEquals(201, response.getStatusCode());
    }

    public void testUploadStream() throws Exception {
        createTestFile();
        testFileCreated = true;

        InputStream inputStream = new FileInputStream(TEST_FILE);

        Response<Void> response = webb
                .post("/upload?stream")
                .body(inputStream)
                .asVoid();

        inputStream.close();

        assertEquals(201, response.getStatusCode());
    }

    public void testUploadCompressedStream() throws Exception {
        createTestFile();
        testFileCreated = true;

        Response<Void> response = webb
                .post("/upload-compressed")
                .compress()
                .body(TEST_FILE)
                .asVoid();

        assertEquals(201, response.getStatusCode());
    }

    public void testEchoCompressedBytes() throws Exception {
        byte[] payload = new byte[5000];
        new Random().nextBytes(payload);

        Response<byte[]> response = webb
                .post("/echoBin") //force-content-encoding
                .header(Webb.HDR_ACCEPT_ENCODING, "gzip")
                .compress()
                .body(payload)
                .asBytes();

        assertEquals(200, response.getStatusCode());
        byte[] echoed = response.getBody();
        assertNotNull(echoed);
        assertArrayEquals(payload, echoed);
    }

    public void testCompressedRandomBytes() throws Exception {
        // random bytes cannot be compressed efficiently, so there should be a fallback to 'identity'
        byte[] payload = new byte[5000];
        new Random().nextBytes(payload);

        Response<byte[]> response = webb
                .post("/echoBin?force-content-encoding=identity")
                .compress()
                .body(payload)
                .asBytes();

        assertEquals(200, response.getStatusCode());
        byte[] echoed = response.getBody();
        assertNotNull(echoed);
        assertArrayEquals(payload, echoed);
    }

    public void testCompressedSqueezableBytes() throws Exception {
        byte[] payload = new byte[5000];
        for (int i = 0; i < payload.length; ++i) {
            payload[i] = (byte) (0xFF & (i / 100));
        }

        Response<byte[]> response = webb
                .post("/echoBin?force-content-encoding=identity")
                .compress()
                .body(payload)
                .asBytes();

        assertEquals(200, response.getStatusCode());
        byte[] echoed = response.getBody();
        assertNotNull(echoed);
        assertTrue(payload.length > echoed.length);

        byte[] gunzip = TestWebbUtils_NoMock.gunzip(echoed);
        assertArrayEquals(payload, gunzip);
    }

    public void testDownloadGzip() throws Exception {

        Response<JSONArray> response = webb
                .get("/compressed.json")
                .header(Webb.HDR_ACCEPT_ENCODING, "gzip")
                .asJsonArray();

        assertEquals(200, response.getStatusCode());
        assertEquals(500, response.getBody().length());
    }

    public void testDownloadDeflate() throws Exception {

        Response<JSONArray> response = webb
                .get("/compressed.json")
                .header(Webb.HDR_ACCEPT_ENCODING, "deflate")
                .asJsonArray();

        assertEquals(200, response.getStatusCode());
        assertEquals(500, response.getBody().length());
    }

    public void testDownloadUnknownEncoding() throws Exception {
        Request request = webb
                .get("/compressed.json")
                .header(Webb.HDR_ACCEPT_ENCODING, "unknown");

        try {
            Response<byte[]> response = request.asBytes();
        } catch (WebbException expected) {
            assertTrue(expected.getMessage().matches("^unsupported.*encoding.*$"));
            return;
        }
        fail("unknown encoding not detected");
    }

    public void testEchoBytes() throws Exception {
        byte[] msg = (SIMPLE_ASCII + ", " + COMPLEX_UTF8).getBytes(Const.UTF8);
        Response<byte[]> response = webb
                .post("/echoBin")
                .body(msg)
                .asBytes();

        assertArrayEquals(msg, response.getBody());
    }

    public void testEchoAsStream() throws Exception {
        byte[] msg = (SIMPLE_ASCII + ", " + COMPLEX_UTF8).getBytes(Const.UTF8);
        Response<InputStream> response = webb
                .post("/echoBin")
                .body(msg)
                .asStream();
        InputStream is = response.getBody();
        try {
            byte[] result = WebbUtils.readBytes(is);
            assertArrayEquals(msg, result);
        } finally {
            is.close();
        }
    }
}