%% ``The contents of this file are subject to the Erlang Public License, %% Version 1.1, (the "License"); you may not use this file except in %% compliance with the License. You should have received a copy of the %% Erlang Public License along with this software. If not, it can be %% retrieved via the world wide web at http://www.erlang.org/. %% %% Software distributed under the License is distributed on an "AS IS" %% basis, WITHOUT WARRANTY OF ANY KIND, either express or implied. See %% the License for the specific language governing rights and limitations %% under the License. %% %% The Initial Developer of the Original Code is Ericsson Utvecklings AB. %% Portions created by Ericsson are Copyright 1999, Ericsson Utvecklings %% AB. All Rights Reserved.'' %% %% @author Ulf Wiger %% -module(gproc_tests). -ifdef(TEST). -include_lib("eunit/include/eunit.hrl"). -include_lib("stdlib/include/qlc.hrl"). reg_test_() -> {setup, fun() -> application:start(gproc) end, fun(_) -> application:stop(gproc) end, [ {spawn, ?_test(t_simple_reg())} , ?_test(t_is_clean()) , {spawn, ?_test(t_simple_prop())} , ?_test(t_is_clean()) , {spawn, ?_test(t_await())} , ?_test(t_is_clean()) , {spawn, ?_test(t_simple_mreg())} , ?_test(t_is_clean()) , {spawn, ?_test(t_gproc_crash())} , ?_test(t_is_clean()) , {spawn, ?_test(t_cancel_wait_and_register())} , ?_test(t_is_clean()) , {spawn, ?_test(t_give_away_to_pid())} , ?_test(t_is_clean()) , {spawn, ?_test(t_give_away_to_self())} , ?_test(t_is_clean()) , {spawn, ?_test(t_give_away_badarg())} , ?_test(t_is_clean()) , {spawn, ?_test(t_give_away_to_unknown())} , ?_test(t_is_clean()) , {spawn, ?_test(t_give_away_and_back())} , ?_test(t_is_clean()) , {spawn, ?_test(t_select())} , ?_test(t_is_clean()) , {spawn, ?_test(t_qlc())} , ?_test(t_is_clean()) , {spawn, ?_test(t_get_env())} , ?_test(t_is_clean()) , {spawn, ?_test(t_get_set_env())} , ?_test(t_is_clean()) , {spawn, ?_test(t_set_env())} , ?_test(t_is_clean()) ]}. t_simple_reg() -> ?assert(gproc:reg({n,l,name}) =:= true), ?assert(gproc:where({n,l,name}) =:= self()), ?assert(gproc:unreg({n,l,name}) =:= true), ?assert(gproc:where({n,l,name}) =:= undefined). t_simple_prop() -> ?assert(gproc:reg({p,l,prop}) =:= true), ?assert(t_other_proc(fun() -> ?assert(gproc:reg({p,l,prop}) =:= true) end) =:= ok), ?assert(gproc:unreg({p,l,prop}) =:= true). t_other_proc(F) -> {_Pid,Ref} = spawn_monitor(fun() -> exit(F()) end), receive {'DOWN',Ref,_,_,R} -> R after 10000 -> erlang:error(timeout) end. t_await() -> Me = self(), {_Pid,Ref} = spawn_monitor( fun() -> exit(?assert(gproc:await({n,l,t_await}) =:= {Me,val})) end), ?assert(gproc:reg({n,l,t_await},val) =:= true), receive {'DOWN', Ref, _, _, R} -> ?assertEqual(R, ok) after 10000 -> erlang:error(timeout) end. t_is_clean() -> sys:get_status(gproc), % in order to synch T = ets:tab2list(gproc), ?assert(T =:= []). t_simple_mreg() -> P = self(), ?assertEqual(true, gproc:mreg(n, l, [{foo, foo_val}, {bar, bar_val}])), ?assertEqual(P, gproc:where({n,l,foo})), ?assertEqual(P, gproc:where({n,l,bar})). t_gproc_crash() -> P = spawn_helper(), ?assert(gproc:where({n,l,P}) =:= P), exit(whereis(gproc), kill), give_gproc_some_time(100), ?assert(whereis(gproc) =/= undefined), %% %% Check that the registration is still there using an ets:lookup(), %% Once we've killed the process, gproc will always return undefined %% if the process is not alive, regardless of whether the registration %% is still there. So, here, the lookup should find something... %% ?assert(ets:lookup(gproc,{{n,l,P},n}) =/= []), ?assert(gproc:where({n,l,P}) =:= P), exit(P, kill), %% ...and here, it shouldn't. %% (sleep for a while first to let gproc handle the EXIT give_gproc_some_time(10), ?assert(ets:lookup(gproc,{{n,l,P},n}) =:= []). t_cancel_wait_and_register() -> Alias = {n, l, foo}, Me = self(), P = spawn(fun() -> {'EXIT',_} = (catch gproc:await(Alias, 100)), ?assert(element(1,sys:get_status(gproc)) == status), Me ! {self(), go_ahead}, timer:sleep(infinity) end), receive {P, go_ahead} -> ?assertEqual(gproc:reg(Alias, undefined), true), exit(P, kill), timer:sleep(500), ?assert(element(1,sys:get_status(gproc)) == status) end. t_give_away_to_pid() -> From = {n, l, foo}, Me = self(), P = spawn_link(fun t_loop/0), ?assertEqual(true, gproc:reg(From, undefined)), ?assertEqual(Me, gproc:where(From)), ?assertEqual(P, gproc:give_away(From, P)), ?assertEqual(P, gproc:where(From)), ?assertEqual(ok, t_call(P, die)). t_give_away_to_self() -> From = {n, l, foo}, Me = self(), ?assertEqual(true, gproc:reg(From, undefined)), ?assertEqual(Me, gproc:where(From)), ?assertEqual(Me, gproc:give_away(From, Me)), ?assertEqual(Me, gproc:where(From)), ?assertEqual(true, gproc:unreg(From)). t_give_away_badarg() -> From = {n, l, foo}, Me = self(), ?assertEqual(undefined, gproc:where(From)), ?assertError(badarg, gproc:give_away(From, Me)). t_give_away_to_unknown() -> From = {n, l, foo}, Unknown = {n, l, unknown}, Me = self(), ?assertEqual(true, gproc:reg(From, undefined)), ?assertEqual(Me, gproc:where(From)), ?assertEqual(undefined, gproc:where(Unknown)), ?assertEqual(undefined, gproc:give_away(From, Unknown)), ?assertEqual(undefined, gproc:where(From)). t_give_away_and_back() -> From = {n, l, foo}, Me = self(), P = spawn_link(fun t_loop/0), ?assertEqual(true, gproc:reg(From, undefined)), ?assertEqual(Me, gproc:where(From)), ?assertEqual(P, gproc:give_away(From, P)), ?assertEqual(P, gproc:where(From)), ?assertEqual(ok, t_call(P, {give_away, From})), ?assertEqual(Me, gproc:where(From)), ?assertEqual(ok, t_call(P, die)). t_select() -> ?assertEqual(true, gproc:reg({n, l, {n,1}}, x)), ?assertEqual(true, gproc:reg({n, l, {n,2}}, y)), ?assertEqual(true, gproc:reg({p, l, {p,1}}, x)), ?assertEqual(true, gproc:reg({p, l, {p,2}}, y)), ?assertEqual(true, gproc:reg({c, l, {c,1}}, 1)), ?assertEqual(true, gproc:reg({a, l, {c,1}}, undefined)), %% local names ?assertEqual( [{{n,l,{n,1}},self(),x}, {{n,l,{n,2}},self(),y}], gproc:select( {local,names}, [{{{n,l,'_'},'_','_'},[],['$_']}])), %% mactch local names on value ?assertEqual( [{{n,l,{n,1}},self(),x}], gproc:select( {local,names}, [{{{n,l,'_'},'_',x},[],['$_']}])), %% match all on value ?assertEqual( [{{n,l,{n,1}},self(),x}, {{p,l,{p,1}},self(),x}], gproc:select( {all,all}, [{{{'_',l,'_'},'_',x},[],['$_']}])), %% match all on pid ?assertEqual( [{{a,l,{c,1}},self(),1}, {{c,l,{c,1}},self(),1}, {{n,l,{n,1}},self(),x}, {{n,l,{n,2}},self(),y}, {{p,l,{p,1}},self(),x}, {{p,l,{p,2}},self(),y} ], gproc:select( {all,all}, [{{'_',self(),'_'},[],['$_']}])). t_qlc() -> ?assertEqual(true, gproc:reg({n, l, {n,1}}, x)), ?assertEqual(true, gproc:reg({n, l, {n,2}}, y)), ?assertEqual(true, gproc:reg({p, l, {p,1}}, x)), ?assertEqual(true, gproc:reg({p, l, {p,2}}, y)), ?assertEqual(true, gproc:reg({c, l, {c,1}}, 1)), ?assertEqual(true, gproc:reg({a, l, {c,1}}, undefined)), %% local names Exp1 = [{{n,l,{n,1}},self(),x}, {{n,l,{n,2}},self(),y}], ?assertEqual(Exp1, qlc:e(qlc:q([N || N <- gproc:table(names)]))), ?assertEqual(Exp1, qlc:e(qlc:q([N || {{n,l,_},_,_} = N <- gproc:table(names)]))), %% mactch local names on value Exp2 = [{{n,l,{n,1}},self(),x}], ?assertEqual(Exp2, qlc:e(qlc:q([N || {{n,l,_},_,x} = N <- gproc:table(names)]))), %% match all on value Exp3 = [{{n,l,{n,1}},self(),x}, {{p,l,{p,1}},self(),x}], ?assertEqual(Exp3, qlc:e(qlc:q([N || {_,_,x} = N <- gproc:table(all)]))), %% match all Exp4 = [{{a,l,{c,1}},self(),1}, {{c,l,{c,1}},self(),1}, {{n,l,{n,1}},self(),x}, {{n,l,{n,2}},self(),y}, {{p,l,{p,1}},self(),x}, {{p,l,{p,2}},self(),y} ], ?assertEqual(Exp4, qlc:e(qlc:q([X || X <- gproc:table(all)]))), %% match on pid ?assertEqual(Exp4, qlc:e(qlc:q([{K,P,V} || {K,P,V} <- gproc:table(all), P =:= self()]))), ?assertEqual(Exp4, qlc:e(qlc:q([{K,P,V} || {K,P,V} <- gproc:table(all), P == self()]))). t_get_env() -> ?assertEqual(ok, application:set_env(gproc, ssss, "s1")), ?assertEqual(true, os:putenv("SSSS", "s2")), ?assertEqual(true, os:putenv("TTTT", "s3")), ?assertEqual(ok, application:set_env(gproc, aaaa, a)), ?assertEqual(undefined, gproc:get_env(l, gproc, ssss, [])), ?assertEqual("s1", gproc:get_env(l, gproc, ssss, [app_env])), ?assertEqual("s2", gproc:get_env(l, gproc, ssss, [os_env])), ?assertEqual("s1", gproc:get_env(l, gproc, ssss, [app_env, os_env])), ?assertEqual("s3", gproc:get_env(l, gproc, ssss, [{os_env,"TTTT"}])), ?assertEqual("s4", gproc:get_env(l, gproc, ssss, [{default,"s4"}])). t_get_set_env() -> ?assertEqual(ok, application:set_env(gproc, aaaa, a)), ?assertEqual(a, gproc:get_set_env(l, gproc, aaaa, [app_env])), ?assertEqual(ok, application:set_env(gproc, aaaa, undefined)), ?assertEqual(a, gproc:get_env(l, gproc, aaaa, [error])). t_set_env() -> ?assertEqual(ok, application:set_env(gproc, aaaa, a)), ?assertEqual(a, gproc:get_set_env(l, gproc, aaaa, [app_env])), ?assertEqual(ok, application:set_env(gproc, aaaa, undefined)), ?assertEqual(b, gproc:set_env(l, gproc, aaaa, b, [app_env])), ?assertEqual({ok,b}, application:get_env(gproc, aaaa)), %% ?assertEqual(true, os:putenv("SSSS", "s0")), ?assertEqual("s0", gproc:get_env(l, gproc, ssss, [os_env])), ?assertEqual("s1", gproc:set_env(l, gproc, ssss, "s1", [os_env])), ?assertEqual("s1", os:getenv("SSSS")), ?assertEqual(true, os:putenv("SSSS", "s0")), ?assertEqual([{self(),"s1"}], gproc:lookup_values({p,l,{gproc_env,gproc,ssss}})). t_loop() -> receive {From, {give_away, Key}} -> ?assertEqual(From, gproc:give_away(Key, From)), From ! {self(), ok}, t_loop(); {From, die} -> From ! {self(), ok} end. t_call(P, Msg) -> P ! {self(), Msg}, receive {P, Reply} -> Reply end. spawn_helper() -> Parent = self(), P = spawn(fun() -> ?assert(gproc:reg({n,l,self()}) =:= true), Ref = erlang:monitor(process, Parent), Parent ! {ok,self()}, receive {'DOWN', Ref, _, _, _} -> ok end end), receive {ok,P} -> P end. give_gproc_some_time(T) -> timer:sleep(T), sys:get_status(gproc). -endif.