777 lines
32 KiB
Erlang
777 lines
32 KiB
Erlang
%%--------------------------------------------------------------------
|
|
%% Copyright (c) 2020 EMQ Technologies Co., Ltd. All Rights Reserved.
|
|
%%
|
|
%% Licensed under the Apache License, Version 2.0 (the "License");
|
|
%% you may not use this file except in compliance with the License.
|
|
%% You may obtain a copy of the License at
|
|
%%
|
|
%% http://www.apache.org/licenses/LICENSE-2.0
|
|
%%
|
|
%% Unless required by applicable law or agreed to in writing, software
|
|
%% distributed under the License is distributed on an "AS IS" BASIS,
|
|
%% WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
%% See the License for the specific language governing permissions and
|
|
%% limitations under the License.
|
|
%%--------------------------------------------------------------------
|
|
|
|
-module(mqtt_protocol_v5_SUITE).
|
|
|
|
-compile(export_all).
|
|
-compile(nowarn_export_all).
|
|
|
|
-include_lib("emqx/include/emqx.hrl").
|
|
-include_lib("emqx/include/emqx_mqtt.hrl").
|
|
-include_lib("eunit/include/eunit.hrl").
|
|
|
|
-import(lists, [nth/2]).
|
|
|
|
-define(TOPICS, [<<"TopicA">>, <<"TopicA/B">>, <<"Topic/C">>, <<"TopicA/C">>,
|
|
<<"/TopicA">>]).
|
|
|
|
-define(WILD_TOPICS, [<<"TopicA/+">>, <<"+/C">>, <<"#">>, <<"/#">>, <<"/+">>,
|
|
<<"+/+">>, <<"TopicA/#">>]).
|
|
|
|
all() -> emqx_ct:all(?MODULE).
|
|
|
|
init_per_suite(Config) ->
|
|
%% Meck emqtt
|
|
ok = meck:new(emqtt, [non_strict, passthrough, no_history, no_link]),
|
|
%% Start Apps
|
|
emqx_ct_helpers:start_apps([]),
|
|
Config.
|
|
|
|
end_per_suite(_Config) ->
|
|
ok = meck:unload(emqtt),
|
|
emqx_ct_helpers:stop_apps([]).
|
|
|
|
%%--------------------------------------------------------------------
|
|
%% Helpers
|
|
%%--------------------------------------------------------------------
|
|
|
|
client_info(Key, Client) ->
|
|
maps:get(Key, maps:from_list(emqtt:info(Client)), undefined).
|
|
|
|
receive_messages(Count) ->
|
|
receive_messages(Count, []).
|
|
|
|
receive_messages(0, Msgs) ->
|
|
Msgs;
|
|
receive_messages(Count, Msgs) ->
|
|
receive
|
|
{publish, Msg} ->
|
|
receive_messages(Count-1, [Msg|Msgs]);
|
|
_Other ->
|
|
receive_messages(Count, Msgs)
|
|
after 1000 ->
|
|
Msgs
|
|
end.
|
|
|
|
receive_disconnect_reasoncode() ->
|
|
receive
|
|
{disconnected, ReasonCode, _} -> ReasonCode;
|
|
_Other -> receive_disconnect_reasoncode()
|
|
after 100 ->
|
|
error("no disconnect packet")
|
|
end.
|
|
|
|
waiting_client_process_exit(C) ->
|
|
receive
|
|
{'EXIT', C, Reason} -> Reason;
|
|
_Oth -> error({got_another_message, _Oth})
|
|
after
|
|
1000 -> error({waiting_timeout, C})
|
|
end.
|
|
|
|
clean_retained(Topic) ->
|
|
{ok, Clean} = emqtt:start_link([{clean_start, true}]),
|
|
{ok, _} = emqtt:connect(Clean),
|
|
{ok, _} = emqtt:publish(Clean, Topic, #{}, <<"">>, [{qos, ?QOS_1}, {retain, true}]),
|
|
ok = emqtt:disconnect(Clean).
|
|
|
|
%%--------------------------------------------------------------------
|
|
%% Test Cases
|
|
%%--------------------------------------------------------------------
|
|
|
|
t_basic_test(_) ->
|
|
Topic = nth(1, ?TOPICS),
|
|
ct:print("Basic test starting"),
|
|
{ok, C} = emqtt:start_link([{proto_ver, v5}]),
|
|
{ok, _} = emqtt:connect(C),
|
|
{ok, _, [1]} = emqtt:subscribe(C, Topic, qos1),
|
|
{ok, _, [2]} = emqtt:subscribe(C, Topic, qos2),
|
|
{ok, _} = emqtt:publish(C, Topic, <<"qos 2">>, 2),
|
|
{ok, _} = emqtt:publish(C, Topic, <<"qos 2">>, 2),
|
|
{ok, _} = emqtt:publish(C, Topic, <<"qos 2">>, 2),
|
|
?assertEqual(3, length(receive_messages(3))),
|
|
ok = emqtt:disconnect(C).
|
|
|
|
%%--------------------------------------------------------------------
|
|
%% Connection
|
|
%%--------------------------------------------------------------------
|
|
|
|
t_connect_clean_start(_) ->
|
|
process_flag(trap_exit, true),
|
|
{ok, Client1} = emqtt:start_link([{clientid, <<"t_connect_clean_start">>},
|
|
{proto_ver, v5},{clean_start, true}]),
|
|
{ok, _} = emqtt:connect(Client1),
|
|
?assertEqual(0, client_info(session_present, Client1)), %% [MQTT-3.1.2-4]
|
|
ok = emqtt:pause(Client1),
|
|
{ok, Client2} = emqtt:start_link([{clientid, <<"t_connect_clean_start">>},
|
|
{proto_ver, v5},{clean_start, false}]),
|
|
{ok, _} = emqtt:connect(Client2),
|
|
?assertEqual(1, client_info(session_present, Client2)), %% [MQTT-3.1.2-5]
|
|
?assertEqual(142, receive_disconnect_reasoncode()),
|
|
waiting_client_process_exit(Client1),
|
|
|
|
ok = emqtt:disconnect(Client2),
|
|
waiting_client_process_exit(Client2),
|
|
|
|
{ok, Client3} = emqtt:start_link([{clientid, <<"new_client">>},
|
|
{proto_ver, v5},{clean_start, false}]),
|
|
{ok, _} = emqtt:connect(Client3),
|
|
?assertEqual(0, client_info(session_present, Client3)), %% [MQTT-3.1.2-6]
|
|
ok = emqtt:disconnect(Client3),
|
|
waiting_client_process_exit(Client3),
|
|
|
|
process_flag(trap_exit, false).
|
|
|
|
t_connect_will_message(_) ->
|
|
Topic = nth(1, ?TOPICS),
|
|
Payload = "will message",
|
|
|
|
{ok, Client1} = emqtt:start_link([
|
|
{proto_ver, v5},
|
|
{clean_start, true},
|
|
{will_flag, true},
|
|
{will_topic, Topic},
|
|
{will_payload, Payload}
|
|
]),
|
|
{ok, _} = emqtt:connect(Client1),
|
|
[ClientPid] = emqx_cm:lookup_channels(client_info(clientid, Client1)),
|
|
Info = emqx_connection:info(sys:get_state(ClientPid)),
|
|
?assertNotEqual(undefined, maps:find(will_msg, Info)), %% [MQTT-3.1.2-7]
|
|
|
|
{ok, Client2} = emqtt:start_link([{proto_ver, v5}]),
|
|
{ok, _} = emqtt:connect(Client2),
|
|
{ok, _, [2]} = emqtt:subscribe(Client2, Topic, qos2),
|
|
|
|
ok = emqtt:disconnect(Client1, 4), %% [MQTT-3.14.2-1]
|
|
[Msg | _ ] = receive_messages(1),
|
|
%% [MQTT-3.1.2-8]
|
|
?assertEqual({ok, iolist_to_binary(Topic)}, maps:find(topic, Msg)),
|
|
?assertEqual({ok, iolist_to_binary(Payload)}, maps:find(payload, Msg)),
|
|
?assertEqual({ok, 0}, maps:find(qos, Msg)),
|
|
ok = emqtt:disconnect(Client2),
|
|
|
|
{ok, Client3} = emqtt:start_link([
|
|
{proto_ver, v5},
|
|
{clean_start, true},
|
|
{will_flag, true},
|
|
{will_topic, Topic},
|
|
{will_payload, Payload}
|
|
]),
|
|
{ok, _} = emqtt:connect(Client3),
|
|
|
|
{ok, Client4} = emqtt:start_link([{proto_ver, v5}]),
|
|
{ok, _} = emqtt:connect(Client4),
|
|
{ok, _, [2]} = emqtt:subscribe(Client4, Topic, qos2),
|
|
ok = emqtt:disconnect(Client3),
|
|
?assertEqual(0, length(receive_messages(1))), %% [MQTT-3.1.2-10]
|
|
ok = emqtt:disconnect(Client4).
|
|
|
|
t_batch_subscribe(_) ->
|
|
{ok, Client} = emqtt:start_link([{proto_ver, v5}, {clientid, <<"batch_test">>}]),
|
|
{ok, _} = emqtt:connect(Client),
|
|
application:set_env(emqx, enable_acl_cache, false),
|
|
TempAcl = emqx_ct_helpers:deps_path(emqx, "test/emqx_access_SUITE_data/acl_temp.conf"),
|
|
file:write_file(TempAcl, "{deny, {client, \"batch_test\"}, subscribe,
|
|
[\"t1\", \"t2\", \"t3\"]}.\n"),
|
|
timer:sleep(10),
|
|
emqx_mod_acl_internal:reload([{acl_file, TempAcl}]),
|
|
{ok, _, [?RC_NOT_AUTHORIZED,
|
|
?RC_NOT_AUTHORIZED,
|
|
?RC_NOT_AUTHORIZED]} = emqtt:subscribe(Client, [{<<"t1">>, qos1},
|
|
{<<"t2">>, qos2},
|
|
{<<"t3">>, qos0}]),
|
|
{ok, _, [?RC_NO_SUBSCRIPTION_EXISTED,
|
|
?RC_NO_SUBSCRIPTION_EXISTED,
|
|
?RC_NO_SUBSCRIPTION_EXISTED]} = emqtt:unsubscribe(Client, [<<"t1">>,
|
|
<<"t2">>,
|
|
<<"t3">>]),
|
|
file:delete(TempAcl),
|
|
emqtt:disconnect(Client).
|
|
|
|
t_connect_will_retain(_) ->
|
|
Topic = nth(1, ?TOPICS),
|
|
Payload = "will message",
|
|
|
|
{ok, Client1} = emqtt:start_link([
|
|
{proto_ver, v5},
|
|
{clean_start, true},
|
|
{will_flag, true},
|
|
{will_topic, Topic},
|
|
{will_payload, Payload},
|
|
{will_retain, false}
|
|
]),
|
|
{ok, _} = emqtt:connect(Client1),
|
|
|
|
{ok, Client2} = emqtt:start_link([{proto_ver, v5}]),
|
|
{ok, _} = emqtt:connect(Client2),
|
|
{ok, _, [2]} = emqtt:subscribe(Client2, #{}, [{Topic, [{rap, true}, {qos, 2}]}]),
|
|
|
|
ok = emqtt:disconnect(Client1, 4),
|
|
[Msg1 | _ ] = receive_messages(1),
|
|
?assertEqual({ok, false}, maps:find(retain, Msg1)), %% [MQTT-3.1.2-14]
|
|
ok = emqtt:disconnect(Client2),
|
|
|
|
{ok, Client3} = emqtt:start_link([
|
|
{proto_ver, v5},
|
|
{clean_start, true},
|
|
{will_flag, true},
|
|
{will_topic, Topic},
|
|
{will_payload, Payload},
|
|
{will_retain, true}
|
|
]),
|
|
{ok, _} = emqtt:connect(Client3),
|
|
|
|
{ok, Client4} = emqtt:start_link([{proto_ver, v5}]),
|
|
{ok, _} = emqtt:connect(Client4),
|
|
{ok, _, [2]} = emqtt:subscribe(Client4, #{}, [{Topic, [{rap, true}, {qos, 2}]}]),
|
|
|
|
ok = emqtt:disconnect(Client3, 4),
|
|
[Msg2 | _ ] = receive_messages(1),
|
|
?assertEqual({ok, true}, maps:find(retain, Msg2)), %% [MQTT-3.1.2-15]
|
|
ok = emqtt:disconnect(Client4),
|
|
clean_retained(Topic).
|
|
|
|
t_connect_idle_timeout(_) ->
|
|
IdleTimeout = 2000,
|
|
emqx_zone:set_env(external, idle_timeout, IdleTimeout),
|
|
|
|
{ok, Sock} = emqtt_sock:connect({127,0,0,1}, 1883, [], 60000),
|
|
timer:sleep(IdleTimeout),
|
|
?assertMatch({error, closed}, emqtt_sock:recv(Sock,1024)).
|
|
|
|
t_connect_limit_timeout(_) ->
|
|
ok = meck:new(proplists, [non_strict, passthrough, no_history, no_link, unstick]),
|
|
meck:expect(proplists, get_value, fun(active_n, _Options, _Default) -> 1;
|
|
(Arg1, ARg2, Arg3) -> meck:passthrough([Arg1, ARg2, Arg3])
|
|
end),
|
|
|
|
Topic = nth(1, ?TOPICS),
|
|
emqx_zone:set_env(external, publish_limit, {3, 5}),
|
|
|
|
{ok, Client} = emqtt:start_link([{proto_ver, v5},{keepalive, 60}]),
|
|
{ok, _} = emqtt:connect(Client),
|
|
[ClientPid] = emqx_cm:lookup_channels(client_info(clientid, Client)),
|
|
|
|
?assertEqual(undefined, emqx_connection:info(limit_timer, sys:get_state(ClientPid))),
|
|
Payload = <<"t_shared_subscriptions_client_terminates_when_qos_eq_2">>,
|
|
ok = emqtt:publish(Client, Topic, Payload, 0),
|
|
ok = emqtt:publish(Client, Topic, Payload, 0),
|
|
ok = emqtt:publish(Client, Topic, Payload, 0),
|
|
timer:sleep(200),
|
|
?assert(is_reference(emqx_connection:info(limit_timer, sys:get_state(ClientPid)))),
|
|
|
|
ok = emqtt:disconnect(Client),
|
|
emqx_zone:set_env(external, publish_limit, undefined),
|
|
meck:unload(proplists).
|
|
|
|
t_connect_emit_stats_timeout(_) ->
|
|
IdleTimeout = 2000,
|
|
emqx_zone:set_env(external, idle_timeout, IdleTimeout),
|
|
|
|
{ok, Client} = emqtt:start_link([{proto_ver, v5},{keepalive, 60}]),
|
|
{ok, _} = emqtt:connect(Client),
|
|
[ClientPid] = emqx_cm:lookup_channels(client_info(clientid, Client)),
|
|
|
|
?assert(is_reference(emqx_connection:info(stats_timer, sys:get_state(ClientPid)))),
|
|
timer:sleep(IdleTimeout),
|
|
?assertEqual(undefined, emqx_connection:info(stats_timer, sys:get_state(ClientPid))),
|
|
ok = emqtt:disconnect(Client).
|
|
|
|
%% [MQTT-3.1.2-22]
|
|
t_connect_keepalive_timeout(_) ->
|
|
Keepalive = 2,
|
|
|
|
{ok, Client} = emqtt:start_link([{proto_ver, v5},
|
|
{keepalive, Keepalive}]),
|
|
{ok, _} = emqtt:connect(Client),
|
|
emqtt:pause(Client),
|
|
receive
|
|
{disconnected, ReasonCode, _Channel} -> ?assertEqual(141, ReasonCode)
|
|
after round(timer:seconds(Keepalive) * 2 * 1.5 ) ->
|
|
error("keepalive timeout")
|
|
end.
|
|
|
|
%% [MQTT-3.1.2-23]
|
|
t_connect_session_expiry_interval(_) ->
|
|
Topic = nth(1, ?TOPICS),
|
|
Payload = "test message",
|
|
|
|
{ok, Client1} = emqtt:start_link([
|
|
{clientid, <<"t_connect_session_expiry_interval">>},
|
|
{proto_ver, v5},
|
|
{properties, #{'Session-Expiry-Interval' => 7200}}
|
|
]),
|
|
{ok, _} = emqtt:connect(Client1),
|
|
{ok, _, [2]} = emqtt:subscribe(Client1, Topic, qos2),
|
|
ok = emqtt:disconnect(Client1),
|
|
|
|
{ok, Client2} = emqtt:start_link([{proto_ver, v5}]),
|
|
{ok, _} = emqtt:connect(Client2),
|
|
{ok, 2} = emqtt:publish(Client2, Topic, Payload, 2),
|
|
ok = emqtt:disconnect(Client2),
|
|
|
|
{ok, Client3} = emqtt:start_link([
|
|
{clientid, <<"t_connect_session_expiry_interval">>},
|
|
{proto_ver, v5},
|
|
{clean_start, false}
|
|
]),
|
|
{ok, _} = emqtt:connect(Client3),
|
|
[Msg | _ ] = receive_messages(1),
|
|
?assertEqual({ok, iolist_to_binary(Topic)}, maps:find(topic, Msg)),
|
|
?assertEqual({ok, iolist_to_binary(Payload)}, maps:find(payload, Msg)),
|
|
?assertEqual({ok, 2}, maps:find(qos, Msg)),
|
|
ok = emqtt:disconnect(Client3).
|
|
|
|
%% [MQTT-3.1.3-9]
|
|
t_connect_will_delay_interval(_) ->
|
|
process_flag(trap_exit, true),
|
|
Topic = nth(1, ?TOPICS),
|
|
Payload = "will message",
|
|
|
|
{ok, Client1} = emqtt:start_link([{proto_ver, v5}]),
|
|
{ok, _} = emqtt:connect(Client1),
|
|
{ok, _, [2]} = emqtt:subscribe(Client1, Topic, qos2),
|
|
|
|
{ok, Client2} = emqtt:start_link([
|
|
{clientid, <<"t_connect_will_delay_interval">>},
|
|
{proto_ver, v5},
|
|
{clean_start, true},
|
|
{will_flag, true},
|
|
{will_qos, 2},
|
|
{will_topic, Topic},
|
|
{will_payload, Payload},
|
|
{will_props, #{'Will-Delay-Interval' => 3}},
|
|
{properties, #{'Session-Expiry-Interval' => 7200}}
|
|
]),
|
|
{ok, _} = emqtt:connect(Client2),
|
|
%% terminate the client without sending the DISCONNECT
|
|
emqtt:stop(Client2),
|
|
%% should not get the will msg in 2.5s
|
|
timer:sleep(1500),
|
|
?assertEqual(0, length(receive_messages(1))),
|
|
%% should get the will msg in 4.5s
|
|
timer:sleep(1000),
|
|
?assertEqual(1, length(receive_messages(1))),
|
|
|
|
%% try again, but let the session expire quickly
|
|
{ok, Client3} = emqtt:start_link([
|
|
{clientid, <<"t_connect_will_delay_interval">>},
|
|
{proto_ver, v5},
|
|
{clean_start, true},
|
|
{will_flag, true},
|
|
{will_qos, 2},
|
|
{will_topic, Topic},
|
|
{will_payload, Payload},
|
|
{will_props, #{'Will-Delay-Interval' => 7200}},
|
|
{properties, #{'Session-Expiry-Interval' => 3}}
|
|
]),
|
|
{ok, _} = emqtt:connect(Client3),
|
|
%% terminate the client without sending the DISCONNECT
|
|
emqtt:stop(Client3),
|
|
%% should not get the will msg in 2.5s
|
|
timer:sleep(1500),
|
|
?assertEqual(0, length(receive_messages(1))),
|
|
%% should get the will msg in 4.5s
|
|
timer:sleep(1000),
|
|
?assertEqual(1, length(receive_messages(1))),
|
|
|
|
ok = emqtt:disconnect(Client1),
|
|
|
|
receive {'EXIT', _, _} -> ok
|
|
after 100 -> ok
|
|
end,
|
|
process_flag(trap_exit, false).
|
|
|
|
%% [MQTT-3.1.4-3]
|
|
t_connect_duplicate_clientid(_) ->
|
|
process_flag(trap_exit, true),
|
|
{ok, Client1} = emqtt:start_link([
|
|
{clientid, <<"t_connect_duplicate_clientid">>},
|
|
{proto_ver, v5}
|
|
]),
|
|
{ok, _} = emqtt:connect(Client1),
|
|
{ok, Client2} = emqtt:start_link([
|
|
{clientid, <<"t_connect_duplicate_clientid">>},
|
|
{proto_ver, v5}
|
|
]),
|
|
{ok, _} = emqtt:connect(Client2),
|
|
?assertEqual(142, receive_disconnect_reasoncode()),
|
|
waiting_client_process_exit(Client1),
|
|
|
|
ok = emqtt:disconnect(Client2),
|
|
waiting_client_process_exit(Client2),
|
|
process_flag(trap_exit, false).
|
|
|
|
%%--------------------------------------------------------------------
|
|
%% Connack
|
|
%%--------------------------------------------------------------------
|
|
|
|
t_connack_session_present(_) ->
|
|
{ok, Client1} = emqtt:start_link([
|
|
{clientid, <<"t_connect_duplicate_clientid">>},
|
|
{proto_ver, v5},
|
|
{properties, #{'Session-Expiry-Interval' => 7200}},
|
|
{clean_start, true}
|
|
]),
|
|
{ok, _} = emqtt:connect(Client1),
|
|
?assertEqual(0, client_info(session_present, Client1)), %% [MQTT-3.2.2-2]
|
|
ok = emqtt:disconnect(Client1),
|
|
|
|
{ok, Client2} = emqtt:start_link([
|
|
{clientid, <<"t_connect_duplicate_clientid">>},
|
|
{proto_ver, v5},
|
|
{properties, #{'Session-Expiry-Interval' => 7200}},
|
|
{clean_start, false}
|
|
]),
|
|
{ok, _} = emqtt:connect(Client2),
|
|
?assertEqual(1, client_info(session_present, Client2)), %% [[MQTT-3.2.2-3]]
|
|
ok = emqtt:disconnect(Client2).
|
|
|
|
t_connack_max_qos_allowed(_) ->
|
|
process_flag(trap_exit, true),
|
|
Topic = nth(1, ?TOPICS),
|
|
|
|
%% max_qos_allowed = 0
|
|
emqx_zone:set_env(external, max_qos_allowed, 0),
|
|
persistent_term:erase({emqx_zone, external, '$mqtt_caps'}),
|
|
persistent_term:erase({emqx_zone, external, '$mqtt_pub_caps'}),
|
|
|
|
{ok, Client1} = emqtt:start_link([{proto_ver, v5}]),
|
|
{ok, Connack1} = emqtt:connect(Client1),
|
|
?assertEqual(0, maps:get('Maximum-QoS', Connack1)), %% [MQTT-3.2.2-9]
|
|
|
|
{ok, _, [0]} = emqtt:subscribe(Client1, Topic, 0), %% [MQTT-3.2.2-10]
|
|
{ok, _, [1]} = emqtt:subscribe(Client1, Topic, 1), %% [MQTT-3.2.2-10]
|
|
{ok, _, [2]} = emqtt:subscribe(Client1, Topic, 2), %% [MQTT-3.2.2-10]
|
|
|
|
{ok, _} = emqtt:publish(Client1, Topic, <<"Unsupported Qos 1">>, qos1),
|
|
?assertEqual(155, receive_disconnect_reasoncode()), %% [MQTT-3.2.2-11]
|
|
waiting_client_process_exit(Client1),
|
|
|
|
{ok, Client2} = emqtt:start_link([
|
|
{proto_ver, v5},
|
|
{will_flag, true},
|
|
{will_topic, Topic},
|
|
{will_payload, <<"Unsupported Qos">>},
|
|
{will_qos, 2}
|
|
]),
|
|
{error, Connack2} = emqtt:connect(Client2),
|
|
?assertMatch({qos_not_supported, _}, Connack2), %% [MQTT-3.2.2-12]
|
|
waiting_client_process_exit(Client2),
|
|
|
|
%% max_qos_allowed = 1
|
|
emqx_zone:set_env(external, max_qos_allowed, 1),
|
|
persistent_term:erase({emqx_zone, external, '$mqtt_caps'}),
|
|
persistent_term:erase({emqx_zone, external, '$mqtt_pub_caps'}),
|
|
|
|
{ok, Client3} = emqtt:start_link([{proto_ver, v5}]),
|
|
{ok, Connack3} = emqtt:connect(Client3),
|
|
?assertEqual(1, maps:get('Maximum-QoS', Connack3)), %% [MQTT-3.2.2-9]
|
|
|
|
{ok, _, [0]} = emqtt:subscribe(Client3, Topic, 0), %% [MQTT-3.2.2-10]
|
|
{ok, _, [1]} = emqtt:subscribe(Client3, Topic, 1), %% [MQTT-3.2.2-10]
|
|
{ok, _, [2]} = emqtt:subscribe(Client3, Topic, 2), %% [MQTT-3.2.2-10]
|
|
|
|
{ok, _} = emqtt:publish(Client3, Topic, <<"Unsupported Qos 2">>, qos2),
|
|
?assertEqual(155, receive_disconnect_reasoncode()), %% [MQTT-3.2.2-11]
|
|
waiting_client_process_exit(Client3),
|
|
|
|
{ok, Client4} = emqtt:start_link([
|
|
{proto_ver, v5},
|
|
{will_flag, true},
|
|
{will_topic, Topic},
|
|
{will_payload, <<"Unsupported Qos">>},
|
|
{will_qos, 2}
|
|
]),
|
|
{error, Connack4} = emqtt:connect(Client4),
|
|
?assertMatch({qos_not_supported, _}, Connack4), %% [MQTT-3.2.2-12]
|
|
waiting_client_process_exit(Client4),
|
|
|
|
%% max_qos_allowed = 2
|
|
emqx_zone:set_env(external, max_qos_allowed, 2),
|
|
persistent_term:erase({emqx_zone, external, '$mqtt_caps'}),
|
|
persistent_term:erase({emqx_zone, external, '$mqtt_pub_caps'}),
|
|
|
|
{ok, Client5} = emqtt:start_link([{proto_ver, v5}]),
|
|
{ok, Connack5} = emqtt:connect(Client5),
|
|
?assertEqual(undefined, maps:get('Maximum-QoS', Connack5, undefined)), %% [MQTT-3.2.2-9]
|
|
ok = emqtt:disconnect(Client5),
|
|
waiting_client_process_exit(Client5),
|
|
|
|
process_flag(trap_exit, false).
|
|
|
|
t_connack_assigned_clienid(_) ->
|
|
{ok, Client1} = emqtt:start_link([{proto_ver, v5}]),
|
|
{ok, _} = emqtt:connect(Client1),
|
|
?assert(is_binary(client_info(clientid, Client1))), %% [MQTT-3.2.2-16]
|
|
ok = emqtt:disconnect(Client1).
|
|
|
|
%%--------------------------------------------------------------------
|
|
%% Publish
|
|
%%--------------------------------------------------------------------
|
|
|
|
t_publish_rap(_) ->
|
|
Topic = nth(1, ?TOPICS),
|
|
|
|
{ok, Client1} = emqtt:start_link([{proto_ver, v5}]),
|
|
{ok, _} = emqtt:connect(Client1),
|
|
{ok, _, [2]} = emqtt:subscribe(Client1, #{}, [{Topic, [{rap, true}, {qos, 2}]}]),
|
|
{ok, _} = emqtt:publish(Client1, Topic, #{}, <<"retained message">>,
|
|
[{qos, ?QOS_1}, {retain, true}]),
|
|
[Msg1 | _] = receive_messages(1),
|
|
?assertEqual(true, maps:get(retain, Msg1)), %% [MQTT-3.3.1-12]
|
|
ok = emqtt:disconnect(Client1),
|
|
|
|
{ok, Client2} = emqtt:start_link([{proto_ver, v5}]),
|
|
{ok, _} = emqtt:connect(Client2),
|
|
{ok, _, [2]} = emqtt:subscribe(Client2, #{}, [{Topic, [{rap, false}, {qos, 2}]}]),
|
|
{ok, _} = emqtt:publish(Client2, Topic, #{}, <<"retained message">>,
|
|
[{qos, ?QOS_1}, {retain, true}]),
|
|
[Msg2 | _] = receive_messages(1),
|
|
?assertEqual(false, maps:get(retain, Msg2)), %% [MQTT-3.3.1-13]
|
|
ok = emqtt:disconnect(Client2),
|
|
|
|
clean_retained(Topic).
|
|
|
|
t_publish_wildtopic(_) ->
|
|
process_flag(trap_exit, true),
|
|
Topic = nth(1, ?WILD_TOPICS),
|
|
|
|
{ok, Client1} = emqtt:start_link([{proto_ver, v5}]),
|
|
{ok, _} = emqtt:connect(Client1),
|
|
ok = emqtt:publish(Client1, Topic, <<"error topic">>),
|
|
?assertEqual(144, receive_disconnect_reasoncode()),
|
|
waiting_client_process_exit(Client1),
|
|
|
|
process_flag(trap_exit, false).
|
|
|
|
t_publish_payload_format_indicator(_) ->
|
|
Topic = nth(1, ?TOPICS),
|
|
Properties = #{'Payload-Format-Indicator' => 233},
|
|
|
|
{ok, Client1} = emqtt:start_link([{proto_ver, v5}]),
|
|
{ok, _} = emqtt:connect(Client1),
|
|
{ok, _, [2]} = emqtt:subscribe(Client1, Topic, qos2),
|
|
ok = emqtt:publish(Client1, Topic, Properties, <<"Payload Format Indicator">>, [{qos, ?QOS_0}]),
|
|
[Msg1 | _] = receive_messages(1),
|
|
?assertEqual(Properties, maps:get(properties, Msg1)), %% [MQTT-3.3.2-6]
|
|
ok = emqtt:disconnect(Client1).
|
|
|
|
t_publish_topic_alias(_) ->
|
|
process_flag(trap_exit, true),
|
|
Topic = nth(1, ?TOPICS),
|
|
|
|
{ok, Client1} = emqtt:start_link([{proto_ver, v5}]),
|
|
{ok, _} = emqtt:connect(Client1),
|
|
ok = emqtt:publish(Client1, Topic, #{'Topic-Alias' => 0}, <<"Topic-Alias">>, [{qos, ?QOS_0}]),
|
|
?assertEqual(148, receive_disconnect_reasoncode()), %% [MQTT-3.3.2-8]
|
|
waiting_client_process_exit(Client1),
|
|
|
|
{ok, Client2} = emqtt:start_link([{proto_ver, v5}]),
|
|
{ok, _} = emqtt:connect(Client2),
|
|
{ok, _, [2]} = emqtt:subscribe(Client2, Topic, qos2),
|
|
ok = emqtt:publish(Client2, Topic, #{'Topic-Alias' => 233},
|
|
<<"Topic-Alias">>, [{qos, ?QOS_0}]),
|
|
ok = emqtt:publish(Client2, <<"">>, #{'Topic-Alias' => 233},
|
|
<<"Topic-Alias">>, [{qos, ?QOS_0}]),
|
|
?assertEqual(2, length(receive_messages(2))), %% [MQTT-3.3.2-12]
|
|
ok = emqtt:disconnect(Client2),
|
|
waiting_client_process_exit(Client2),
|
|
|
|
process_flag(trap_exit, false).
|
|
|
|
t_publish_response_topic(_) ->
|
|
process_flag(trap_exit, true),
|
|
Topic = nth(1, ?TOPICS),
|
|
|
|
{ok, Client1} = emqtt:start_link([{proto_ver, v5}]),
|
|
{ok, _} = emqtt:connect(Client1),
|
|
ok = emqtt:publish(Client1, Topic, #{'Response-Topic' => nth(1, ?WILD_TOPICS)},
|
|
<<"Response-Topic">>, [{qos, ?QOS_0}]),
|
|
?assertEqual(130, receive_disconnect_reasoncode()), %% [MQTT-3.3.2-14]
|
|
waiting_client_process_exit(Client1),
|
|
|
|
process_flag(trap_exit, false).
|
|
|
|
t_publish_properties(_) ->
|
|
Topic = nth(1, ?TOPICS),
|
|
Properties = #{
|
|
'Response-Topic' => Topic, %% [MQTT-3.3.2-15]
|
|
'Correlation-Data' => <<"233">>, %% [MQTT-3.3.2-16]
|
|
'User-Property' => [{<<"a">>, <<"2333">>}], %% [MQTT-3.3.2-18]
|
|
'Content-Type' => <<"2333">> %% [MQTT-3.3.2-20]
|
|
},
|
|
|
|
{ok, Client1} = emqtt:start_link([{proto_ver, v5}]),
|
|
{ok, _} = emqtt:connect(Client1),
|
|
{ok, _, [2]} = emqtt:subscribe(Client1, Topic, qos2),
|
|
ok = emqtt:publish(Client1, Topic, Properties, <<"Publish Properties">>, [{qos, ?QOS_0}]),
|
|
[Msg1 | _] = receive_messages(1),
|
|
?assertEqual(Properties, maps:get(properties, Msg1)), %% [MQTT-3.3.2-16]
|
|
ok = emqtt:disconnect(Client1).
|
|
|
|
t_publish_overlapping_subscriptions(_) ->
|
|
Topic = nth(1, ?TOPICS),
|
|
Properties = #{'Subscription-Identifier' => 2333},
|
|
|
|
{ok, Client1} = emqtt:start_link([{proto_ver, v5}]),
|
|
{ok, _} = emqtt:connect(Client1),
|
|
{ok, _, [1]} = emqtt:subscribe(Client1, Properties, nth(1, ?WILD_TOPICS), qos1),
|
|
{ok, _, [0]} = emqtt:subscribe(Client1, Properties, nth(3, ?WILD_TOPICS), qos0),
|
|
{ok, _} = emqtt:publish(Client1, Topic, #{},
|
|
<<"t_publish_overlapping_subscriptions">>, [{qos, ?QOS_2}]),
|
|
|
|
[Msg1 | _ ] = receive_messages(2),
|
|
?assert( maps:get(qos, Msg1) < 2 ), %% [MQTT-3.3.4-2]
|
|
?assertEqual(Properties, maps:get(properties, Msg1)), %% [MQTT-3.3.4-3]
|
|
ok = emqtt:disconnect(Client1).
|
|
|
|
%%--------------------------------------------------------------------
|
|
%% Subsctibe
|
|
%%--------------------------------------------------------------------
|
|
|
|
t_subscribe_topic_alias(_) ->
|
|
Topic1 = nth(1, ?TOPICS),
|
|
Topic2 = nth(2, ?TOPICS),
|
|
{ok, Client1} = emqtt:start_link([{proto_ver, v5},
|
|
{properties, #{'Topic-Alias-Maximum' => 1}}
|
|
]),
|
|
{ok, _} = emqtt:connect(Client1),
|
|
{ok, _, [2]} = emqtt:subscribe(Client1, Topic1, qos2),
|
|
{ok, _, [2]} = emqtt:subscribe(Client1, Topic2, qos2),
|
|
|
|
ok = emqtt:publish(Client1, Topic1, #{}, <<"Topic-Alias">>, [{qos, ?QOS_0}]),
|
|
[Msg1] = receive_messages(1),
|
|
?assertEqual({ok, #{'Topic-Alias' => 1}}, maps:find(properties, Msg1)),
|
|
?assertEqual({ok, Topic1}, maps:find(topic, Msg1)),
|
|
|
|
ok = emqtt:publish(Client1, Topic1, #{}, <<"Topic-Alias">>, [{qos, ?QOS_0}]),
|
|
[Msg2] = receive_messages(1),
|
|
?assertEqual({ok, #{'Topic-Alias' => 1}}, maps:find(properties, Msg2)),
|
|
?assertEqual({ok, <<>>}, maps:find(topic, Msg2)),
|
|
|
|
ok = emqtt:publish(Client1, Topic2, #{}, <<"Topic-Alias">>, [{qos, ?QOS_0}]),
|
|
[Msg3] = receive_messages(1),
|
|
?assertEqual({ok, #{}}, maps:find(properties, Msg3)),
|
|
?assertEqual({ok, Topic2}, maps:find(topic, Msg3)),
|
|
|
|
ok = emqtt:disconnect(Client1).
|
|
|
|
t_subscribe_no_local(_) ->
|
|
Topic = nth(1, ?TOPICS),
|
|
|
|
{ok, Client1} = emqtt:start_link([{proto_ver, v5}]),
|
|
{ok, _} = emqtt:connect(Client1),
|
|
{ok, _, [2]} = emqtt:subscribe(Client1, #{}, [{Topic, [{nl, true}, {qos, 2}]}]),
|
|
|
|
{ok, Client2} = emqtt:start_link([{proto_ver, v5}]),
|
|
{ok, _} = emqtt:connect(Client2),
|
|
{ok, _, [2]} = emqtt:subscribe(Client2, #{}, [{Topic, [{nl, true}, {qos, 2}]}]),
|
|
|
|
ok = emqtt:publish(Client1, Topic, <<"t_subscribe_no_local">>, 0),
|
|
?assertEqual(1, length(receive_messages(2))), %% [MQTT-3.8.3-3]
|
|
ok = emqtt:disconnect(Client1).
|
|
|
|
t_subscribe_actions(_) ->
|
|
Topic = nth(1, ?TOPICS),
|
|
Properties = #{'Subscription-Identifier' => 2333},
|
|
|
|
{ok, Client1} = emqtt:start_link([{proto_ver, v5}]),
|
|
{ok, _} = emqtt:connect(Client1),
|
|
{ok, _, [2]} = emqtt:subscribe(Client1, Properties, Topic, qos2),
|
|
{ok, _, [1]} = emqtt:subscribe(Client1, Properties, Topic, qos1),
|
|
{ok, _} = emqtt:publish(Client1, Topic, <<"t_subscribe_actions">>, 2),
|
|
[Msg1 | _ ] = receive_messages(1),
|
|
?assertEqual(1, maps:get(qos, Msg1)), %% [MQTT-3.8.4-3] [MQTT-3.8.4-8]
|
|
%% [MQTT-3.8.4-5] [MQTT-3.8.4-6] [MQTT-3.8.4-7]
|
|
{ok, _, [2,2]} = emqtt:subscribe(Client1, [{nth(1, ?TOPICS), qos2},
|
|
{nth(2, ?TOPICS), qos2}] ),
|
|
ok = emqtt:disconnect(Client1).
|
|
%%--------------------------------------------------------------------
|
|
%% Unsubsctibe Unsuback
|
|
%%--------------------------------------------------------------------
|
|
|
|
t_unscbsctibe(_) ->
|
|
Topic1 = nth(1, ?TOPICS),
|
|
Topic2 = nth(2, ?TOPICS),
|
|
|
|
{ok, Client1} = emqtt:start_link([{proto_ver, v5}]),
|
|
{ok, _} = emqtt:connect(Client1),
|
|
{ok, _, [2]} = emqtt:subscribe(Client1, Topic1, qos2),
|
|
{ok, _, [0]} = emqtt:unsubscribe(Client1, Topic1), %% [MQTT-3.10.4-4]
|
|
{ok, _, [17]} = emqtt:unsubscribe(Client1, <<"noExistTopic">>), %% [MQTT-3.10.4-5]
|
|
|
|
{ok, _, [2, 2]} = emqtt:subscribe(Client1, [{Topic1, qos2}, {Topic2, qos2}]),
|
|
%% [[MQTT-3.10.4-6]] [MQTT-3.11.3-1] [MQTT-3.11.3-2]
|
|
{ok, _, [0, 0, 17]} = emqtt:unsubscribe(Client1, [Topic1, Topic2, <<"noExistTopic">>]),
|
|
ok = emqtt:disconnect(Client1).
|
|
|
|
%%--------------------------------------------------------------------
|
|
%% Pingreq
|
|
%%--------------------------------------------------------------------
|
|
|
|
t_pingreq(_) ->
|
|
{ok, Client1} = emqtt:start_link([{proto_ver, v5}]),
|
|
{ok, _} = emqtt:connect(Client1),
|
|
pong = emqtt:ping(Client1), %% [MQTT-3.12.4-1]
|
|
ok = emqtt:disconnect(Client1).
|
|
|
|
%%--------------------------------------------------------------------
|
|
%% Shared Subscriptions
|
|
%%--------------------------------------------------------------------
|
|
|
|
t_shared_subscriptions_client_terminates_when_qos_eq_2(_) ->
|
|
process_flag(trap_exit, true),
|
|
application:set_env(emqx, shared_dispatch_ack_enabled, true),
|
|
|
|
Topic = nth(1, ?TOPICS),
|
|
SharedTopic = list_to_binary("$share/sharename/" ++ binary_to_list(<<"TopicA">>)),
|
|
|
|
CRef = counters:new(1, [atomics]),
|
|
meck:expect(emqtt, connected,
|
|
fun(cast, ?PUBLISH_PACKET(?QOS_2, _PacketId), _State) ->
|
|
ok = counters:add(CRef, 1, 1),
|
|
{stop, {shutdown, for_testiong}};
|
|
(Arg1, ARg2, Arg3) -> meck:passthrough([Arg1, ARg2, Arg3])
|
|
end),
|
|
|
|
{ok, Sub1} = emqtt:start_link([{proto_ver, v5},
|
|
{clientid, <<"sub_client_1">>},
|
|
{keepalive, 5}]),
|
|
{ok, _} = emqtt:connect(Sub1),
|
|
{ok, _, [2]} = emqtt:subscribe(Sub1, SharedTopic, qos2),
|
|
|
|
{ok, Sub2} = emqtt:start_link([{proto_ver, v5},
|
|
{clientid, <<"sub_client_2">>},
|
|
{keepalive, 5}]),
|
|
{ok, _} = emqtt:connect(Sub2),
|
|
{ok, _, [2]} = emqtt:subscribe(Sub2, SharedTopic, qos2),
|
|
|
|
{ok, Pub} = emqtt:start_link([{proto_ver, v5}, {clientid, <<"pub_client">>}]),
|
|
{ok, _} = emqtt:connect(Pub),
|
|
{ok, _} = emqtt:publish(Pub, Topic,
|
|
<<"t_shared_subscriptions_client_terminates_when_qos_eq_2">>, 2),
|
|
|
|
receive
|
|
{'EXIT', _,{shutdown, for_testiong}} ->
|
|
ok
|
|
after 1000 ->
|
|
error("disconnected timeout")
|
|
end,
|
|
|
|
?assertEqual(1, counters:get(CRef, 1)),
|
|
process_flag(trap_exit, false).
|