syn_test_suite_helper.erl 3.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899
  1. %% ==========================================================================================================
  2. %% Syn - A global process registry.
  3. %%
  4. %% Copyright (C) 2015, Roberto Ostinelli <roberto@ostinelli.net>.
  5. %% All rights reserved.
  6. %%
  7. %% The MIT License (MIT)
  8. %%
  9. %% Copyright (c) 2015 Roberto Ostinelli
  10. %%
  11. %% Permission is hereby granted, free of charge, to any person obtaining a copy
  12. %% of this software and associated documentation files (the "Software"), to deal
  13. %% in the Software without restriction, including without limitation the rights
  14. %% to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  15. %% copies of the Software, and to permit persons to whom the Software is
  16. %% furnished to do so, subject to the following conditions:
  17. %%
  18. %% The above copyright notice and this permission notice shall be included in
  19. %% all copies or substantial portions of the Software.
  20. %%
  21. %% THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  22. %% IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  23. %% FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  24. %% AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  25. %% LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  26. %% OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
  27. %% THE SOFTWARE.
  28. %% ==========================================================================================================
  29. -module(syn_test_suite_helper).
  30. %% API
  31. -export([start_slave/1, stop_slave/1]).
  32. -export([connect_node/1, disconnect_node/1]).
  33. -export([clean_after_test/0, clean_after_test/1]).
  34. -export([start_process/0, start_process/1, kill_process/1, process_main/0]).
  35. %% ===================================================================
  36. %% API
  37. %% ===================================================================
  38. start_slave(NodeShortName) ->
  39. EbinFilePath = filename:join([filename:dirname(code:lib_dir(syn, ebin)), "ebin"]),
  40. TestFilePath = filename:join([filename:dirname(code:lib_dir(syn, ebin)), "test"]),
  41. %% start slave
  42. {ok, NodeName} = ct_slave:start(NodeShortName, [
  43. {boot_timeout, 10},
  44. {erl_flags, lists:concat(["-pa ", EbinFilePath, " ", TestFilePath])}
  45. ]),
  46. {ok, NodeName}.
  47. stop_slave(NodeShortName) ->
  48. {ok, _} = ct_slave:stop(NodeShortName).
  49. connect_node(NodeName) ->
  50. net_kernel:connect_node(NodeName).
  51. disconnect_node(NodeName) ->
  52. erlang:disconnect_node(NodeName).
  53. clean_after_test() ->
  54. %% delete table
  55. {atomic, ok} = mnesia:delete_table(syn_processes_table),
  56. %% stop mnesia
  57. mnesia:stop(),
  58. %% delete schema
  59. mnesia:delete_schema([node()]),
  60. %% stop syn
  61. syn:stop().
  62. clean_after_test(undefined) ->
  63. clean_after_test();
  64. clean_after_test(NodeName) ->
  65. %% delete table
  66. {atomic, ok} = mnesia:delete_table(syn_processes_table),
  67. %% stop mnesia
  68. mnesia:stop(),
  69. rpc:call(NodeName, mnesia, stop, []),
  70. %% delete schema
  71. mnesia:delete_schema([node(), NodeName]),
  72. %% stop syn
  73. syn:stop(),
  74. rpc:call(NodeName, syn, stop, []).
  75. start_process() ->
  76. Pid = spawn(?MODULE, process_main, []),
  77. Pid.
  78. start_process(NodeName) ->
  79. %% Pid = rpc:call(NodeName, erlang, spawn, [?MODULE, process_main, []]),
  80. Pid = spawn(NodeName, ?MODULE, process_main, []),
  81. Pid.
  82. kill_process(Pid) ->
  83. exit(Pid, kill).
  84. process_main() ->
  85. receive
  86. shutdown -> ok
  87. end.