test: remove unused things, refactor some functions

This commit is contained in:
Thales Macedo Garitezi 2023-08-16 15:50:01 -03:00
parent e8c73b06e1
commit c46b8de938
2 changed files with 2 additions and 34 deletions

View File

@ -23,18 +23,12 @@
-compile(export_all).
-compile(nowarn_export_all).
-define(NOW,
(calendar:system_time_to_rfc3339(erlang:system_time(millisecond), [{unit, millisecond}]))
).
-define(DS_SHARD, <<"local">>).
all() ->
emqx_common_test_helpers:all(?MODULE).
init_per_suite(Config) ->
%% avoid inter-suite flakiness...
application:stop(emqx),
application:stop(emqx_durable_storage),
TCApps = emqx_cth_suite:start(
app_specs(),
#{work_dir => ?config(priv_dir, Config)}
@ -232,12 +226,8 @@ receive_messages(0, Msgs) ->
receive_messages(Count, Msgs) ->
receive
{publish, Msg} ->
receive_messages(Count - 1, [Msg | Msgs]);
{deliver, _Topic, Msg} ->
receive_messages(Count - 1, [Msg | Msgs]);
_Other ->
receive_messages(Count, Msgs)
after 5000 ->
receive_messages(Count - 1, [Msg | Msgs])
after 5_000 ->
Msgs
end.

View File

@ -267,8 +267,6 @@ receive_messages(Count, Msgs) ->
receive
{publish, Msg} ->
receive_messages(Count - 1, [Msg | Msgs]);
{deliver, _Topic, Msg} ->
receive_messages(Count - 1, [Msg | Msgs]);
_Other ->
receive_messages(Count, Msgs)
after 5000 ->
@ -375,26 +373,6 @@ do_publish(Payloads = [_ | _], PublishFun, WaitForUnregister) ->
do_publish(Payload, PublishFun, WaitForUnregister) ->
do_publish([Payload], PublishFun, WaitForUnregister).
get_replay_messages(ReplayID) ->
DSShard = <<"local">>,
case emqx_ds_storage_layer:restore_iterator(DSShard, ReplayID) of
{ok, It} ->
do_get_replay_messages(It, []);
Error ->
error({"error restoring iterator", #{error => Error, replay_id => ReplayID}})
end.
do_get_replay_messages(It, Acc) ->
case emqx_ds_storage_layer:next(It) of
{value, Val, NewIt} ->
Msg = erlang:binary_to_term(Val),
do_get_replay_messages(NewIt, [Msg | Acc]);
none ->
{ok, lists:reverse(Acc)};
{error, Reason} ->
{error, Reason}
end.
%%--------------------------------------------------------------------
%% Test Cases
%%--------------------------------------------------------------------