1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
|
#!/usr/bin/env escript
%% -*- erlang -*-
%%! -name wm_test@127.0.0.1
main([Ebin]) ->
code:add_path(Ebin),
code:add_paths(filelib:wildcard("deps/*/ebin", Ebin)),
code:add_paths(filelib:wildcard("deps/*/deps/*/ebin", Ebin)),
{ok, [{application, webmachine, App}]} =
file:consult(filename:join([Ebin, "webmachine.app"])),
{ok, NonTestRe} = re:compile("_tests$"),
Modules = lists:filter(
fun(M) when M =:= gen_server2 ->
false;
(M) when M =:= priority_queue ->
false;
(M) when M =:= json_pp ->
false;
(M) ->
nomatch == re:run(atom_to_list(M), NonTestRe)
end,
proplists:get_value(modules, App)),
start_cover(Modules),
eunit:test(Modules, [verbose]),
analyze_cover(Modules);
main(_) ->
io:format("usage: run_tests.escript RIAK_EBIN_DIRECTORY~n"),
halt(1).
start_cover(Modules) ->
{ok, _Cover} = cover:start(),
io:format("Cover compiling...~n"),
Compiled = [ M || {ok, M} <- [ cover:compile(
filename:join(["src",atom_to_list(M)]))
|| M <- Modules ] ],
case length(Modules) == length(Compiled) of
true -> ok;
false ->
io:format("Warning: the following modules were not"
" cover-compiled:~n ~p~n", [Compiled])
end.
analyze_cover(Modules) ->
io:format("Analyzing cover...~n"),
CoverBase = filename:join(["test", "cover"]),
ok = filelib:ensure_dir(filename:join([CoverBase, "fake"])),
Coverages = lists:foldl(
fun(M, Acc) ->
[analyze_module(CoverBase, M)|Acc]
end,
[], Modules),
IndexFilename = filename:join([CoverBase, "index.html"]),
{ok, Index} = file:open(IndexFilename, [write]),
{LineTotal, CoverTotal} =
lists:foldl(fun({_,_,Lines,Covered}, {LineAcc, CovAcc}) ->
{LineAcc+Lines, CovAcc+Covered}
end, {0,0}, Coverages),
file:write(Index,
"<html><head><title>Riak Coverage</title></head>\n"
"<body><h1>Riak Coverage</h1><ul>\n"),
file:write(Index,
io_lib:format("<h2>Total: ~.2f%</h2>\n",
[percentage(CoverTotal, LineTotal)])),
[ file:write(Index,
io_lib:format(
"<li><a href=\"~s\">~p</a>: ~.2f%</li>~n",
[Filename, Module, percentage(Covered, Lines)]))
|| {Filename, Module, Lines, Covered} <- Coverages ],
file:write(Index,"</ul></body></html>"),
file:close(Index),
io:format("Cover analysis in ~s~n", [IndexFilename]).
analyze_module(CoverBase, Module) ->
{ok, Filename} =
cover:analyze_to_file(
Module,
filename:join(CoverBase, atom_to_list(Module)++".COVER.html"),
[html]),
Lines = count_lines(Filename, "[[:digit:]]\.\.|"),
Covered = count_lines(Filename, "[[:space:]]0\.\.|"),
{filename:basename(Filename), Module, Lines, Lines-Covered}.
count_lines(Filename, Pattern) ->
{ok, [Lines],_} = io_lib:fread(
"~d",
os:cmd(io_lib:format("grep -e \"~s\" ~s | wc -l",
[Pattern, Filename]))),
Lines.
percentage(_, 0) -> 1000.0;
percentage(Part, Total) ->
(Part/Total)*100.
|
