[tests] Update stream_id from stream:stream after auth

diff --git a/test/suite.erl b/test/suite.erl
index efe587253452a0bffd25a34832d9c89c408c818f..ecdbd71fb1266441eae1a1fb61f4c90ded8e2011 100644 (file)
--- a/test/suite.erl
+++ b/test/suite.erl
@@ -471,7 +471,9 @@ wait_auth_SASL_result(Config, ShouldFail) ->
            NS = if Type == client -> ?NS_CLIENT;
                    Type == server -> ?NS_SERVER
                 end,
-           receive #stream_start{xmlns = NS, version = {1,0}} -> ok end,
+           Config2 = receive #stream_start{id = ID, xmlns = NS, version = {1,0}} ->
+               set_opt(stream_id, ID, Config)
+           end,
             receive #stream_features{sub_els = Fs} ->
                    if Type == client ->
                            #xmpp_session{optional = true} =
@@ -488,7 +490,7 @@ wait_auth_SASL_result(Config, ShouldFail) ->
                              set_opt(rosterver, true, ConfigAcc);
                         (_, ConfigAcc) ->
                              ConfigAcc
-                     end, Config, Fs)
+                     end, Config2, Fs)
            end;
         #sasl_challenge{text = ClientIn} ->
             {Response, SASL} = (?config(sasl, Config))(ClientIn),