Merge pull request #40 from esl/fix-stream-features-for-stattls

read stream features after starttls or compresion
esl · Sep 8, 2014 · ea9d2d3 · ea9d2d3
2 parents af0a385 + c0ff4f7
commit ea9d2d3
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/escalus_session.erl b/src/escalus_session.erl
@@ -163,7 +163,8 @@ maybe_use_ssl(Conn, Props, Features) ->
     case use_ssl(Props, Features) of
         true ->
             {Conn1, Props1} = starttls(Conn, Props),
-            {Conn1, Props1, Features};
+            {Conn2, Props2, Features2} = stream_features(Conn1, Props1, []),
+            {Conn2, Props2, Features2};
         false ->
             {Conn, Props, Features}
     end.
@@ -189,7 +190,8 @@ maybe_use_compression(Conn, Props, Features) ->
     case can_use_compression(Props, Features) of
         true ->
             {Conn1, Props1} = compress(Conn, Props),
-            {Conn1, Props1, Features};
+            {Conn2, Props2, Features2} = stream_features(Conn1, Props1, []),
+            {Conn2, Props2, Features2};
         false ->
             {Conn, Props, Features}
     end.