emqx4.3/apps/emqx/test/props/prop_emqx_frame.erl

%%--------------------------------------------------------------------
%% Copyright (c) 2020-2023 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(prop_emqx_frame).

-include_lib("emqx/include/emqx_mqtt.hrl").
-include_lib("proper/include/proper.hrl").

%%--------------------------------------------------------------------
%% Properties
%%--------------------------------------------------------------------

prop_serialize_parse_connect() ->
    ?FORALL(
        Opts = #{version := ProtoVer},
        parse_opts(),
        begin
            ProtoName = proplists:get_value(ProtoVer, ?PROTOCOL_NAMES),
            Packet = ?CONNECT_PACKET(#mqtt_packet_connect{
                proto_name = ProtoName,
                proto_ver = ProtoVer,
                clientid = <<"clientId">>,
                will_qos = ?QOS_1,
                will_flag = true,
                will_retain = true,
                will_topic = <<"will">>,
                will_props = #{},
                will_payload = <<"bye">>,
                clean_start = true,
                properties = #{}
            }),
            Packet =:= parse_serialize(Packet, Opts)
        end
    ).

%%--------------------------------------------------------------------
%% Helpers
%%--------------------------------------------------------------------

parse_serialize(Packet, Opts) when is_map(Opts) ->
    Ver = maps:get(version, Opts, ?MQTT_PROTO_V4),
    Bin = iolist_to_binary(emqx_frame:serialize(Packet, Ver)),
    ParseState = emqx_frame:initial_parse_state(Opts),
    {ok, NPacket, <<>>, _} = emqx_frame:parse(Bin, ParseState),
    NPacket.

%%--------------------------------------------------------------------
%% Generators
%%--------------------------------------------------------------------

parse_opts() ->
    ?LET(PropList, [{strict_mode, boolean()}, {version, range(4, 5)}], maps:from_list(PropList)).