-
Notifications
You must be signed in to change notification settings - Fork 8
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Start a cluster with TLS and verify that ered can communicate with the cluster, and verify the behavior when using an expired client certificate. Some tests require the tool 'faketime' to modify the system time, but tests are skipped if the tool is not available in PATH. See: https://manpages.ubuntu.com/manpages/trusty/man1/faketime.1.html
- Loading branch information
Showing
6 changed files
with
319 additions
and
105 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,84 @@ | ||
| -module(ered_test_utils). | ||
|
|
||
| -include("ered_test_utils.hrl"). | ||
|
|
||
| -export([start_cluster/2, | ||
| check_consistent_cluster/2, | ||
| wait_for_consistent_cluster/2]). | ||
|
|
||
| %% Start a cluster client and wait for cluster_ok. | ||
| start_cluster(Ports, Opts) -> | ||
| [Port1, Port2 | PortsRest] = Ports, | ||
| InitialNodes = [{"127.0.0.1", Port} || Port <- [Port1, Port2]], | ||
|
|
||
| {ok, P} = ered:start_link(InitialNodes, [{info_pid, [self()]}] ++ Opts), | ||
|
|
||
| ConnectedInit = [?MSG(#{msg_type := connected, addr := {"127.0.0.1", Port}}) | ||
| || Port <- [Port1, Port2]], | ||
|
|
||
| #{slot_map := SlotMap} = ?MSG(#{msg_type := slot_map_updated}, 1000), | ||
|
|
||
| IdMap = maps:from_list(lists:flatmap( | ||
| fun([_,_|Nodes]) -> | ||
| [{Port, Id} || [_Addr, Port, Id |_]<- Nodes] | ||
| end, SlotMap)), | ||
|
|
||
| ConnectedRest = [#{msg_type := connected} = ?MSG(#{addr := {"127.0.0.1", Port}}) | ||
| || Port <- PortsRest], | ||
|
|
||
| ClusterIds = [Id || #{cluster_id := Id} <- ConnectedInit ++ ConnectedRest], | ||
| ClusterIds = [maps:get(Port, IdMap) || Port <- Ports], | ||
|
|
||
| ?MSG(#{msg_type := cluster_ok}), | ||
|
|
||
| %% Clear all old data | ||
| [{ok, _} = ered:command_client(Client, [<<"FLUSHDB">>]) || Client <- ered:get_clients(P)], | ||
|
|
||
| no_more_msgs(), | ||
| P. | ||
|
|
||
| %% Check if all nodes have the same single view of the slot map and that | ||
| %% all cluster nodes are included in the slot map. | ||
| check_consistent_cluster(Ports, ClientOpts) -> | ||
| SlotMaps = [fun(Port) -> | ||
| {ok, Pid} = ered_client:start_link("127.0.0.1", Port, ClientOpts), | ||
| {ok, SlotMap} = ered_client:command(Pid, [<<"CLUSTER">>, <<"SLOTS">>]), | ||
| ered_client:stop(Pid), | ||
| SlotMap | ||
| end(P) || P <- Ports], | ||
| Consistent = case lists:usort(SlotMaps) of | ||
| [SlotMap] -> | ||
| Ports =:= [Port || {_Ip, Port} <- ered_lib:slotmap_all_nodes(SlotMap)]; | ||
| _NotAllIdentical -> | ||
| false | ||
| end, | ||
| case Consistent of | ||
| true -> ok; | ||
| false -> {error, SlotMaps} | ||
| end. | ||
|
|
||
| %% Wait until cluster is consistent, i.e all nodes have the same single view | ||
| %% of the slot map and all cluster nodes are included in the slot map. | ||
| wait_for_consistent_cluster(Ports, ClientOpts) -> | ||
| fun Loop(N) -> | ||
| case ered_test_utils:check_consistent_cluster(Ports, ClientOpts) of | ||
| ok -> | ||
| true; | ||
| {error, _} when N > 0 -> | ||
| timer:sleep(500), | ||
| Loop(N-1); | ||
| {error, SlotMaps} -> | ||
| error({timeout_consistent_cluster, SlotMaps}) | ||
| end | ||
| end(20). | ||
|
|
||
| no_more_msgs() -> | ||
| {messages,Msgs} = erlang:process_info(self(), messages), | ||
| case Msgs of | ||
| [] -> | ||
| ok; | ||
| Msgs -> | ||
| error({unexpected,Msgs}) | ||
| end. | ||
|
|
||
|
|
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,20 @@ | ||
| %% Expect to receive a message within timeout. | ||
| -define(MSG(Pattern, Timeout), | ||
| (fun () -> | ||
| receive | ||
| Pattern = M -> M | ||
| after | ||
| Timeout -> error({timeout, ??Pattern, erlang:process_info(self(), messages)}) | ||
| end | ||
| end)()). | ||
|
|
||
| %% Expect to receive a message within a second. | ||
| -define(MSG(Pattern), ?MSG(Pattern, 1000)). | ||
|
|
||
| %% Check message queue for optional messages. | ||
| -define(OPTIONAL_MSG(Pattern), | ||
| receive | ||
| Pattern -> ok | ||
| after | ||
| 0 -> ok | ||
| end). |
Oops, something went wrong.