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
95
|
% 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(cpse_gather).
-export([
module/1
]).
module(ModName) ->
Exports = ModName:module_info(exports),
SetupMod = get_setup_all(ModName, Exports),
TeardownMod = get_teardown_all(ModName, Exports),
SetupTest = get_fun(ModName, setup_each, 0, Exports),
TeardownTest = get_fun(ModName, teardown_each, 1, Exports),
RevTests = lists:foldl(fun({Fun, Arity}, Acc) ->
case {atom_to_list(Fun), Arity} of
{[$c, $p, $s, $e, $_ | _], Arity} when Arity == 0; Arity == 1 ->
TestFun = make_test_fun(ModName, Fun, Arity),
[TestFun | Acc];
_ ->
Acc
end
end, [], Exports),
Tests = lists:reverse(RevTests),
{
setup,
spawn,
SetupMod,
TeardownMod,
[
{
foreach,
SetupTest,
TeardownTest,
Tests
}
]
}.
get_setup_all(ModName, Exports) ->
case lists:member({setup_all, 0}, Exports) of
true -> fun ModName:setup_all/0;
false -> fun cpse_util:setup_all/0
end.
get_teardown_all(ModName, Exports) ->
case lists:member({teardown_all, 1}, Exports) of
true -> fun ModName:teardown_all/1;
false -> fun cpse_util:teardown_all/1
end.
get_fun(ModName, FunName, Arity, Exports) ->
case lists:member({FunName, Arity}, Exports) of
true -> fun ModName:FunName/Arity;
false when Arity == 0 -> fun() -> ok end;
false when Arity == 1 -> fun(_) -> ok end
end.
make_test_fun(Module, Fun, Arity) ->
Name = atom_to_list(Fun),
case Arity of
0 ->
fun(_) ->
{timeout, 60, {Name, fun() ->
process_flag(trap_exit, true),
Module:Fun()
end}}
end;
1 ->
fun(Arg) ->
{timeout, 60, {Name, fun() ->
process_flag(trap_exit, true),
Module:Fun(Arg)
end}}
end
end.
|
