mirror of
https://github.com/valitydev/riak_test.git
synced 2024-11-07 00:55:21 +00:00
78 lines
3.0 KiB
Erlang
78 lines
3.0 KiB
Erlang
|
|
||
|
|
-module(verify_wm_timeouts).
|
||
|
|
-compile(export_all).
|
||
|
|
-include_lib("eunit/include/eunit.hrl").
|
||
|
|
|
||
|
|
confirm() ->
|
||
|
|
[Node] = rt:build_cluster(1),
|
||
|
|
rt:wait_until_pingable(Node),
|
||
|
|
|
||
|
|
lager:info("setting up initial data and loading remote code"),
|
||
|
|
_Prep = os:cmd(io_lib:format("curl -s -S ~s/buckets/foo/keys/bar "
|
||
|
|
"-X PUT -H 'content-type: text/plain' "
|
||
|
|
"-d 'foobarbaz\n'",
|
||
|
|
[rt:http_url(Node)])),
|
||
|
|
|
||
|
|
{_Mod, Bin, File} = code:get_object_code(?MODULE),
|
||
|
|
rpc:call(Node, code, load_binary, [?MODULE,
|
||
|
|
File, Bin]),
|
||
|
|
ok = rpc:call(Node, ?MODULE, setup_mocks, []),
|
||
|
|
|
||
|
|
lager:info("testing GET timeout"),
|
||
|
|
%%curl must be installed locally for this to work.
|
||
|
|
GET = os:cmd(io_lib:format("curl -s -S ~s/buckets/foo/keys/bar "
|
||
|
|
"-H 'x-riak-timeout: 100'",
|
||
|
|
[rt:http_url(Node)])),
|
||
|
|
?assertEqual(GET, "request timed out\n"),
|
||
|
|
|
||
|
|
lager:info("testing PUT timeout"),
|
||
|
|
PUT = os:cmd(io_lib:format("curl -s -S ~s/buckets/foo/keys/fnord "
|
||
|
|
"-X PUT -H 'content-type: text/plain' "
|
||
|
|
"-H 'x-riak-timeout: 100' "
|
||
|
|
"-d 'foobarbazdasdasd\n'",
|
||
|
|
[rt:http_url(Node)])),
|
||
|
|
?assertEqual(PUT, "request timed out\n"),
|
||
|
|
|
||
|
|
lager:info("testing DELETE timeout"),
|
||
|
|
DEL = os:cmd(io_lib:format("curl -s -S ~s/buckets/foo/keys/bar "
|
||
|
|
"-X DELETE -H 'x-riak-timeout: 100'",
|
||
|
|
[rt:http_url(Node)])),
|
||
|
|
?assertEqual(DEL, "request timed out\n"),
|
||
|
|
|
||
|
|
lager:info("testing invalid timeout value"),
|
||
|
|
INV = os:cmd(io_lib:format("curl -s -S ~s/buckets/foo/keys/bar "
|
||
|
|
"-H 'x-riak-timeout: asdasasdasd'",
|
||
|
|
[rt:http_url(Node)])),
|
||
|
|
?assertEqual(INV, "foobarbaz\n"),
|
||
|
|
|
||
|
|
lager:info("testing GET still works before long timeout"),
|
||
|
|
LGET = os:cmd(io_lib:format("curl -s -S ~s/buckets/foo/keys/bar "
|
||
|
|
"-H 'x-riak-timeout: 4000'",
|
||
|
|
[rt:http_url(Node)])),
|
||
|
|
?assertEqual(LGET, "foobarbaz\n"),
|
||
|
|
|
||
|
|
ok = rpc:call(Node, ?MODULE, unload_mocks, []),
|
||
|
|
pass.
|
||
|
|
|
||
|
|
setup_mocks() ->
|
||
|
|
error_logger:info_msg("Beginning setup_mocks"),
|
||
|
|
meck:new(riak_kv_get_fsm, [unstick, passthrough, no_link]),
|
||
|
|
meck:new(riak_kv_put_fsm, [unstick, passthrough, no_link]),
|
||
|
|
meck:expect(riak_kv_put_fsm, prepare,
|
||
|
|
fun delay/2),
|
||
|
|
meck:expect(riak_kv_get_fsm, prepare,
|
||
|
|
fun delay/2),
|
||
|
|
error_logger:info_msg("Installed mocks"),
|
||
|
|
ok.
|
||
|
|
|
||
|
|
delay(timeout, State) ->
|
||
|
|
error_logger:info_msg("Adding delay"),
|
||
|
|
timer:sleep(1000),
|
||
|
|
meck:passthrough([timeout, State]).
|
||
|
|
|
||
|
|
unload_mocks() ->
|
||
|
|
error_logger:info_msg("unloading mocks"),
|
||
|
|
meck:unload(),
|
||
|
|
error_logger:info_msg("mocks unloaded"),
|
||
|
|
ok.
|