gproc_tests.erl 12 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375
  1. %% ``The contents of this file are subject to the Erlang Public License,
  2. %% Version 1.1, (the "License"); you may not use this file except in
  3. %% compliance with the License. You should have received a copy of the
  4. %% Erlang Public License along with this software. If not, it can be
  5. %% retrieved via the world wide web at http://www.erlang.org/.
  6. %%
  7. %% Software distributed under the License is distributed on an "AS IS"
  8. %% basis, WITHOUT WARRANTY OF ANY KIND, either express or implied. See
  9. %% the License for the specific language governing rights and limitations
  10. %% under the License.
  11. %%
  12. %% The Initial Developer of the Original Code is Ericsson Utvecklings AB.
  13. %% Portions created by Ericsson are Copyright 1999, Ericsson Utvecklings
  14. %% AB. All Rights Reserved.''
  15. %%
  16. %% @author Ulf Wiger <ulf.wiger@erlang-solutions.com>
  17. %%
  18. -module(gproc_tests).
  19. -ifdef(TEST).
  20. -include_lib("eunit/include/eunit.hrl").
  21. -include_lib("stdlib/include/qlc.hrl").
  22. reg_test_() ->
  23. {setup,
  24. fun() ->
  25. application:start(gproc)
  26. end,
  27. fun(_) ->
  28. application:stop(gproc)
  29. end,
  30. [
  31. {spawn, ?_test(t_simple_reg())}
  32. , ?_test(t_is_clean())
  33. , {spawn, ?_test(t_simple_prop())}
  34. , ?_test(t_is_clean())
  35. , {spawn, ?_test(t_await())}
  36. , ?_test(t_is_clean())
  37. , {spawn, ?_test(t_simple_mreg())}
  38. , ?_test(t_is_clean())
  39. , {spawn, ?_test(t_gproc_crash())}
  40. , ?_test(t_is_clean())
  41. , {spawn, ?_test(t_cancel_wait_and_register())}
  42. , ?_test(t_is_clean())
  43. , {spawn, ?_test(t_give_away_to_pid())}
  44. , ?_test(t_is_clean())
  45. , {spawn, ?_test(t_give_away_to_self())}
  46. , ?_test(t_is_clean())
  47. , {spawn, ?_test(t_give_away_badarg())}
  48. , ?_test(t_is_clean())
  49. , {spawn, ?_test(t_give_away_to_unknown())}
  50. , ?_test(t_is_clean())
  51. , {spawn, ?_test(t_give_away_and_back())}
  52. , ?_test(t_is_clean())
  53. , {spawn, ?_test(t_select())}
  54. , ?_test(t_is_clean())
  55. , {spawn, ?_test(t_qlc())}
  56. , ?_test(t_is_clean())
  57. , {spawn, ?_test(t_get_env())}
  58. , ?_test(t_is_clean())
  59. , {spawn, ?_test(t_get_set_env())}
  60. , ?_test(t_is_clean())
  61. , {spawn, ?_test(t_set_env())}
  62. , ?_test(t_is_clean())
  63. , {spawn, ?_test(t_get_env_inherit())}
  64. , ?_test(t_is_clean())
  65. ]}.
  66. t_simple_reg() ->
  67. ?assert(gproc:reg({n,l,name}) =:= true),
  68. ?assert(gproc:where({n,l,name}) =:= self()),
  69. ?assert(gproc:unreg({n,l,name}) =:= true),
  70. ?assert(gproc:where({n,l,name}) =:= undefined).
  71. t_simple_prop() ->
  72. ?assert(gproc:reg({p,l,prop}) =:= true),
  73. ?assert(t_other_proc(fun() ->
  74. ?assert(gproc:reg({p,l,prop}) =:= true)
  75. end) =:= ok),
  76. ?assert(gproc:unreg({p,l,prop}) =:= true).
  77. t_other_proc(F) ->
  78. {_Pid,Ref} = spawn_monitor(fun() -> exit(F()) end),
  79. receive
  80. {'DOWN',Ref,_,_,R} ->
  81. R
  82. after 10000 ->
  83. erlang:error(timeout)
  84. end.
  85. t_await() ->
  86. Me = self(),
  87. {_Pid,Ref} = spawn_monitor(
  88. fun() -> exit(?assert(gproc:await({n,l,t_await}) =:= {Me,val})) end),
  89. ?assert(gproc:reg({n,l,t_await},val) =:= true),
  90. receive
  91. {'DOWN', Ref, _, _, R} ->
  92. ?assertEqual(R, ok)
  93. after 10000 ->
  94. erlang:error(timeout)
  95. end.
  96. t_is_clean() ->
  97. sys:get_status(gproc), % in order to synch
  98. T = ets:tab2list(gproc),
  99. ?assert(T =:= []).
  100. t_simple_mreg() ->
  101. P = self(),
  102. ?assertEqual(true, gproc:mreg(n, l, [{foo, foo_val},
  103. {bar, bar_val}])),
  104. ?assertEqual(P, gproc:where({n,l,foo})),
  105. ?assertEqual(P, gproc:where({n,l,bar})).
  106. t_gproc_crash() ->
  107. P = spawn_helper(),
  108. ?assert(gproc:where({n,l,P}) =:= P),
  109. exit(whereis(gproc), kill),
  110. give_gproc_some_time(100),
  111. ?assert(whereis(gproc) =/= undefined),
  112. %%
  113. %% Check that the registration is still there using an ets:lookup(),
  114. %% Once we've killed the process, gproc will always return undefined
  115. %% if the process is not alive, regardless of whether the registration
  116. %% is still there. So, here, the lookup should find something...
  117. %%
  118. ?assert(ets:lookup(gproc,{{n,l,P},n}) =/= []),
  119. ?assert(gproc:where({n,l,P}) =:= P),
  120. exit(P, kill),
  121. %% ...and here, it shouldn't.
  122. %% (sleep for a while first to let gproc handle the EXIT
  123. give_gproc_some_time(10),
  124. ?assert(ets:lookup(gproc,{{n,l,P},n}) =:= []).
  125. t_cancel_wait_and_register() ->
  126. Alias = {n, l, foo},
  127. Me = self(),
  128. P = spawn(fun() ->
  129. {'EXIT',_} = (catch gproc:await(Alias, 100)),
  130. ?assert(element(1,sys:get_status(gproc)) == status),
  131. Me ! {self(), go_ahead},
  132. timer:sleep(infinity)
  133. end),
  134. receive
  135. {P, go_ahead} ->
  136. ?assertEqual(gproc:reg(Alias, undefined), true),
  137. exit(P, kill),
  138. timer:sleep(500),
  139. ?assert(element(1,sys:get_status(gproc)) == status)
  140. end.
  141. t_give_away_to_pid() ->
  142. From = {n, l, foo},
  143. Me = self(),
  144. P = spawn_link(fun t_loop/0),
  145. ?assertEqual(true, gproc:reg(From, undefined)),
  146. ?assertEqual(Me, gproc:where(From)),
  147. ?assertEqual(P, gproc:give_away(From, P)),
  148. ?assertEqual(P, gproc:where(From)),
  149. ?assertEqual(ok, t_call(P, die)).
  150. t_give_away_to_self() ->
  151. From = {n, l, foo},
  152. Me = self(),
  153. ?assertEqual(true, gproc:reg(From, undefined)),
  154. ?assertEqual(Me, gproc:where(From)),
  155. ?assertEqual(Me, gproc:give_away(From, Me)),
  156. ?assertEqual(Me, gproc:where(From)),
  157. ?assertEqual(true, gproc:unreg(From)).
  158. t_give_away_badarg() ->
  159. From = {n, l, foo},
  160. Me = self(),
  161. ?assertEqual(undefined, gproc:where(From)),
  162. ?assertError(badarg, gproc:give_away(From, Me)).
  163. t_give_away_to_unknown() ->
  164. From = {n, l, foo},
  165. Unknown = {n, l, unknown},
  166. Me = self(),
  167. ?assertEqual(true, gproc:reg(From, undefined)),
  168. ?assertEqual(Me, gproc:where(From)),
  169. ?assertEqual(undefined, gproc:where(Unknown)),
  170. ?assertEqual(undefined, gproc:give_away(From, Unknown)),
  171. ?assertEqual(undefined, gproc:where(From)).
  172. t_give_away_and_back() ->
  173. From = {n, l, foo},
  174. Me = self(),
  175. P = spawn_link(fun t_loop/0),
  176. ?assertEqual(true, gproc:reg(From, undefined)),
  177. ?assertEqual(Me, gproc:where(From)),
  178. ?assertEqual(P, gproc:give_away(From, P)),
  179. ?assertEqual(P, gproc:where(From)),
  180. ?assertEqual(ok, t_call(P, {give_away, From})),
  181. ?assertEqual(Me, gproc:where(From)),
  182. ?assertEqual(ok, t_call(P, die)).
  183. t_select() ->
  184. ?assertEqual(true, gproc:reg({n, l, {n,1}}, x)),
  185. ?assertEqual(true, gproc:reg({n, l, {n,2}}, y)),
  186. ?assertEqual(true, gproc:reg({p, l, {p,1}}, x)),
  187. ?assertEqual(true, gproc:reg({p, l, {p,2}}, y)),
  188. ?assertEqual(true, gproc:reg({c, l, {c,1}}, 1)),
  189. ?assertEqual(true, gproc:reg({a, l, {c,1}}, undefined)),
  190. %% local names
  191. ?assertEqual(
  192. [{{n,l,{n,1}},self(),x},
  193. {{n,l,{n,2}},self(),y}], gproc:select(
  194. {local,names},
  195. [{{{n,l,'_'},'_','_'},[],['$_']}])),
  196. %% mactch local names on value
  197. ?assertEqual(
  198. [{{n,l,{n,1}},self(),x}], gproc:select(
  199. {local,names},
  200. [{{{n,l,'_'},'_',x},[],['$_']}])),
  201. %% match all on value
  202. ?assertEqual(
  203. [{{n,l,{n,1}},self(),x},
  204. {{p,l,{p,1}},self(),x}], gproc:select(
  205. {all,all},
  206. [{{{'_',l,'_'},'_',x},[],['$_']}])),
  207. %% match all on pid
  208. ?assertEqual(
  209. [{{a,l,{c,1}},self(),1},
  210. {{c,l,{c,1}},self(),1},
  211. {{n,l,{n,1}},self(),x},
  212. {{n,l,{n,2}},self(),y},
  213. {{p,l,{p,1}},self(),x},
  214. {{p,l,{p,2}},self(),y}
  215. ], gproc:select(
  216. {all,all},
  217. [{{'_',self(),'_'},[],['$_']}])).
  218. t_qlc() ->
  219. ?assertEqual(true, gproc:reg({n, l, {n,1}}, x)),
  220. ?assertEqual(true, gproc:reg({n, l, {n,2}}, y)),
  221. ?assertEqual(true, gproc:reg({p, l, {p,1}}, x)),
  222. ?assertEqual(true, gproc:reg({p, l, {p,2}}, y)),
  223. ?assertEqual(true, gproc:reg({c, l, {c,1}}, 1)),
  224. ?assertEqual(true, gproc:reg({a, l, {c,1}}, undefined)),
  225. %% local names
  226. Exp1 = [{{n,l,{n,1}},self(),x},
  227. {{n,l,{n,2}},self(),y}],
  228. ?assertEqual(Exp1,
  229. qlc:e(qlc:q([N || N <- gproc:table(names)]))),
  230. ?assertEqual(Exp1,
  231. qlc:e(qlc:q([N || {{n,l,_},_,_} = N <- gproc:table(names)]))),
  232. %% mactch local names on value
  233. Exp2 = [{{n,l,{n,1}},self(),x}],
  234. ?assertEqual(Exp2,
  235. qlc:e(qlc:q([N || {{n,l,_},_,x} = N <- gproc:table(names)]))),
  236. %% match all on value
  237. Exp3 = [{{n,l,{n,1}},self(),x},
  238. {{p,l,{p,1}},self(),x}],
  239. ?assertEqual(Exp3,
  240. qlc:e(qlc:q([N || {_,_,x} = N <- gproc:table(all)]))),
  241. %% match all
  242. Exp4 = [{{a,l,{c,1}},self(),1},
  243. {{c,l,{c,1}},self(),1},
  244. {{n,l,{n,1}},self(),x},
  245. {{n,l,{n,2}},self(),y},
  246. {{p,l,{p,1}},self(),x},
  247. {{p,l,{p,2}},self(),y}
  248. ],
  249. ?assertEqual(Exp4,
  250. qlc:e(qlc:q([X || X <- gproc:table(all)]))),
  251. %% match on pid
  252. ?assertEqual(Exp4,
  253. qlc:e(qlc:q([{K,P,V} || {K,P,V} <-
  254. gproc:table(all), P =:= self()]))),
  255. ?assertEqual(Exp4,
  256. qlc:e(qlc:q([{K,P,V} || {K,P,V} <-
  257. gproc:table(all), P == self()]))).
  258. t_get_env() ->
  259. ?assertEqual(ok, application:set_env(gproc, ssss, "s1")),
  260. ?assertEqual(true, os:putenv("SSSS", "s2")),
  261. ?assertEqual(true, os:putenv("TTTT", "s3")),
  262. ?assertEqual(ok, application:set_env(gproc, aaaa, a)),
  263. ?assertEqual(undefined, gproc:get_env(l, gproc, ssss, [])),
  264. %%
  265. ?assertEqual("s1", gproc:get_env(l, gproc, ssss, [app_env])),
  266. ?assertEqual("s2", gproc:get_env(l, gproc, ssss, [os_env])),
  267. ?assertEqual("s1", gproc:get_env(l, gproc, ssss, [app_env, os_env])),
  268. ?assertEqual("s3", gproc:get_env(l, gproc, ssss, [{os_env,"TTTT"}])),
  269. ?assertEqual("s4", gproc:get_env(l, gproc, ssss, [{default,"s4"}])),
  270. %%
  271. ?assertEqual(ok, application:start(mnesia)),
  272. ?assertEqual({atomic,ok}, mnesia:create_table(t, [{ram_copies, [node()]}])),
  273. ?assertEqual(ok, mnesia:dirty_write({t, foo, bar})),
  274. ?assertEqual(bar, gproc:get_env(l, gproc, some_env, [{mnesia,transaction,
  275. {t, foo}, 3}])),
  276. ?assertEqual("erl", gproc:get_env(l, gproc, progname, [init_arg])).
  277. t_get_set_env() ->
  278. ?assertEqual(ok, application:set_env(gproc, aaaa, a)),
  279. ?assertEqual(a, gproc:get_set_env(l, gproc, aaaa, [app_env])),
  280. ?assertEqual(ok, application:set_env(gproc, aaaa, undefined)),
  281. ?assertEqual(a, gproc:get_env(l, gproc, aaaa, [error])).
  282. t_set_env() ->
  283. ?assertEqual(ok, application:set_env(gproc, aaaa, a)),
  284. ?assertEqual(a, gproc:get_set_env(l, gproc, aaaa, [app_env])),
  285. ?assertEqual(ok, application:set_env(gproc, aaaa, undefined)),
  286. ?assertEqual(b, gproc:set_env(l, gproc, aaaa, b, [app_env])),
  287. ?assertEqual({ok,b}, application:get_env(gproc, aaaa)),
  288. %%
  289. ?assertEqual(true, os:putenv("SSSS", "s0")),
  290. ?assertEqual("s0", gproc:get_env(l, gproc, ssss, [os_env])),
  291. ?assertEqual("s1", gproc:set_env(l, gproc, ssss, "s1", [os_env])),
  292. ?assertEqual("s1", os:getenv("SSSS")),
  293. ?assertEqual(true, os:putenv("SSSS", "s0")),
  294. ?assertEqual([{self(),"s1"}],
  295. gproc:lookup_values({p,l,{gproc_env,gproc,ssss}})).
  296. t_get_env_inherit() ->
  297. P = spawn_link(fun() ->
  298. ?assertEqual(bar, gproc:set_env(l,gproc,foo,bar,[])),
  299. gproc:reg({n,l,get_env_p}),
  300. t_loop()
  301. end),
  302. ?assertEqual({P,undefined}, gproc:await({n,l,get_env_p},1000)),
  303. ?assertEqual(bar, gproc:get_env(l, gproc, foo, [{inherit, P}])),
  304. ?assertEqual(bar, gproc:get_env(l, gproc, foo, [{inherit, {n,l,get_env_p}}])),
  305. ?assertEqual(ok, t_call(P, die)).
  306. t_loop() ->
  307. receive
  308. {From, {give_away, Key}} ->
  309. ?assertEqual(From, gproc:give_away(Key, From)),
  310. From ! {self(), ok},
  311. t_loop();
  312. {From, die} ->
  313. From ! {self(), ok}
  314. end.
  315. t_call(P, Msg) ->
  316. P ! {self(), Msg},
  317. receive
  318. {P, Reply} ->
  319. Reply
  320. end.
  321. spawn_helper() ->
  322. Parent = self(),
  323. P = spawn(fun() ->
  324. ?assert(gproc:reg({n,l,self()}) =:= true),
  325. Ref = erlang:monitor(process, Parent),
  326. Parent ! {ok,self()},
  327. receive
  328. {'DOWN', Ref, _, _, _} ->
  329. ok
  330. end
  331. end),
  332. receive
  333. {ok,P} ->
  334. P
  335. end.
  336. give_gproc_some_time(T) ->
  337. timer:sleep(T),
  338. sys:get_status(gproc).
  339. -endif.