diff options
Diffstat (limited to 'test/handlers')
-rw-r--r-- | test/handlers/compress_h.erl | 25 | ||||
-rw-r--r-- | test/handlers/resp_h.erl | 17 |
2 files changed, 41 insertions, 1 deletions
diff --git a/test/handlers/compress_h.erl b/test/handlers/compress_h.erl index 4b4dbe7..1509d82 100644 --- a/test/handlers/compress_h.erl +++ b/test/handlers/compress_h.erl @@ -27,7 +27,30 @@ init(Req0, State=stream_reply) -> <<"large">> -> stream_reply(#{}, Req0); <<"content-encoding">> -> - stream_reply(#{<<"content-encoding">> => <<"compress">>}, Req0) + stream_reply(#{<<"content-encoding">> => <<"compress">>}, Req0); + <<"sendfile">> -> + Data = lists:duplicate(10000, $a), + AppFile = code:where_is_file("cowboy.app"), + Size = filelib:file_size(AppFile), + Req1 = cowboy_req:stream_reply(200, Req0), + %% We send a few files interspersed into other data. + cowboy_req:stream_body(Data, nofin, Req1), + cowboy_req:stream_body({sendfile, 0, Size, AppFile}, nofin, Req1), + cowboy_req:stream_body(Data, nofin, Req1), + cowboy_req:stream_body({sendfile, 0, Size, AppFile}, nofin, Req1), + cowboy_req:stream_body(Data, fin, Req1), + Req1; + <<"sendfile_fin">> -> + Data = lists:duplicate(10000, $a), + AppFile = code:where_is_file("cowboy.app"), + Size = filelib:file_size(AppFile), + Req1 = cowboy_req:stream_reply(200, Req0), + %% We send a few files interspersed into other data. + cowboy_req:stream_body(Data, nofin, Req1), + cowboy_req:stream_body({sendfile, 0, Size, AppFile}, nofin, Req1), + cowboy_req:stream_body(Data, nofin, Req1), + cowboy_req:stream_body({sendfile, 0, Size, AppFile}, fin, Req1), + Req1 end, {ok, Req, State}. diff --git a/test/handlers/resp_h.erl b/test/handlers/resp_h.erl index bfcb95c..7a80e75 100644 --- a/test/handlers/resp_h.erl +++ b/test/handlers/resp_h.erl @@ -219,6 +219,23 @@ do(<<"stream_body">>, Req0, Opts) -> Req = cowboy_req:stream_reply(200, Req0), cowboy_req:stream_body(<<"Hello world!">>, nofin, Req), {ok, Req, Opts}; + <<"sendfile">> -> + AppFile = code:where_is_file("cowboy.app"), + AppSize = filelib:file_size(AppFile), + Req = cowboy_req:stream_reply(200, Req0), + cowboy_req:stream_body(<<"Hello ">>, nofin, Req), + cowboy_req:stream_body({sendfile, 0, AppSize, AppFile}, nofin, Req), + cowboy_req:stream_body(<<" interspersed ">>, nofin, Req), + cowboy_req:stream_body({sendfile, 0, AppSize, AppFile}, nofin, Req), + cowboy_req:stream_body(<<" world!">>, fin, Req), + {ok, Req, Opts}; + <<"sendfile_fin">> -> + AppFile = code:where_is_file("cowboy.app"), + AppSize = filelib:file_size(AppFile), + Req = cowboy_req:stream_reply(200, Req0), + cowboy_req:stream_body(<<"Hello! ">>, nofin, Req), + cowboy_req:stream_body({sendfile, 0, AppSize, AppFile}, fin, Req), + {ok, Req, Opts}; _ -> %% Call stream_body without initiating streaming. cowboy_req:stream_body(<<0:800000>>, fin, Req0), |