From d6f9ce49c41e60e00200f7fb539d1d07ea1a59f0 Mon Sep 17 00:00:00 2001
From: Juan Facorro <juan.facorro@gmail.com>
Date: Tue, 2 Dec 2014 16:39:37 -0300
Subject: [PATCH] [Fixes #53 #54].

---
 src/shotgun.erl | 21 +++++++++++----------
 1 file changed, 11 insertions(+), 10 deletions(-)

diff --git a/src/shotgun.erl b/src/shotgun.erl
index bd0e859..ede2ad3 100644
--- a/src/shotgun.erl
+++ b/src/shotgun.erl
@@ -347,15 +347,16 @@ wait_response({gun_response, _Pid, _StreamRef, fin, StatusCode, Headers},
         end,
     {next_state, at_rest, State#{responses => NewResponses}};
 wait_response({gun_response, _Pid, _StreamRef, nofin, StatusCode, Headers},
-              #{from := From, stream := StreamRef} = State) ->
-    StateName = case lists:keyfind(<<"transfer-encoding">>, 1, Headers) of
-                    {<<"transfer-encoding">>, <<"chunked">>} ->
-                        Result = {ok, StreamRef},
-                        gen_fsm:reply(From, Result),
-                        receive_chunk;
-                    _ ->
-                        receive_data
-                end,
+              #{from := From, stream := StreamRef, async := Async} = State) ->
+    StateName =
+      case lists:keyfind(<<"transfer-encoding">>, 1, Headers) of
+        {<<"transfer-encoding">>, <<"chunked">>} when Async == true->
+          Result = {ok, StreamRef},
+          gen_fsm:reply(From, Result),
+          receive_chunk;
+        _ ->
+          receive_data
+      end,
     {
       next_state,
       StateName,
@@ -417,7 +418,7 @@ clean_state() ->
        status_code => undefined,
        headers => undefined,
        async => false,
-       chunk_mode => binary,
+       async_mode => binary,
        buffer => <<"">>
      }.