1 files changed, 104 insertions, 0 deletions
diff --git a/lib/eunit/src/eunit_autoexport.erl b/lib/eunit/src/eunit_autoexport.erl
new file mode 100644
index 0000000000..7b153c1194
--- /dev/null
+++ b/lib/eunit/src/eunit_autoexport.erl
@@ -0,0 +1,104 @@
+%% This library is free software; you can redistribute it and/or modify
+%% it under the terms of the GNU Lesser General Public License as
+%% published by the Free Software Foundation; either version 2 of the
+%% License, or (at your option) any later version.
+%%
+%% This library is distributed in the hope that it will be useful, but
+%% WITHOUT ANY WARRANTY; without even the implied warranty of
+%% MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+%% Lesser General Public License for more details.
+%%
+%% You should have received a copy of the GNU Lesser General Public
+%% License along with this library; if not, write to the Free Software
+%% Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
+%% USA
+%%
+%% $Id: eunit_autoexport.erl 329 2009-03-01 11:23:32Z rcarlsson $ 
+%%
+%% @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) ->
+    {if Test ->
+	     EUnit = {record_field,0,{atom,0,''},{atom,0,eunit}},
+	     [{function,0,test,0,
+	       [{clause,0,[],[],
+		 [{call,0,{remote,0,EUnit,{atom,0,test}},
+		   [{atom,0,Module}]}]}]}
+	      | As];
+	true ->
+	     As
+     end,
+     Test}.
+
+module_decl(Name, M, Fs, Exports) ->
+    Module = if is_atom(Name) -> Name;
+		true -> list_to_atom(packages:concat(Name))
+	     end,
+    {Fs1, Test} = rewrite(Fs, [], Module, true),
+    Es = if Test -> [{test,0} | Exports];
+	    true -> Exports
+	 end,
+    [M, {attribute,0,export,Es} | lists:reverse(Fs1)].