yuanbiao
/
emqx
1
0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add mqtt connect tests cases

This commit is contained in:
HuangDan 2018-08-28 20:05:56 +08:00
parent 0fbf813ebf
commit 37c1570e94
1 changed files with 69 additions and 67 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

134
test/emqx_SUITE.erl
View File

@ -25,21 +25,34 @@
-include_lib("common_test/include/ct.hrl"). -include_lib("common_test/include/ct.hrl").
-include("emqx_mqtt.hrl").
-record(ssl_socket, {tcp, ssl}).
-type(socket() :: inet:socket() | #ssl_socket{}).
-define(CLIENT, ?CONNECT_PACKET(#mqtt_packet_connect{ -define(CLIENT, ?CONNECT_PACKET(#mqtt_packet_connect{
client_id = <<"mqtt_client">>, client_id = <<"mqtt_client">>,
username = <<"admin">>, username = <<"admin">>,
password = <<"public">>})). password = <<"public">>})).
-define(CLIENT2, ?CONNECT_PACKET(#mqtt_packet_connect{
username = <<"admin">>,
clean_start = false,
password = <<"public">>})).
all() -> all() ->
[{group, connect}, [{group, connect}%,
{group, cleanSession}]. % {group, cleanSession}
].
groups() -> groups() ->
[{connect, [non_parallel_tests], [{connect, [non_parallel_tests],
[mqtt_connect, [
% mqtt_connect,
% mqtt_connect_with_tcp, % mqtt_connect_with_tcp,
mqtt_connect_with_ssl_oneway, % mqtt_connect_with_ssl_oneway,
mqtt_connect_with_ssl_twoway%, % mqtt_connect_with_ssl_twoway%,
% mqtt_connect_with_ws mqtt_connect_with_ws%,
]}, ]},
{cleanSession, [sequence], {cleanSession, [sequence],
[cleanSession_validate] [cleanSession_validate]
@ -48,7 +61,6 @@ groups() ->
init_per_suite(Config) -> init_per_suite(Config) ->
emqx_ct_broker_helpers:run_setup_steps(), emqx_ct_broker_helpers:run_setup_steps(),
% ct:log("Apps:~p", [Apps]),
Config. Config.
end_per_suite(_Config) -> end_per_suite(_Config) ->
@ -65,76 +77,65 @@ mqtt_connect(_) ->
?assertEqual(<<32,2,0,0>>, connect_broker_(<<16,12,0,4,77,81,84,84,4,2,0,90,0,0>>, 4)). ?assertEqual(<<32,2,0,0>>, connect_broker_(<<16,12,0,4,77,81,84,84,4,2,0,90,0,0>>, 4)).
connect_broker_(Packet, RecvSize) -> connect_broker_(Packet, RecvSize) ->
{ok, Sock} = gen_tcp:connect({127,0,0,1}, 1883, [binary, {packet, raw}, {active, false}]), {ok, Sock} = emqx_client_sock:connect({127,0,0,1}, 1883, [binary, {packet, raw}, {active, false}], 3000),
gen_tcp:send(Sock, Packet), emqx_client_sock:send(Sock, Packet),
{ok, Data} = gen_tcp:recv(Sock, RecvSize, 3000), {ok, Data} = gen_tcp:recv(Sock, RecvSize, 3000),
gen_tcp:close(Sock), emqx_client_sock:close(Sock),
Data. Data.
%% mqtt_connect_with_tcp(_) -> mqtt_connect_with_tcp(_) ->
%% %% Issue #599 %% Issue #599
%% %% Empty clientId and clean_session = false %% Empty clientId and clean_session = false
%% {ok, Sock} = gen_tcp:connect({127,0,0,1}, 1883, [binary, {packet, raw}, {active, false}]), {ok, Sock} = emqx_client_sock:connect({127,0,0,1}, 1883, [binary, {packet, raw}, {active, false}], 3000),
%% Packet = raw_send_serialise(?CLIENT), Packet = raw_send_serialise(?CLIENT2),
%% gen_tcp:send(Sock, Packet), emqx_client_sock:send(Sock, Packet),
%% {ok, Data} = gen_tcp:recv(Sock, 0), {ok, Data} = gen_tcp:recv(Sock, 0),
%% % {ok, ?CONNACK_PACKET(?CONNACK_ACCEPT), _} = raw_recv_pase(Data), {ok, ?CONNACK_PACKET(?CONNACK_INVALID_ID), _} = raw_recv_pase(Data),
%% gen_tcp:close(Sock). emqx_client_sock:close(Sock).
mqtt_connect_with_ssl_oneway(_) -> mqtt_connect_with_ssl_oneway(_) ->
emqx:stop(), emqx:shutdown(),
emqx_ct_broker_helpers:change_opts(ssl_oneway), emqx_ct_broker_helpers:change_opts(ssl_oneway),
emqx:start(), emqx:start(),
timer:sleep(5000), ClientSsl = emqx_ct_broker_helpers:client_ssl(),
{ok, SslOneWay} = emqttc:start_link([{host, "localhost"}, {ok, #ssl_socket{tcp = Sock, ssl = SslSock}}
{port, 8883}, = emqx_client_sock:connect("127.0.0.1", 8883, [{ssl_opts, ClientSsl}], 3000),
{logger, debug}, %% Packet = raw_send_serialise(?CLIENT),
{client_id, <<"ssloneway">>}, ssl]), %% ssl:send(SslSock, Packet),
timer:sleep(100), %% receive Data ->
emqttc:subscribe(SslOneWay, <<"topic">>, qos1), %% ct:log("Data:~p~n", [Data])
{ok, Pub} = emqttc:start_link([{host, "localhost"}, %% after 30000 ->
{client_id, <<"pub">>}]), %% ok
emqttc:publish(Pub, <<"topic">>, <<"SSL oneWay test">>, [{qos, 1}]), %% end,
timer:sleep(100), ssl:close(SslSock).
receive {publish, _Topic, RM} ->
?assertEqual(<<"SSL oneWay test">>, RM)
after 1000 -> false
end,
timer:sleep(100),
emqttc:disconnect(SslOneWay),
emqttc:disconnect(Pub).
mqtt_connect_with_ssl_twoway(_Config) -> mqtt_connect_with_ssl_twoway(_Config) ->
emqx:stop(), emqx:shutdown(),
emqx_ct_broker_helpers:change_opts(ssl_twoway), emqx_ct_broker_helpers:change_opts(ssl_twoway),
emqx:start(), emqx:start(),
timer:sleep(3000), ClientSsl = emqx_ct_broker_helpers:client_ssl_twoway(),
ClientSSl = emqx_ct_broker_helpers:client_ssl(), {ok, #ssl_socket{tcp = _Sock1, ssl = SslSock} = Sock}
{ok, SslTwoWay} = emqttc:start_link([{host, "localhost"}, = emqx_client_sock:connect("127.0.0.1", 8883, [{ssl_opts, ClientSsl}], 3000),
{port, 8883}, Packet = raw_send_serialise(?CLIENT),
{client_id, <<"ssltwoway">>}, emqx_client_sock:setopts(Sock, [{active, once}]),
{ssl, ClientSSl}]), emqx_client_sock:send(Sock, Packet),
{ok, Sub} = emqttc:start_link([{host, "localhost"}, timer:sleep(500),
{client_id, <<"sub">>}]), receive {ssl, _, Data}->
emqttc:subscribe(Sub, <<"topic">>, qos1), {ok, ?CONNACK_PACKET(?CONNACK_ACCEPT), _} = raw_recv_pase(Data)
emqttc:publish(SslTwoWay, <<"topic">>, <<"ssl client pub message">>, [{qos, 1}]), after 1000 ->
timer:sleep(10), ok
receive {publish, _Topic, RM} ->
?assertEqual(<<"ssl client pub message">>, RM)
after 1000 -> false
end, end,
emqttc:disconnect(SslTwoWay), emqx_client_sock:close(Sock).
emqttc:disconnect(Sub).
%% mqtt_connect_with_ws(_Config) -> mqtt_connect_with_ws(_Config) ->
%% WS = rfc6455_client:new("ws://127.0.0.1:8083" ++ "/mqtt", self()), WS = rfc6455_client:new("ws://127.0.0.1:8083" ++ "/mqtt", self()),
%% {ok, _} = rfc6455_client:open(WS), {ok, _} = rfc6455_client:open(WS),
%% Packet = raw_send_serialise(?CLIENT), Packet = raw_send_serialise(?CLIENT),
%% ok = rfc6455_client:send_binary(WS, Packet), ok = rfc6455_client:send_binary(WS, Packet),
%% {binary, P} = rfc6455_client:recv(WS), {binary, P} = rfc6455_client:recv(WS),
%% % {ok, ?CONNACK_PACKET(?CONNACK_ACCEPT), _} = raw_recv_pase(P), ct:log(":~p", [P]),
%% {close, _} = rfc6455_client:close(WS), {close, _} = rfc6455_client:close(WS),
%% ok. ok.
cleanSession_validate(_) -> cleanSession_validate(_) ->
{ok, C1} = emqttc:start_link([{host, "localhost"}, {ok, C1} = emqttc:start_link([{host, "localhost"},
@ -163,8 +164,9 @@ cleanSession_validate(_) ->
emqttc:disconnect(C11). emqttc:disconnect(C11).
raw_send_serialise(Packet) -> raw_send_serialise(Packet) ->
emqttc_serialiser:serialise(Packet). emqx_frame:serialize(Packet).
raw_recv_pase(P) -> raw_recv_pase(P) ->
emqttc_parser:parse(P, emqttc_parser:new()). emqx_frame:parse(P, {none, #{max_packet_size => ?MAX_PACKET_SIZE,
version => ?MQTT_PROTO_V4} }).