From e56252dac6a2448c85b8a5d200d2db171bea8800 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=91=A8=E5=AD=90=E5=8D=9A?= <349832309@qq.com>
Date: Fri, 26 Oct 2018 14:34:22 +0800
Subject: [PATCH] Fix bugs in test cases

---
 test/emqx_mock_client.erl | 17 +++++++++--------
 test/emqx_sm_SUITE.erl    | 11 +++++++++--
 2 files changed, 18 insertions(+), 10 deletions(-)

diff --git a/test/emqx_mock_client.erl b/test/emqx_mock_client.erl
index 4a49a1fc5..24de66d0a 100644
--- a/test/emqx_mock_client.erl
+++ b/test/emqx_mock_client.erl
@@ -46,14 +46,15 @@ init([ClientId]) ->
     }.
 
 handle_call({start_session, ClientPid, ClientId, Zone}, _From, State) ->
-    Attrs = #{ zone             => Zone,
-               client_id        => ClientId,
-               conn_pid         => ClientPid,
-               clean_start      => true,
-               username         => undefined,
-               expiry_interval  => 0,
-               max_inflight     => 0,
-               topic_alias_maximum => 0
+    Attrs = #{ zone                 => Zone,
+               client_id            => ClientId,
+               conn_pid             => ClientPid,
+               clean_start          => true,
+               username             => undefined,
+               expiry_interval      => 0,
+               max_inflight         => 0,
+               topic_alias_maximum  => 0,
+               will_msg             => undefined
              },
     {ok, SessPid} = emqx_sm:open_session(Attrs),
     {reply, {ok, SessPid},
diff --git a/test/emqx_sm_SUITE.erl b/test/emqx_sm_SUITE.erl
index 110e13026..2b83b6afb 100644
--- a/test/emqx_sm_SUITE.erl
+++ b/test/emqx_sm_SUITE.erl
@@ -24,8 +24,15 @@ all() -> [t_open_close_session].
 t_open_close_session(_) ->
     emqx_ct_broker_helpers:run_setup_steps(),
     {ok, ClientPid} = emqx_mock_client:start_link(<<"client">>),
-    Attrs = #{clean_start => true, client_id => <<"client">>, conn_pid => ClientPid,
-              zone => internal, username => <<"zhou">>, expiry_interval => 0, max_inflight => 0, topic_alias_maximum => 0},
+    Attrs = #{clean_start         => true, 
+              client_id           => <<"client">>, 
+              conn_pid            => ClientPid,
+              zone                => internal, 
+              username            => <<"zhou">>, 
+              expiry_interval     => 0, 
+              max_inflight        => 0, 
+              topic_alias_maximum => 0,
+              will_msg            => undefined},
     {ok, SPid} = emqx_sm:open_session(Attrs),
     [{<<"client">>, SPid}] = emqx_sm:lookup_session(<<"client">>),
     SPid = emqx_sm:lookup_session_pid(<<"client">>),