etest_http.hrl

%% @author Johannes Huning <johannes.huning@wooga.com>
%% @doc ETest HTTP related assertions.
-ifndef(ETEST_HTTP_HRL).
-define(ETEST_HTTP_HRL, true).


%% @doc The HTTP response record.
%%      Contains all required fields for the assertions to operate on:
%%      Right now that is nothing but status-code, headers and the body.
-record (etest_http_res, {
    status  :: non_neg_integer(),
    headers :: [{binary(), binary() | string()}],
    body    :: binary()
}).


-define (perform_get(Url), ?perform_get(Url, [])).
-define (perform_get(Url, Headers), ?perform_get(Url, Headers, [])).

-define (perform_get(Url, Headers, Queries),
    ?perform_request(get, Url, Headers, Queries, <<>>)).


-define (perform_post(Url), ?perform_post(Url, [])).
-define (perform_post(Url, Headers), ?perform_post(Url, Headers, <<>>)).
-define (perform_post(Url, Headers, Body),
    ?perform_post(Url, Headers, Body, [])).

-define (perform_post(Url, Headers, Body, Queries),
    ?perform_request(post, Url, Headers, Queries, Body)).


-define (perform_request(Method, Url, Headers, Queries, Body),
((fun() ->
    case etest_http:perform_request(Method, Url, Headers, Queries, Body) of
        {error, Reason} ->
            .erlang:error({perform_request,
                [{module, ?MODULE},
                 {line,   ?LINE},
                 {request,
                    {(??Method), (??Url), (??Headers), (??Queries), (??Body)}},
                 {error, Reason}] });
        __Response -> __Response
    end
end)())).


-define (assert_contains(Haystack, Needle),
((fun() ->
    case string:str(Haystack, Needle) of
        0 -> .erlang:error({assert_contains,
                [{module,   ?MODULE},
                 {line,     ?LINE},
                 {haystack, (??Haystack)},
                 {needle,   (??Needle)}] });
        _ -> ok
    end
end)())).

-define (assert_body_contains(Res, Needle),
    ?assert_contains(Res#etest_http_res.body, Needle)).

-define (assert_body(Res, Body), ?assert_equal(Body, Res#etest_http_res.body)).


-define (assert_header(Res, HeaderName),
((fun() ->
    Headers = Res#etest_http_res.headers,
    case proplists:is_defined(HeaderName, Headers) of
        false ->
            .erlang:error({assert_header,
                [{module,   ?MODULE},
                 {line,     ?LINE},
                 {expected, (??HeaderName)}] });
        _ -> ok
    end
end)())).


-define (assert_header_val(Res, HeaderName, HeaderVal0),
((fun(HeaderVal) ->
    __Headers = Res#etest_http_res.headers,
    case proplists:get_value(HeaderName, __Headers, undefined) of
        HeaderVal -> ok;
        __V -> .erlang:error({assert_header_val,
                    [{module,   ?MODULE},
                     {line,     ?LINE},
                     {header,   (??HeaderName)},
                     {expected, (??HeaderVal0)},
                     {value,    __V}] })
    end
end)(HeaderVal0))).


-define (assert_status(Res, StatusCode0),
((fun(StatusCode) ->
    case Res#etest_http_res.status of
        StatusCode -> ok;
        __V -> .erlang:error({assert_status,
                    [{module,   ?MODULE},
                     {line,     ?LINE},
                     {expected, (??StatusCode0)},
                     {value,    __V}] })
    end
end)(StatusCode0))).


-define (assert_json(Res, JsonStruct),
((fun() ->
    __Value    = etest_http_json:decode(Res#etest_http_res.body),
    __Expected = etest_http_json:decode(etest_http_json:encode(JsonStruct)),
    ?assert_equal(__Expected, __Value)
end)())).


-define (assert_json_key(Res, Key),
((fun() ->
    __JsonStruct = etest_http_json:decode(Res#etest_http_res.body),
    case etest_http_json:fetch(Key, __JsonStruct, '__undefined__') of
        '__undefined__' ->
            .erlang:error({assert_json_key,
                [{module,   ?MODULE},
                 {line,     ?LINE},
                 {expected, (??Key)},
                 {value,    undefined}] });
        _ -> ok
    end
end)())).


-define (assert_json_val(Res, Key, Value0),
((fun(Value) ->
    __JsonStruct = etest_http_json:decode(Res#etest_http_res.body),
    case etest_http_json:fetch(Key, __JsonStruct, '__undefined__') of
        Value -> ok;
        __V -> .erlang:error({assert_json_val,
                    [{module,   ?MODULE},
                     {line,     ?LINE},
                     {expected, (??Value0)},
                     {value,    __V}] })
    end
end)(Value0))).

-endif. % ETEST_HTTP_HRL.