Skip to content
Fetching contributors…
Cannot retrieve contributors at this time
96 lines (80 sloc) 2.44 KB
% 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.
-module(test_util).
-export([init_code_path/0]).
-export([source_file/1, build_file/1, config_files/0]).
-export([run/2]).
-export([request/3, request/4]).
-define(REQ_TIMEOUT, 30000).
srcdir() ->
"@abs_top_srcdir@".
builddir() ->
"@abs_top_builddir@".
init_code_path() ->
Paths = [
"etap",
"couchdb",
"ejson",
"erlang-oauth",
"lhttpc",
"mochiweb",
"snappy"
],
lists:foreach(fun(Name) ->
code:add_patha(filename:join([builddir(), "src", Name]))
end, Paths).
source_file(Name) ->
filename:join([srcdir(), Name]).
build_file(Name) ->
filename:join([builddir(), Name]).
config_files() ->
[
build_file("etc/couchdb/default_dev.ini"),
source_file("test/etap/random_port.ini"),
build_file("etc/couchdb/local_dev.ini")
].
run(Plan, Fun) ->
test_util:init_code_path(),
etap:plan(Plan),
case (catch Fun()) of
ok ->
etap:end_tests();
Other ->
etap:diag(io_lib:format("Test died abnormally:~n~p", [Other])),
timer:sleep(500),
etap:bail(Other)
end,
ok.
request(Url, Headers, Method) ->
request(Url, Headers, Method, []).
request(Url, Headers, Method, Body) ->
request(Url, Headers, Method, Body, 3).
request(_Url, _Headers, _Method, _Body, 0) ->
{error, request_failed};
request(Url, Headers, Method, Body, N) ->
case code:is_loaded(lhttpc) of
false ->
_ = ssl:start(),
ok = lhttpc:start();
_ ->
ok
end,
case lhttpc:request(Url, Method, Headers, Body, ?REQ_TIMEOUT) of
{ok, {{Code, _}, RespHeaders, RespBody}} ->
{ok, Code, RespHeaders, RespBody};
{error, connection_closed} ->
% Connection closed right after a successful request that
% used the same connection.
request(Url, Headers, Method, Body, N - 1);
Error ->
Error
end.
Jump to Line
Something went wrong with that request. Please try again.