gproc_dist_tests.erl 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613
  1. %% -*- erlang-indent-level: 4; indent-tabs-mode: nil -*-
  2. %% ``The contents of this file are subject to the Erlang Public License,
  3. %% Version 1.1, (the "License"); you may not use this file except in
  4. %% compliance with the License. You should have received a copy of the
  5. %% Erlang Public License along with this software. If not, it can be
  6. %% retrieved via the world wide web at http://www.erlang.org/.
  7. %%
  8. %% Software distributed under the License is distributed on an "AS IS"
  9. %% basis, WITHOUT WARRANTY OF ANY KIND, either express or implied. See
  10. %% the License for the specific language governing rights and limitations
  11. %% under the License.
  12. %%
  13. %% The Initial Developer of the Original Code is Ericsson Utvecklings AB.
  14. %% Portions created by Ericsson are Copyright 1999, Ericsson Utvecklings
  15. %% AB. All Rights Reserved.''
  16. %%
  17. %% @author Ulf Wiger <ulf@wiger.net>
  18. %%
  19. -module(gproc_dist_tests).
  20. -ifdef(TEST).
  21. -include_lib("eunit/include/eunit.hrl").
  22. -export([t_spawn/1, t_spawn_reg/2]).
  23. dist_test_() ->
  24. {timeout, 120,
  25. [{setup,
  26. fun() ->
  27. case run_dist_tests() of
  28. true ->
  29. Ns = start_slaves([dist_test_n1, dist_test_n2]),
  30. ?assertMatch({[ok,ok],[]},
  31. rpc:multicall(Ns, application, set_env,
  32. [gproc, gproc_dist, Ns])),
  33. ?assertMatch({[ok,ok],[]},
  34. rpc:multicall(
  35. Ns, application, start, [gproc])),
  36. Ns;
  37. false ->
  38. skip
  39. end
  40. end,
  41. fun(_Ns) ->
  42. ok
  43. end,
  44. fun(skip) -> [];
  45. (Ns) when is_list(Ns) ->
  46. {inorder,
  47. [
  48. {inorder, [
  49. fun() ->
  50. ?debugVal(t_simple_reg(Ns))
  51. end,
  52. fun() ->
  53. ?debugVal(t_simple_reg_or_locate(Ns))
  54. end,
  55. fun() ->
  56. ?debugVal(t_simple_counter(Ns))
  57. end,
  58. fun() ->
  59. ?debugVal(t_aggr_counter(Ns))
  60. end,
  61. fun() ->
  62. ?debugVal(t_awaited_aggr_counter(Ns))
  63. end,
  64. fun() ->
  65. ?debugVal(t_simple_resource_count(Ns))
  66. end,
  67. fun() ->
  68. ?debugVal(t_awaited_resource_count(Ns))
  69. end,
  70. fun() ->
  71. ?debugVal(t_resource_count_on_zero(Ns))
  72. end,
  73. fun() ->
  74. ?debugVal(t_update_counters(Ns))
  75. end,
  76. fun() ->
  77. ?debugVal(t_shared_counter(Ns))
  78. end,
  79. fun() ->
  80. ?debugVal(t_prop(Ns))
  81. end,
  82. fun() ->
  83. ?debugVal(t_mreg(Ns))
  84. end,
  85. fun() ->
  86. ?debugVal(t_await_reg(Ns))
  87. end,
  88. fun() ->
  89. ?debugVal(t_await_self(Ns))
  90. end,
  91. fun() ->
  92. ?debugVal(t_await_reg_exists(Ns))
  93. end,
  94. fun() ->
  95. ?debugVal(t_give_away(Ns))
  96. end,
  97. fun() ->
  98. ?debugVal(t_sync(Ns))
  99. end,
  100. fun() ->
  101. ?debugVal(t_monitor(Ns))
  102. end,
  103. fun() ->
  104. ?debugVal(t_standby_monitor(Ns))
  105. end,
  106. fun() ->
  107. ?debugVal(t_follow_monitor(Ns))
  108. end,
  109. fun() ->
  110. ?debugVal(t_subscribe(Ns))
  111. end
  112. ]
  113. },
  114. fun() ->
  115. ?debugVal(t_sync_cand_dies(Ns))
  116. end,
  117. {timeout, 90, [fun() ->
  118. ?debugVal(t_fail_node(Ns))
  119. end]}
  120. ]}
  121. end
  122. }]}.
  123. run_dist_tests() ->
  124. case os:getenv("GPROC_DIST") of
  125. "true" -> true;
  126. "false" -> false;
  127. false ->
  128. case code:ensure_loaded(gen_leader) of
  129. {error, nofile} ->
  130. false;
  131. _ ->
  132. true
  133. end
  134. end.
  135. -define(T_NAME, {n, g, {?MODULE, ?LINE, erlang:now()}}).
  136. -define(T_KVL, [{foo, "foo"}, {bar, "bar"}]).
  137. -define(T_COUNTER, {c, g, {?MODULE, ?LINE}}).
  138. -define(T_RESOURCE, {r, g, {?MODULE, ?LINE}}).
  139. -define(T_PROP, {p, g, ?MODULE}).
  140. t_simple_reg([H|_] = Ns) ->
  141. Name = ?T_NAME,
  142. P = t_spawn_reg(H, Name),
  143. ?assertMatch(ok, t_lookup_everywhere(Name, Ns, P)),
  144. ?assertMatch(true, t_call(P, {apply, gproc, unreg, [Name]})),
  145. ?assertMatch(ok, t_lookup_everywhere(Name, Ns, undefined)),
  146. ?assertMatch(ok, t_call(P, die)).
  147. t_simple_reg_or_locate([A,B|_] = _Ns) ->
  148. Name = ?T_NAME,
  149. P1 = t_spawn(A),
  150. Ref = erlang:monitor(process, P1),
  151. ?assertMatch({P1, the_value},
  152. t_call(P1, {apply, gproc, reg_or_locate, [Name, the_value]})),
  153. P2 = t_spawn(B),
  154. Ref2 = erlang:monitor(process, P2),
  155. ?assertMatch({P1, the_value},
  156. t_call(P2, {apply, gproc, reg_or_locate, [Name, other_value]})),
  157. ?assertMatch(ok, t_call(P1, die)),
  158. ?assertMatch(ok, t_call(P2, die)),
  159. flush_down(Ref),
  160. flush_down(Ref2).
  161. flush_down(Ref) ->
  162. receive
  163. {'DOWN', Ref, _, _, _} ->
  164. ok
  165. after 1000 ->
  166. erlang:error({timeout, [flush_down, Ref]})
  167. end.
  168. t_simple_counter([H|_] = Ns) ->
  169. Ctr = ?T_COUNTER,
  170. P = t_spawn_reg(H, Ctr, 3),
  171. ?assertMatch(ok, t_read_everywhere(Ctr, P, Ns, 3)),
  172. ?assertMatch(5, t_call(P, {apply, gproc, update_counter, [Ctr, 2]})),
  173. ?assertMatch(ok, t_read_everywhere(Ctr, P, Ns, 5)),
  174. ?assertMatch(ok, t_call(P, die)).
  175. t_shared_counter([H|_] = Ns) ->
  176. Ctr = ?T_COUNTER,
  177. P = t_spawn_reg_shared(H, Ctr, 3),
  178. ?assertMatch(ok, t_read_everywhere(Ctr, shared, Ns, 3)),
  179. ?assertMatch(5, t_call(P, {apply, gproc, update_shared_counter, [Ctr, 2]})),
  180. ?assertMatch(ok, t_read_everywhere(Ctr, shared, Ns, 5)),
  181. ?assertMatch(ok, t_call(P, die)),
  182. ?assertMatch(ok, t_read_everywhere(Ctr, shared, Ns, 5)),
  183. ?assertMatch(ok, t_read_everywhere(Ctr, shared, Ns, 5)), % twice
  184. P1 = t_spawn(H),
  185. ?assertMatch(true, t_call(P1, {apply, gproc, unreg_shared, [Ctr]})),
  186. ?assertMatch(ok, t_read_everywhere(Ctr, shared, Ns, badarg)).
  187. t_aggr_counter([H1,H2|_] = Ns) ->
  188. {c,g,Nm} = Ctr = ?T_COUNTER,
  189. Aggr = {a,g,Nm},
  190. Pc1 = t_spawn_reg(H1, Ctr, 3),
  191. Pa = t_spawn_reg(H2, Aggr),
  192. ?assertMatch(ok, t_read_everywhere(Ctr, Pc1, Ns, 3)),
  193. ?assertMatch(ok, t_read_everywhere(Aggr, Pa, Ns, 3)),
  194. Pc2 = t_spawn_reg(H2, Ctr, 3),
  195. ?assertMatch(ok, t_read_everywhere(Ctr, Pc2, Ns, 3)),
  196. ?assertMatch(ok, t_read_everywhere(Aggr, Pa, Ns, 6)),
  197. ?assertMatch(5, t_call(Pc1, {apply, gproc, update_counter, [Ctr, 2]})),
  198. ?assertMatch(ok, t_read_everywhere(Ctr, Pc1, Ns, 5)),
  199. ?assertMatch(ok, t_read_everywhere(Aggr, Pa, Ns, 8)),
  200. ?assertMatch(ok, t_call(Pc1, die)),
  201. ?assertMatch(ok, t_read_everywhere(Aggr, Pa, Ns, 3)),
  202. ?assertMatch(ok, t_call(Pc2, die)),
  203. ?assertMatch(ok, t_call(Pa, die)).
  204. t_awaited_aggr_counter([H1,H2|_] = Ns) ->
  205. {c,g,Nm} = Ctr = ?T_COUNTER,
  206. Aggr = {a,g,Nm},
  207. Pc1 = t_spawn_reg(H1, Ctr, 3),
  208. P = t_spawn(H2),
  209. Ref = erlang:monitor(process, P),
  210. P ! {self(), Ref, {apply, gproc, await, [Aggr]}},
  211. t_sleep(),
  212. P1 = t_spawn_reg(H2, Aggr),
  213. ?assert(P1 == receive
  214. {P, Ref, Res} ->
  215. element(1, Res);
  216. {'DOWN', Ref, _, _, Reason} ->
  217. erlang:error(Reason);
  218. Other ->
  219. erlang:error({received, Other})
  220. end),
  221. ?assertMatch(ok, t_read_everywhere(Aggr, P1, Ns, 3)),
  222. ?assertMatch(ok, t_call(Pc1, die)),
  223. ?assertMatch(ok, t_call(P, die)),
  224. flush_down(Ref),
  225. ?assertMatch(ok, t_call(P1, die)).
  226. t_simple_resource_count([H1,H2|_] = Ns) ->
  227. {r,g,Nm} = R = ?T_RESOURCE,
  228. RC = {rc,g,Nm},
  229. Pr1 = t_spawn_reg(H1, R, 3),
  230. Prc = t_spawn_reg(H2, RC),
  231. ?assertMatch(ok, t_read_everywhere(R, Pr1, Ns, 3)),
  232. ?assertMatch(ok, t_read_everywhere(RC, Prc, Ns, 1)),
  233. Pr2 = t_spawn_reg(H2, R, 4),
  234. ?assertMatch(ok, t_read_everywhere(R, Pr2, Ns, 4)),
  235. ?assertMatch(ok, t_read_everywhere(RC, Prc, Ns, 2)),
  236. ?assertMatch(ok, t_call(Pr1, die)),
  237. ?assertMatch(ok, t_read_everywhere(RC, Prc, Ns, 1)),
  238. ?assertMatch(ok, t_call(Pr2, die)),
  239. ?assertMatch(ok, t_call(Prc, die)).
  240. t_awaited_resource_count([H1,H2|_] = Ns) ->
  241. {r,g,Nm} = R = ?T_RESOURCE,
  242. RC = {rc,g,Nm},
  243. Pr1 = t_spawn_reg(H1, R, 3),
  244. P = t_spawn(H2),
  245. Ref = erlang:monitor(process, P),
  246. P ! {self(), Ref, {apply, gproc, await, [RC]}},
  247. t_sleep(),
  248. P1 = t_spawn_reg(H2, RC),
  249. ?assert(P1 == receive
  250. {P, Ref, Res} ->
  251. element(1, Res);
  252. {'DOWN', Ref, _, _, Reason} ->
  253. erlang:error(Reason);
  254. Other ->
  255. erlang:error({received, Other})
  256. end),
  257. ?assertMatch(ok, t_read_everywhere(RC, P1, Ns, 1)),
  258. ?assertMatch(ok, t_call(Pr1, die)),
  259. ?assertMatch(ok, t_call(P, die)),
  260. flush_down(Ref),
  261. ?assertMatch(ok, t_call(P1, die)).
  262. t_resource_count_on_zero([H1,H2|_] = Ns) ->
  263. {r,g,Nm} = R = ?T_RESOURCE,
  264. Prop = ?T_PROP,
  265. RC = {rc,g,Nm},
  266. Pr1 = t_spawn_reg(H1, R, 3),
  267. Pp = t_spawn_reg(H2, Prop),
  268. ?assertMatch(ok, t_call(Pp, {selective, true})),
  269. Prc = t_spawn_reg(H2, RC, undefined, [{on_zero, [{send, Prop}]}]),
  270. ?assertMatch(ok, t_read_everywhere(R, Pr1, Ns, 3)),
  271. ?assertMatch(ok, t_read_everywhere(RC, Prc, Ns, 1)),
  272. ?assertMatch(ok, t_call(Pr1, die)),
  273. ?assertMatch(ok, t_read_everywhere(RC, Prc, Ns, 0)),
  274. ?assertMatch({gproc, resource_on_zero, g, Nm, Prc},
  275. t_call(Pp, {apply_fun, fun() ->
  276. receive
  277. {gproc, _, _, _, _} = M ->
  278. M
  279. after 10000 ->
  280. timeout
  281. end
  282. end})),
  283. ?assertMatch(ok, t_call(Pp, {selective, false})),
  284. ?assertMatch(ok, t_call(Pp, die)),
  285. ?assertMatch(ok, t_call(Prc, die)).
  286. t_update_counters([H1,H2|_] = Ns) ->
  287. {c,g,N1} = C1 = ?T_COUNTER,
  288. A1 = {a,g,N1},
  289. C2 = ?T_COUNTER,
  290. P1 = t_spawn_reg(H1, C1, 2),
  291. P12 = t_spawn_reg(H2, C1, 2),
  292. P2 = t_spawn_reg(H2, C2, 1),
  293. Pa1 = t_spawn_reg(H2, A1),
  294. ?assertMatch(ok, t_read_everywhere(C1, P1, Ns, 2)),
  295. ?assertMatch(ok, t_read_everywhere(C1, P12, Ns, 2)),
  296. ?assertMatch(ok, t_read_everywhere(C2, P2, Ns, 1)),
  297. ?assertMatch(ok, t_read_everywhere(A1, Pa1, Ns, 4)),
  298. ?assertMatch([{C1,P1, 3},
  299. {C1,P12,4},
  300. {C2,P2, 0}], t_call(P1, {apply, gproc, update_counters,
  301. [g, [{C1,P1,1},{C1,P12,2},{C2,P2,{-2,0,0}}]]})),
  302. ?assertMatch(ok, t_read_everywhere(C1, P1, Ns, 3)),
  303. ?assertMatch(ok, t_read_everywhere(C1, P12, Ns, 4)),
  304. ?assertMatch(ok, t_read_everywhere(C2, P2, Ns, 0)),
  305. ?assertMatch(ok, t_read_everywhere(A1, Pa1, Ns, 7)),
  306. ?assertMatch(ok, t_call(P1, die)),
  307. ?assertMatch(ok, t_call(P12, die)),
  308. ?assertMatch(ok, t_call(P2, die)).
  309. t_prop([H1,H2|_] = Ns) ->
  310. {p, g, _} = P = ?T_PROP,
  311. P1 = t_spawn_reg(H1, P, 1),
  312. P2 = t_spawn_reg(H2, P, 2),
  313. ?assertMatch(ok, t_read_everywhere(P, P1, Ns, 1)),
  314. ?assertMatch(ok, t_read_everywhere(P, P2, Ns, 2)),
  315. ?assertMatch(ok, t_call(P1, die)),
  316. ?assertMatch(ok, t_read_everywhere(P, P1, Ns, badarg)),
  317. ?assertMatch(ok, t_call(P2, die)).
  318. t_mreg([H|_] = Ns) ->
  319. Kvl = ?T_KVL,
  320. Keys = [K || {K,_} <- Kvl],
  321. P = t_spawn_mreg(H, Kvl),
  322. [?assertMatch(ok, t_lookup_everywhere({n,g,K}, Ns, P)) || K <- Keys],
  323. ?assertMatch(true, t_call(P, {apply, gproc, munreg, [n, g, Keys]})),
  324. [?assertMatch(ok, t_lookup_everywhere({n,g,K},Ns,undefined)) || K <- Keys],
  325. ?assertMatch(ok, t_call(P, die)).
  326. t_await_reg([A,B|_] = Ns) ->
  327. Name = ?T_NAME,
  328. P = t_spawn(A),
  329. Ref = erlang:monitor(process, P),
  330. P ! {self(), Ref, {apply, gproc, await, [Name]}},
  331. t_sleep(),
  332. P1 = t_spawn_reg(B, Name),
  333. ?assert(P1 == receive
  334. {P, Ref, Res} ->
  335. element(1, Res);
  336. {'DOWN', Ref, _, _, Reason} ->
  337. erlang:error(Reason);
  338. Other ->
  339. erlang:error({received,Other})
  340. end),
  341. ?assertMatch(ok, t_call(P, die)),
  342. flush_down(Ref),
  343. ?assertMatch(ok, t_lookup_everywhere(Name, Ns, P1)),
  344. ?assertMatch(ok, t_call(P1, die)).
  345. t_await_self([A|_]) ->
  346. Name = ?T_NAME,
  347. P = t_spawn(A, false), % don't buffer unknowns
  348. Ref = t_call(P, {apply, gproc, nb_wait, [Name]}),
  349. ?assertMatch(ok, t_call(P, {selective, true})),
  350. ?assertMatch(true, t_call(P, {apply, gproc, reg, [Name, some_value]})),
  351. ?assertMatch({registered, {Name, P, some_value}},
  352. t_call(P, {apply_fun, fun() ->
  353. receive
  354. {gproc, Ref, R, Wh} ->
  355. {R, Wh}
  356. after 10000 ->
  357. timeout
  358. end
  359. end})),
  360. ?assertMatch(ok, t_call(P, {selective, false})),
  361. ?assertMatch(true, t_call(P, {apply, gproc, unreg, [Name]})).
  362. t_await_reg_exists([A,B|_]) ->
  363. Name = ?T_NAME,
  364. P = t_spawn(A),
  365. Ref = erlang:monitor(process, P),
  366. P1 = t_spawn_reg(B, Name),
  367. P ! {self(), Ref, {apply, gproc, await, [Name]}},
  368. ?assert(P1 == receive
  369. {P, Ref, Res} ->
  370. element(1, Res);
  371. {'DOWN', Ref, _, _, Reason} ->
  372. erlang:error(Reason);
  373. Other ->
  374. erlang:error({received,Other})
  375. end),
  376. ?assertMatch(ok, t_call(P, die)),
  377. ?assertMatch(ok, t_call(P1, die)).
  378. t_give_away([A,B|_] = Ns) ->
  379. Na = ?T_NAME,
  380. Nb = ?T_NAME,
  381. Pa = t_spawn_reg(A, Na),
  382. Pb = t_spawn_reg(B, Nb),
  383. ?assertMatch(ok, t_lookup_everywhere(Na, Ns, Pa)),
  384. ?assertMatch(ok, t_lookup_everywhere(Nb, Ns, Pb)),
  385. ?assertMatch(Pb, t_call(Pa, {apply, gproc, give_away, [Na, Nb]})),
  386. ?assertMatch(ok, t_lookup_everywhere(Na, Ns, Pb)),
  387. ?assertMatch(Pa, t_call(Pb, {apply, gproc, give_away, [Na, Pa]})),
  388. ?assertMatch(ok, t_lookup_everywhere(Na, Ns, Pa)),
  389. ?assertMatch(ok, t_call(Pa, die)),
  390. ?assertMatch(ok, t_call(Pb, die)).
  391. t_sync(Ns) ->
  392. %% Don't really know how to test this...
  393. [?assertMatch(true, rpc:call(N, gproc_dist, sync, []))
  394. || N <- Ns].
  395. t_monitor([A,B|_]) ->
  396. Na = ?T_NAME,
  397. Pa = t_spawn_reg(A, Na),
  398. Pb = t_spawn(B, _Selective = true),
  399. Ref = t_call(Pb, {apply, gproc, monitor, [Na]}),
  400. ?assert(is_reference(Ref)),
  401. ?assertMatch(ok, t_call(Pa, die)),
  402. ?assertMatch({gproc,unreg,Ref,Na}, got_msg(Pb, gproc)),
  403. Pc = t_spawn_reg(A, Na),
  404. Ref1 = t_call(Pb, {apply, gproc, monitor, [Na]}),
  405. ?assertMatch(true, t_call(Pc, {apply, gproc, unreg, [Na]})),
  406. ?assertMatch({gproc,unreg,Ref1,Na}, got_msg(Pb, gproc)).
  407. t_standby_monitor([A,B|_] = Ns) ->
  408. Na = ?T_NAME,
  409. Pa = t_spawn_reg(A, Na),
  410. Pb = t_spawn(B, _Selective = true),
  411. Ref = t_call(Pb, {apply, gproc, monitor, [Na, standby]}),
  412. ?assert(is_reference(Ref)),
  413. ?assertMatch(ok, t_call(Pa, die)),
  414. ?assertMatch({gproc,{failover,Pb},Ref,Na}, got_msg(Pb, gproc)),
  415. ?assertMatch(ok, t_lookup_everywhere(Na, Ns, Pb)),
  416. Pc = t_spawn(A, true),
  417. Ref1 = t_call(Pc, {apply, gproc, monitor, [Na, standby]}),
  418. ?assertMatch(true, t_call(Pb, {apply, gproc, unreg, [Na]})),
  419. ?assertMatch({gproc,unreg,Ref1,Na}, got_msg(Pc, gproc)),
  420. ?assertMatch(ok, t_lookup_everywhere(Na, Ns, undefined)).
  421. t_follow_monitor([A,B|_]) ->
  422. Na = ?T_NAME,
  423. Pa = t_spawn(A, _Selective = true),
  424. Ref = t_call(Pa, {apply, gproc, monitor, [Na, follow]}),
  425. {gproc,unreg,Ref,Na} = got_msg(Pa),
  426. Pb = t_spawn_reg(B, Na),
  427. {gproc,registered,Ref,Na} = got_msg(Pa),
  428. ok = t_call(Pb, die),
  429. ok = t_call(Pa, die).
  430. t_subscribe([A,B|_] = Ns) ->
  431. Na = ?T_NAME,
  432. Pb = t_spawn(B, _Selective = true),
  433. ?assertEqual(ok, t_call(Pb, {apply, gproc_monitor, subscribe, [Na]})),
  434. ?assertMatch({gproc_monitor, Na, undefined}, got_msg(Pb, gproc_monitor)),
  435. Pa = t_spawn_reg(A, Na),
  436. ?assertMatch({gproc_monitor, Na, Pa}, got_msg(Pb, gproc_monitor)),
  437. Pc = t_spawn(A),
  438. t_call(Pa, {apply, gproc, give_away, [Na, Pc]}),
  439. ?assertMatch(ok, t_lookup_everywhere(Na, Ns, Pc)),
  440. ?assertEqual({gproc_monitor,Na,{migrated,Pc}}, got_msg(Pb, gproc_monitor)),
  441. ?assertEqual(ok, t_call(Pc, die)),
  442. ?assertEqual({gproc_monitor,Na,undefined}, got_msg(Pb, gproc_monitor)),
  443. ok.
  444. %% got_msg(Pb, Tag) ->
  445. %% t_call(Pb,
  446. %% {apply_fun,
  447. %% fun() ->
  448. %% receive
  449. %% M when element(1, M) == Tag ->
  450. %% M
  451. %% after 1000 ->
  452. %% erlang:error({timeout, got_msg, [Pb, Tag]})
  453. %% end
  454. %% end}).
  455. %% Verify that the gproc_dist:sync() call returns true even if a candidate dies
  456. %% while the sync is underway. This test makes use of sys:suspend() to ensure that
  457. %% the other candidate doesn't respond too quickly.
  458. t_sync_cand_dies([A,B|_]) ->
  459. Leader = rpc:call(A, gproc_dist, get_leader, []),
  460. Other = case Leader of
  461. A -> B;
  462. B -> A
  463. end,
  464. ?assertMatch(ok, rpc:call(Other, sys, suspend, [gproc_dist])),
  465. P = rpc:call(Other, erlang, whereis, [gproc_dist]),
  466. Key = rpc:async_call(Leader, gproc_dist, sync, []),
  467. %% The overall timeout for gproc_dist:sync() is 5 seconds. Here, we should
  468. %% still be waiting.
  469. ?assertMatch(timeout, rpc:nb_yield(Key, 1000)),
  470. exit(P, kill),
  471. %% The leader should detect that the other candidate died and respond
  472. %% immediately. Therefore, we should have our answer well within 1 sec.
  473. ?assertMatch({value, true}, rpc:nb_yield(Key, 1000)).
  474. t_fail_node([A,B|_] = Ns) ->
  475. Na = ?T_NAME,
  476. Nb = ?T_NAME,
  477. Pa = t_spawn_reg(A, Na),
  478. Pb = t_spawn_reg(B, Nb),
  479. ?assertMatch(ok, rpc:call(A, application, stop, [gproc])),
  480. ?assertMatch(ok, t_lookup_everywhere(Na, Ns -- [A], undefined)),
  481. ?assertMatch(ok, t_lookup_everywhere(Nb, Ns -- [A], Pb)),
  482. ?assertMatch(ok, rpc:call(A, application, start, [gproc])),
  483. ?assertMatch(ok, t_lookup_everywhere(Na, Ns, undefined)),
  484. ?assertMatch(ok, t_lookup_everywhere(Nb, Ns, Pb)),
  485. ?assertMatch(ok, t_call(Pa, die)),
  486. ?assertMatch(ok, t_call(Pb, die)).
  487. t_sleep() ->
  488. timer:sleep(500).
  489. t_lookup_everywhere(Key, Nodes, Exp) ->
  490. t_lookup_everywhere(Key, Nodes, Exp, 3).
  491. t_lookup_everywhere(Key, _, Exp, 0) ->
  492. {lookup_failed, Key, Exp};
  493. t_lookup_everywhere(Key, Nodes, Exp, I) ->
  494. Expected = [{N, Exp} || N <- Nodes],
  495. Found = [{N,rpc:call(N, gproc, where, [Key])} || N <- Nodes],
  496. if Expected =/= Found ->
  497. ?debugFmt("lookup ~p failed~n"
  498. "(Expected: ~p;~n"
  499. " Found : ~p), retrying...~n",
  500. [Key, Expected, Found]),
  501. t_sleep(),
  502. t_lookup_everywhere(Key, Nodes, Exp, I-1);
  503. true ->
  504. ok
  505. end.
  506. t_read_everywhere(Key, Pid, Nodes, Exp) ->
  507. t_read_everywhere(Key, Pid, Nodes, Exp, 3).
  508. t_read_everywhere(Key, _, _, Exp, 0) ->
  509. {read_failed, Key, Exp};
  510. t_read_everywhere(Key, Pid, Nodes, Exp, I) ->
  511. Expected = [{N, Exp} || N <- Nodes],
  512. Found = [{N, read_result(rpc:call(N, gproc, get_value, [Key, Pid]))}
  513. || N <- Nodes],
  514. if Expected =/= Found ->
  515. ?debugFmt("read ~p failed~n"
  516. "(Expected: ~p;~n"
  517. " Found : ~p), retrying...~n",
  518. [{Key, Pid}, Expected, Found]),
  519. t_sleep(),
  520. t_read_everywhere(Key, Pid, Nodes, Exp, I-1);
  521. true ->
  522. ok
  523. end.
  524. read_result({badrpc, {'EXIT', {badarg, _}}}) -> badarg;
  525. read_result(R) -> R.
  526. t_spawn(Node) -> gproc_test_lib:t_spawn(Node).
  527. t_spawn(Node, Selective) -> gproc_test_lib:t_spawn(Node, Selective).
  528. t_spawn_mreg(Node, KVL) -> gproc_test_lib:t_spawn_mreg(Node, KVL).
  529. t_spawn_reg(Node, N) -> gproc_test_lib:t_spawn_reg(Node, N).
  530. t_spawn_reg(Node, N, V) -> gproc_test_lib:t_spawn_reg(Node, N, V).
  531. t_spawn_reg(Node, N, V, As) -> gproc_test_lib:t_spawn_reg(Node, N, V, As).
  532. t_spawn_reg_shared(Node, N, V) -> gproc_test_lib:t_spawn_reg_shared(Node, N, V).
  533. got_msg(P) -> gproc_test_lib:got_msg(P).
  534. got_msg(P, Tag) -> gproc_test_lib:got_msg(P, Tag).
  535. t_call(P, Req) ->
  536. gproc_test_lib:t_call(P, Req).
  537. start_slaves(Ns) ->
  538. [H|T] = Nodes = [start_slave(N) || N <- Ns],
  539. _ = [rpc:call(H, net_adm, ping, [N]) || N <- T],
  540. Nodes.
  541. start_slave(Name) ->
  542. case node() of
  543. nonode@nohost ->
  544. os:cmd("epmd -daemon"),
  545. {ok, _} = net_kernel:start([gproc_master, shortnames]);
  546. _ ->
  547. ok
  548. end,
  549. {Pa, Pz} = paths(),
  550. Paths = "-pa ./ -pz ../ebin" ++
  551. lists:flatten([[" -pa " ++ Path || Path <- Pa],
  552. [" -pz " ++ Path || Path <- Pz]]),
  553. {ok, Node} = slave:start(host(), Name, Paths),
  554. Node.
  555. paths() ->
  556. Path = code:get_path(),
  557. {ok, [[Root]]} = init:get_argument(root),
  558. {Pas, Rest} = lists:splitwith(fun(P) ->
  559. not lists:prefix(Root, P)
  560. end, Path),
  561. {_, Pzs} = lists:splitwith(fun(P) ->
  562. lists:prefix(Root, P)
  563. end, Rest),
  564. {Pas, Pzs}.
  565. host() ->
  566. [_Name, Host] = re:split(atom_to_list(node()), "@", [{return, list}]),
  567. list_to_atom(Host).
  568. -endif.