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
96
97
98
99
100
101
102
103
104
105
|
%% 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.
%%
%% Alternatively, you may use this file under the terms of the GNU Lesser
%% General Public License (the "LGPL") as published by the Free Software
%% Foundation; either version 2.1, or (at your option) any later version.
%% If you wish to allow use of your version of this file only under the
%% terms of the LGPL, you should delete the provisions above and replace
%% them with the notice and other provisions required by the LGPL; see
%% <http://www.gnu.org/licenses/>. If you do not delete the provisions
%% above, a recipient may use your version of this file under the terms of
%% either the Apache License or the LGPL.
%%
%% @author Richard Carlsson <[email protected]>
%% @copyright 2006 Richard Carlsson
%% @private
%% @see eunit
%% @doc Parse transform for automatic exporting of test functions.
-module(eunit_autoexport).
-include("eunit_internal.hrl").
-export([parse_transform/2]).
parse_transform(Forms, Options) ->
TestSuffix = proplists:get_value(eunit_test_suffix, Options,
?DEFAULT_TEST_SUFFIX),
GeneratorSuffix = proplists:get_value(eunit_generator_suffix,
Options,
?DEFAULT_GENERATOR_SUFFIX),
ExportSuffix = proplists:get_value(eunit_export_suffix, Options,
?DEFAULT_EXPORT_SUFFIX),
F = fun (Form, Set) ->
form(Form, Set, TestSuffix, GeneratorSuffix,
ExportSuffix)
end,
Exports = sets:to_list(lists:foldl(F, sets:new(), Forms)),
rewrite(Forms, Exports).
form({function, _L, Name, 0, _Cs}, S, TestSuffix, GeneratorSuffix,
ExportSuffix) ->
N = atom_to_list(Name),
case lists:suffix(TestSuffix, N) of
true ->
sets:add_element({Name, 0}, S);
false ->
case lists:suffix(GeneratorSuffix, N) of
true ->
sets:add_element({Name, 0}, S);
false ->
case lists:suffix(ExportSuffix, N) of
true ->
sets:add_element({Name, 0}, S);
false ->
S
end
end
end;
form({function, _L, ?DEFAULT_MODULE_WRAPPER_NAME, 1, _Cs}, S, _, _, _) ->
sets:add_element({?DEFAULT_MODULE_WRAPPER_NAME,1}, S);
form(_, S, _, _, _) ->
S.
rewrite([{attribute,_,module,{Name,_Ps}}=M | Fs], Exports) ->
module_decl(Name, M, Fs, Exports);
rewrite([{attribute,_,module,Name}=M | Fs], Exports) ->
module_decl(Name, M, Fs, Exports);
rewrite([F | Fs], Exports) ->
[F | rewrite(Fs, Exports)];
rewrite([], _Exports) ->
[]. %% fail-safe, in case there is no module declaration
rewrite([{function,_,test,0,_}=F | Fs], As, Module, _Test) ->
rewrite(Fs, [F | As], Module, false);
rewrite([F | Fs], As, Module, Test) ->
rewrite(Fs, [F | As], Module, Test);
rewrite([], As, Module, Test) ->
L = erl_anno:new(0),
{if Test ->
[{function,L,test,0,
[{clause,L,[],[],
[{call,L,{remote,L,{atom,L,eunit},{atom,L,test}},
[{atom,L,Module}]}]}]}
| As];
true ->
As
end,
Test}.
module_decl(Name, M, Fs, Exports) ->
Module = Name,
{Fs1, Test} = rewrite(Fs, [], Module, true),
Es = if Test -> [{test,0} | Exports];
true -> Exports
end,
[M, {attribute,erl_anno:new(0),export,Es} | lists:reverse(Fs1)].
|
