From fca1c5d6ed6f5813ae23f0150e33418bc150040d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Lo=C3=AFc=20Hoguin?= <essen@ninenines.eu>
Date: Fri, 1 Aug 2014 10:59:43 +0200
Subject: Add another chunked transfer-encoding test

---
 src/cow_http_te.erl | 22 ++++++++++++++++++++--
 1 file changed, 20 insertions(+), 2 deletions(-)

(limited to 'src')

diff --git a/src/cow_http_te.erl b/src/cow_http_te.erl
index 7bd6880..5ab71f4 100644
--- a/src/cow_http_te.erl
+++ b/src/cow_http_te.erl
@@ -41,9 +41,9 @@ dripfeed(<< C, Rest/bits >>, Acc, State, F) ->
 			dripfeed(Rest, << Acc/binary, C >>, State, F);
 		{more, _, State2} ->
 			dripfeed(Rest, <<>>, State2, F);
-		{more, _, _, State2} ->
+		{more, _, Length, State2} when is_integer(Length) ->
 			dripfeed(Rest, <<>>, State2, F);
-		{more, _, _, Acc2, State2} ->
+		{more, _, Acc2, State2} ->
 			dripfeed(Rest, Acc2, State2, F);
 		{done, _, <<>>} ->
 			ok;
@@ -270,6 +270,24 @@ stream_chunked_dripfeed_test() ->
 		"0\r\n"
 		"\r\n">>, <<>>, {0, 0}, fun stream_chunked/2).
 
+do_body_to_chunks(_, <<>>, Acc) ->
+	lists:reverse([<<"0\r\n\r\n">>|Acc]);
+do_body_to_chunks(ChunkSize, Body, Acc) ->
+	BodySize = byte_size(Body),
+	ChunkSize2 = case BodySize < ChunkSize of
+		true -> BodySize;
+		false -> ChunkSize
+	end,
+	<< Chunk:ChunkSize2/binary, Rest/binary >> = Body,
+	ChunkSizeBin = list_to_binary(integer_to_list(ChunkSize2, 16)),
+	do_body_to_chunks(ChunkSize, Rest,
+		[<< ChunkSizeBin/binary, "\r\n", Chunk/binary, "\r\n" >>|Acc]).
+
+stream_chunked_dripfeed2_test() ->
+	Body = list_to_binary(io_lib:format("~p", [lists:seq(1, 100)])),
+	Body2 = iolist_to_binary(do_body_to_chunks(50, Body, [])),
+	dripfeed(Body2, <<>>, {0, 0}, fun stream_chunked/2).
+
 stream_chunked_error_test_() ->
 	Tests = [
 		{<<>>, undefined},
-- 
cgit v1.2.3