TestFileUploadSize.java

package com.vaadin.tests.components.upload;

import java.io.ByteArrayOutputStream;
import java.io.OutputStream;

import com.vaadin.shared.ui.MarginInfo;
import com.vaadin.tests.components.TestBase;
import com.vaadin.ui.GridLayout;
import com.vaadin.ui.Label;
import com.vaadin.ui.Notification;
import com.vaadin.ui.Upload;
import com.vaadin.ui.Upload.FinishedEvent;
import com.vaadin.ui.Upload.Receiver;
import com.vaadin.ui.Upload.StartedEvent;

public class TestFileUploadSize extends TestBase implements Receiver {

    private Label label = new Label("No finished uploads.");
    private Label receivedSize = new Label("-");
    private Label expectedSize = new Label("-");

    ByteArrayOutputStream baos = new ByteArrayOutputStream();

    @Override
    protected void setup() {
        getLayout().setMargin(new MarginInfo(true, false, false, false));
        getLayout().setSpacing(true);

        Upload u = new Upload("Upload", new Upload.Receiver() {

            @Override
            public OutputStream receiveUpload(String filename,
                    String mimeType) {
                return baos;
            }
        });
        u.setId("UPL");
        u.addStartedListener(new Upload.StartedListener() {

            @Override
            public void uploadStarted(StartedEvent event) {
                expectedSize.setValue(String.valueOf(event.getContentLength()));
            }
        });
        u.addFinishedListener(new Upload.FinishedListener() {

            @Override
            public void uploadFinished(FinishedEvent event) {
                label.setValue("Upload finished. Name: " + event.getFilename());
                receivedSize.setValue(String.valueOf(baos.size()));
                baos.reset();
            }
        });

        expectedSize.setId("expected");
        receivedSize.setId("received");

        GridLayout grid = new GridLayout(2, 2);
        grid.addComponent(new Label("Expected size:"), 0, 0);
        grid.addComponent(new Label("Received size:"), 0, 1);
        grid.addComponent(expectedSize, 1, 0);
        grid.addComponent(receivedSize, 1, 1);

        addComponent(label);
        addComponent(grid);
        addComponent(u);
    }

    @Override
    public OutputStream receiveUpload(String filename, String MIMEType) {
        Notification.show("Receiving upload");
        return new ByteArrayOutputStream();
    }

    @Override
    protected Integer getTicketNumber() {
        return 9548;
    }

    @Override
    protected String getDescription() {
        return "Multibyte characters in filenames should not cause the upload size to be computed incorrectly";
    }

}