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
|
use Mojo::Base -strict;
# Disable IPv6 and libev
BEGIN {
$ENV{MOJO_NO_IPV6} = 1;
$ENV{MOJO_REACTOR} = 'Mojo::Reactor::Poll';
}
# "On the count of three, you will awaken feeling refreshed,
# as if Futurama had never been canceled by idiots,
# then brought back by bigger idiots. One. Two."
use Test::More tests => 15;
use Mojolicious::Lite;
use Scalar::Util 'weaken';
use Test::Mojo;
# Emit "request" event early for multipart requests under "/upload"
hook after_build_tx => sub {
my $tx = shift;
weaken $tx;
$tx->req->content->on(
upgrade => sub {
$tx->emit('request') if $tx->req->url->path->contains('/upload');
}
);
};
# POST /upload/*
my %cache;
post '/upload/:id' => sub {
my $self = shift;
# First invocation, prepare streaming
my $id = $self->param('id');
$self->req->content->on(
part => sub {
my ($multi, $single) = @_;
$single->on(
body => sub {
my $single = shift;
return unless $single->headers->content_disposition =~ /my_file/;
$single->unsubscribe('read');
$single->on(read => sub { $cache{$id} .= pop });
}
);
}
);
return unless $self->req->is_finished;
# Second invocation, render response
$self->render(data => $cache{$id});
};
# GET /download/*
get '/download/:id' => sub {
my $self = shift;
$self->render(data => $cache{$self->param('id')});
};
my $t = Test::Mojo->new;
# POST /upload/23 (small upload)
$t->post_form_ok('/upload/23' => {my_file => {content => 'whatever'}})
->status_is(200)->content_is('whatever');
# GET /download/23 (small download)
$t->get_ok('/download/23')->status_is(200)->content_is('whatever');
# POST /upload/24 (big upload)
$t->post_form_ok('/upload/24' => {my_file => {content => '1234' x 131072}})
->status_is(200)->content_is('1234' x 131072);
# GET /download/24 (big download)
$t->get_ok('/download/24')->status_is(200)->content_is('1234' x 131072);
# GET /download/23 (small download again)
$t->get_ok('/download/23')->status_is(200)->content_is('whatever');
|
