%%--------------------------------------------------------------------
%% 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(prop_emqx_frame).

-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)).