%%
%%   Copyright (c) 2016, Dmitry Kolesnikov
%%   Copyright (c) 2016, Mario Cardona
%%   All Rights Reserved.
%%
%%   Licensed under the Apache License, Version 2.0 (the "License");
%%   you may not use this file except in compliance with the License.
%%   You may obtain a copy of the License at
%%
%%       http://www.apache.org/licenses/LICENSE-2.0
%%
%%   Unless required by applicable law or agreed to in writing, software
%%   distributed under the License is distributed on an "AS IS" BASIS,
%%   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
%%   See the License for the specific language governing permissions and
%%   limitations under the License.
%%
%% @doc
%%   mock networking stack
-module(m_http_mock).

-export([
   init/0
,  init/3
,  free/0
,  response/3
]).

%%
%%
init() ->
   meck:new(hackney, [unstick]).

%% @todo: mock input via routing function fun(Url) -> {Code, Head, Body}
init(Code, Head, Body) ->
   init(),
   response(Code, Head, Body).

%%
%%
free() ->
   meck:unload(hackney).

%%
%%
response(Code, Head, Body) ->
   meck:expect(hackney, request,
      fun(Mthd, Url, ReqHead, ReqBody, _Opts) ->
         MockHead = [
            {<<"X-Mock-Mthd">>, typecast:s(Mthd)}
         ,  {<<"X-Mock-Url">>,  typecast:s(Url)}
         ,  {<<"X-Mock-Body">>, typecast:s(ReqBody)}
         |  [{<<"X-Mock-", H/binary>>, V}  || {H, V} <- ReqHead]
         ], 
         {ok, Code, Head ++ MockHead, undefined}
      end
   ),
   stream_http(Body).

stream_http([]) ->
   meck:expect(hackney, stream_body, fun(_) -> done end);

stream_http([Head | Tail]) ->
   meck:expect(hackney, stream_body, fun(_) -> stream_http(Tail), {ok, Head} end).