aboutsummaryrefslogtreecommitdiffstats
path: root/uitest/src/main/java/com/vaadin/tests/resources/DownloadWithPush.java
blob: 13bb890d9b9002c1eb07d819c0b454ec0b183789 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
package com.vaadin.tests.resources;

import java.io.IOException;
import java.io.InputStream;

import com.vaadin.annotations.Push;
import com.vaadin.server.DownloadStream;
import com.vaadin.server.Resource;
import com.vaadin.server.StreamResource;
import com.vaadin.server.VaadinRequest;
import com.vaadin.tests.components.AbstractTestUIWithLog;
import com.vaadin.ui.Button;

@Push
public class DownloadWithPush extends AbstractTestUIWithLog {

    private static class GeneratedStream extends InputStream {
        int read = 0;
        int next = 'a';
        private final int size;

        public GeneratedStream(int size) {
            this.size = size;
        }

        @Override
        public int read() throws IOException {
            if (available() == 0) {
                return -1;
            }

            read++;
            next++;
            if (next > 'z') {
                next = 'a';
            }
            return next;
        }

        @Override
        public int available() throws IOException {
            return size - read;
        }
    }

    private final class DownloadResoure extends StreamResource {
        private DownloadResoure(StreamSource streamSource, String filename) {
            super(streamSource, filename);
        }

        @Override
        public DownloadStream getStream() {
            DownloadStream ds = super.getStream();
            ds.setParameter("Content-Disposition",
                    "attachment; filename=" + getFilename() + ";");
            return ds;
        }
    }

    private Resource hugeFileResource = createResource();;
    private int fileSize = 300 * (1024 * 1024);

    @Override
    protected void setup(VaadinRequest request) {
        Button b = new Button("Download a "
                + String.format("%.1f", fileSize / 1024.0 / 1024.0) + "MB file",
                event -> getUI().getPage().open(hugeFileResource, "", false));
        addComponent(b);
    }

    private Resource createResource() {
        Resource hugeFileResource = new DownloadResoure(
                () -> new GeneratedStream(fileSize), "hugefile.txt");
        return hugeFileResource;
    }

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

}