aboutsummaryrefslogtreecommitdiffstats
path: root/test
diff options
context:
space:
mode:
authorLoïc Hoguin <[email protected]>2018-11-15 18:53:42 +0100
committerLoïc Hoguin <[email protected]>2018-11-15 18:53:42 +0100
commit240da3f2d9bc02611b23c1ea0e7bbe8db9d99cd8 (patch)
tree7c59e4ebd16731bab4fd9ac5deddb55a7762cf1c /test
parentfbfec873f6026b858c3604d74d88470ce45f4296 (diff)
downloadcowboy-240da3f2d9bc02611b23c1ea0e7bbe8db9d99cd8.tar.gz
cowboy-240da3f2d9bc02611b23c1ea0e7bbe8db9d99cd8.tar.bz2
cowboy-240da3f2d9bc02611b23c1ea0e7bbe8db9d99cd8.zip
Add the set_options stream handler command
The first two options to benefit from this are the cowboy_compress_h options.
Diffstat (limited to 'test')
-rw-r--r--test/compress_SUITE.erl133
-rw-r--r--test/handlers/compress_h.erl34
2 files changed, 132 insertions, 35 deletions
diff --git a/test/compress_SUITE.erl b/test/compress_SUITE.erl
index 052d82d..1b82cc6 100644
--- a/test/compress_SUITE.erl
+++ b/test/compress_SUITE.erl
@@ -149,53 +149,132 @@ gzip_stream_reply_content_encoding(Config) ->
opts_compress_buffering_false(Config0) ->
doc("Confirm that the compress_buffering option can be set to false, "
"which is the default."),
+ Name = name(),
Fun = case config(ref, Config0) of
https_compress -> init_https;
h2_compress -> init_http2;
_ -> init_http
end,
- Config = cowboy_test:Fun(name(), #{
+ Config = cowboy_test:Fun(Name, #{
env => #{dispatch => init_dispatch(Config0)},
stream_handlers => [cowboy_compress_h, cowboy_stream_h],
compress_buffering => false
}, Config0),
- ConnPid = gun_open(Config),
- Ref = gun:get(ConnPid, "/stream_reply/delayed",
- [{<<"accept-encoding">>, <<"gzip">>}]),
- {response, nofin, 200, Headers} = gun:await(ConnPid, Ref),
- {_, <<"gzip">>} = lists:keyfind(<<"content-encoding">>, 1, Headers),
- Z = zlib:open(),
- zlib:inflateInit(Z, 31),
- {data, nofin, Data1} = gun:await(ConnPid, Ref, 100),
- <<"data: Hello!\r\n\r\n">> = iolist_to_binary(zlib:inflate(Z, Data1)),
- timer:sleep(1000),
- {data, nofin, Data2} = gun:await(ConnPid, Ref, 100),
- <<"data: World!\r\n\r\n">> = iolist_to_binary(zlib:inflate(Z, Data2)),
- gun:close(ConnPid),
- cowboy:stop_listener(name()).
+ try
+ ConnPid = gun_open(Config),
+ Ref = gun:get(ConnPid, "/stream_reply/delayed",
+ [{<<"accept-encoding">>, <<"gzip">>}]),
+ {response, nofin, 200, Headers} = gun:await(ConnPid, Ref),
+ {_, <<"gzip">>} = lists:keyfind(<<"content-encoding">>, 1, Headers),
+ Z = zlib:open(),
+ zlib:inflateInit(Z, 31),
+ {data, nofin, Data1} = gun:await(ConnPid, Ref, 100),
+ <<"data: Hello!\r\n\r\n">> = iolist_to_binary(zlib:inflate(Z, Data1)),
+ timer:sleep(1000),
+ {data, nofin, Data2} = gun:await(ConnPid, Ref, 100),
+ <<"data: World!\r\n\r\n">> = iolist_to_binary(zlib:inflate(Z, Data2)),
+ gun:close(ConnPid)
+ after
+ cowboy:stop_listener(Name)
+ end.
opts_compress_buffering_true(Config0) ->
doc("Confirm that the compress_buffering option can be set to true, "
"and that the data received is buffered."),
+ Name = name(),
Fun = case config(ref, Config0) of
https_compress -> init_https;
h2_compress -> init_http2;
_ -> init_http
end,
- Config = cowboy_test:Fun(name(), #{
+ Config = cowboy_test:Fun(Name, #{
env => #{dispatch => init_dispatch(Config0)},
stream_handlers => [cowboy_compress_h, cowboy_stream_h],
compress_buffering => true
}, Config0),
- ConnPid = gun_open(Config),
- Ref = gun:get(ConnPid, "/stream_reply/delayed",
- [{<<"accept-encoding">>, <<"gzip">>}]),
- {response, nofin, 200, Headers} = gun:await(ConnPid, Ref),
+ try
+ ConnPid = gun_open(Config),
+ Ref = gun:get(ConnPid, "/stream_reply/delayed",
+ [{<<"accept-encoding">>, <<"gzip">>}]),
+ {response, nofin, 200, Headers} = gun:await(ConnPid, Ref),
+ {_, <<"gzip">>} = lists:keyfind(<<"content-encoding">>, 1, Headers),
+ Z = zlib:open(),
+ zlib:inflateInit(Z, 31),
+ %% The data gets buffered because it is too small.
+ {data, nofin, Data1} = gun:await(ConnPid, Ref, 100),
+ <<>> = iolist_to_binary(zlib:inflate(Z, Data1)),
+ gun:close(ConnPid)
+ after
+ cowboy:stop_listener(Name)
+ end.
+
+set_options_compress_buffering_false(Config0) ->
+ doc("Confirm that the compress_buffering option can be dynamically "
+ "set to false by a handler and that the data received is not buffered."),
+ Name = name(),
+ Fun = case config(ref, Config0) of
+ https_compress -> init_https;
+ h2_compress -> init_http2;
+ _ -> init_http
+ end,
+ Config = cowboy_test:Fun(Name, #{
+ env => #{dispatch => init_dispatch(Config0)},
+ stream_handlers => [cowboy_compress_h, cowboy_stream_h],
+ compress_buffering => true
+ }, Config0),
+ try
+ ConnPid = gun_open(Config),
+ Ref = gun:get(ConnPid, "/stream_reply/set_options_buffering_false",
+ [{<<"accept-encoding">>, <<"gzip">>}]),
+ {response, nofin, 200, Headers} = gun:await(ConnPid, Ref),
+ {_, <<"gzip">>} = lists:keyfind(<<"content-encoding">>, 1, Headers),
+ Z = zlib:open(),
+ zlib:inflateInit(Z, 31),
+ {data, nofin, Data1} = gun:await(ConnPid, Ref, 100),
+ <<"data: Hello!\r\n\r\n">> = iolist_to_binary(zlib:inflate(Z, Data1)),
+ timer:sleep(1000),
+ {data, nofin, Data2} = gun:await(ConnPid, Ref, 100),
+ <<"data: World!\r\n\r\n">> = iolist_to_binary(zlib:inflate(Z, Data2)),
+ gun:close(ConnPid)
+ after
+ cowboy:stop_listener(Name)
+ end.
+
+set_options_compress_buffering_true(Config0) ->
+ doc("Confirm that the compress_buffering option can be dynamically "
+ "set to true by a handler and that the data received is buffered."),
+ Name = name(),
+ Fun = case config(ref, Config0) of
+ https_compress -> init_https;
+ h2_compress -> init_http2;
+ _ -> init_http
+ end,
+ Config = cowboy_test:Fun(Name, #{
+ env => #{dispatch => init_dispatch(Config0)},
+ stream_handlers => [cowboy_compress_h, cowboy_stream_h],
+ compress_buffering => false
+ }, Config0),
+ try
+ ConnPid = gun_open(Config),
+ Ref = gun:get(ConnPid, "/stream_reply/set_options_buffering_true",
+ [{<<"accept-encoding">>, <<"gzip">>}]),
+ {response, nofin, 200, Headers} = gun:await(ConnPid, Ref),
+ {_, <<"gzip">>} = lists:keyfind(<<"content-encoding">>, 1, Headers),
+ Z = zlib:open(),
+ zlib:inflateInit(Z, 31),
+ %% The data gets buffered because it is too small.
+ {data, nofin, Data1} = gun:await(ConnPid, Ref, 100),
+ <<>> = iolist_to_binary(zlib:inflate(Z, Data1)),
+ gun:close(ConnPid)
+ after
+ cowboy:stop_listener(Name)
+ end.
+
+set_options_compress_threshold_0(Config) ->
+ doc("Confirm that the compress_threshold option can be dynamically "
+ "set to change how large response bodies must be to be compressed."),
+ {200, Headers, GzBody} = do_get("/reply/set_options_threshold0",
+ [{<<"accept-encoding">>, <<"gzip">>}], Config),
{_, <<"gzip">>} = lists:keyfind(<<"content-encoding">>, 1, Headers),
- Z = zlib:open(),
- zlib:inflateInit(Z, 31),
- %% The data gets buffered because it is too small.
- {data, nofin, Data1} = gun:await(ConnPid, Ref, 100),
- <<>> = iolist_to_binary(zlib:inflate(Z, Data1)),
- gun:close(ConnPid),
- cowboy:stop_listener(name()).
+ _ = zlib:gunzip(GzBody),
+ ok.
diff --git a/test/handlers/compress_h.erl b/test/handlers/compress_h.erl
index ffea05f..32830d9 100644
--- a/test/handlers/compress_h.erl
+++ b/test/handlers/compress_h.erl
@@ -19,7 +19,12 @@ init(Req0, State=reply) ->
<<"sendfile">> ->
AppFile = code:where_is_file("cowboy.app"),
Size = filelib:file_size(AppFile),
- cowboy_req:reply(200, #{}, {sendfile, 0, Size, AppFile}, Req0)
+ cowboy_req:reply(200, #{}, {sendfile, 0, Size, AppFile}, Req0);
+ <<"set_options_threshold0">> ->
+ %% @todo This should be replaced by a cowboy_req:cast/cowboy_stream:cast.
+ #{pid := Pid, streamid := StreamID} = Req0,
+ Pid ! {{Pid, StreamID}, {set_options, #{compress_threshold => 0}}},
+ cowboy_req:reply(200, #{}, lists:duplicate(100, $a), Req0)
end,
{ok, Req, State};
init(Req0, State=stream_reply) ->
@@ -52,13 +57,17 @@ init(Req0, State=stream_reply) ->
cowboy_req:stream_body({sendfile, 0, Size, AppFile}, fin, Req1),
Req1;
<<"delayed">> ->
- Req1 = cowboy_req:stream_reply(200, Req0),
- cowboy_req:stream_body(<<"data: Hello!\r\n\r\n">>, nofin, Req1),
- timer:sleep(1000),
- cowboy_req:stream_body(<<"data: World!\r\n\r\n">>, nofin, Req1),
- timer:sleep(1000),
- cowboy_req:stream_body(<<"data: Closing!\r\n\r\n">>, fin, Req1),
- Req1
+ stream_delayed(Req0);
+ <<"set_options_buffering_false">> ->
+ %% @todo This should be replaced by a cowboy_req:cast/cowboy_stream:cast.
+ #{pid := Pid, streamid := StreamID} = Req0,
+ Pid ! {{Pid, StreamID}, {set_options, #{compress_buffering => false}}},
+ stream_delayed(Req0);
+ <<"set_options_buffering_true">> ->
+ %% @todo This should be replaced by a cowboy_req:cast/cowboy_stream:cast.
+ #{pid := Pid, streamid := StreamID} = Req0,
+ Pid ! {{Pid, StreamID}, {set_options, #{compress_buffering => true}}},
+ stream_delayed(Req0)
end,
{ok, Req, State}.
@@ -68,3 +77,12 @@ stream_reply(Headers, Req0) ->
_ = [cowboy_req:stream_body(Data, nofin, Req) || _ <- lists:seq(1,9)],
cowboy_req:stream_body(Data, fin, Req),
Req.
+
+stream_delayed(Req0) ->
+ Req = cowboy_req:stream_reply(200, Req0),
+ cowboy_req:stream_body(<<"data: Hello!\r\n\r\n">>, nofin, Req),
+ timer:sleep(1000),
+ cowboy_req:stream_body(<<"data: World!\r\n\r\n">>, nofin, Req),
+ timer:sleep(1000),
+ cowboy_req:stream_body(<<"data: Closing!\r\n\r\n">>, fin, Req),
+ Req.