From 410ac9b9666b5f0f27aad7aa1c8baa3b4405a674 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Pawe=C5=82=20Chmielowski?= <pchmielowski@process-one.net>
Date: Fri, 14 Sep 2018 20:27:33 +0200
Subject: [PATCH] [tests] Update stream_id from stream:stream after auth

---
 test/suite.erl | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/test/suite.erl b/test/suite.erl
index efe587253..ecdbd71fb 100644
--- 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),