From c25bd1c886cb47171a8da99766c35394de850c27 Mon Sep 17 00:00:00 2001
From: Eduardo Gurgel <edgurgel@gmail.com>
Date: Sun, 5 Jan 2014 16:36:01 -0300
Subject: [PATCH] Add '/stream/:n'

---
 lib/httparrot.ex             |  1 +
 test/stream_handler_test.exs | 67 ++++++++++++++++++++++++++++++++++++
 2 files changed, 68 insertions(+)
 create mode 100644 test/stream_handler_test.exs

diff --git a/lib/httparrot.ex b/lib/httparrot.ex
index 5a37293..19a89c6 100644
--- a/lib/httparrot.ex
+++ b/lib/httparrot.ex
@@ -18,6 +18,7 @@ defmodule HTTParrot do
              {'/cookies/delete', HTTParrot.DeleteCookiesHandler, []},
              {'/basic-auth/:user/:passwd', HTTParrot.BasicAuthHandler, []},
              {'/hidden-basic-auth/:user/:passwd', HTTParrot.HiddenBasicAuthHandler, []},
+             {'/stream/:n', HTTParrot.StreamHandler, []},
              {'/html', :cowboy_static, {:priv_file, :httparrot, "html.html"}} ] }
     ])
     {:ok, port} = :application.get_env(:httparrot, :port)
diff --git a/test/stream_handler_test.exs b/test/stream_handler_test.exs
new file mode 100644
index 0000000..684173b
--- /dev/null
+++ b/test/stream_handler_test.exs
@@ -0,0 +1,67 @@
+defmodule HTTParrot.StreamHandlerTest do
+  use ExUnit.Case
+  import :meck
+  import HTTParrot.StreamHandler
+
+  setup do
+    new :cowboy_req
+    new HTTParrot.GeneralRequestInfo
+    new JSEX
+  end
+
+  teardown do
+    unload :cowboy_req
+    unload HTTParrot.GeneralRequestInfo
+    unload JSEX
+  end
+
+  test "malformed_request returns false if it's not an integer" do
+    expect(:cowboy_req, :binding, [{[:n, :req1], {"a2B=", :req2}}])
+
+    assert malformed_request(:req1, :state) == {true, :req2, :state}
+
+    assert validate :cowboy_req
+  end
+
+  test "malformed_request returns false if it's an integer" do
+    expect(:cowboy_req, :binding, [{[:n, :req1], {"2", :req2}}])
+
+    assert malformed_request(:req1, :state) == {false, :req2, 2}
+
+    assert validate :cowboy_req
+  end
+
+  test "malformed_request returns 1 if 'n' is less than 1" do
+    expect(:cowboy_req, :binding, [{[:n, :req1], {"0", :req2}}])
+
+    assert malformed_request(:req1, :state) == {false, :req2, 1}
+
+    assert validate :cowboy_req
+  end
+
+  test "malformed_request returns 100 if 'n' is greater than 100" do
+    expect(:cowboy_req, :binding, [{[:n, :req1], {"200", :req2}}])
+
+    assert malformed_request(:req1, :state) == {false, :req2, 100}
+
+    assert validate :cowboy_req
+  end
+
+  test "response must stream chunks" do
+    expect(HTTParrot.GeneralRequestInfo, :retrieve, 1, {[:info], :req2})
+    expect(JSEX, :encode!, [{[[{:id, 0}, :info]], :json1},
+                            {[[{:id, 1}, :info]], :json2}])
+
+    assert {{:chunked, func}, :req2, nil} = get_json(:req1, 2)
+    assert is_function(func)
+
+    send_func = fn(body) -> self <- {:chunk, body} end
+    func.(send_func)
+
+    assert_receive {:chunk, :json1}
+    assert_receive {:chunk, :json2}
+
+    assert validate HTTParrot.GeneralRequestInfo
+    assert validate JSEX
+  end
+end