fix types and tests in peer discovery

(cherry picked from commit 1c7b4e3)
rabbitmq · Dec 13, 2023 · 2f0d6c7 · 2f0d6c7
1 parent cd7f1f9
commit 2f0d6c7
Show file tree

Hide file tree

Showing 4 changed files with 26 additions and 15 deletions.
diff --git a/deps/rabbitmq_peer_discovery_aws/test/unit_SUITE.erl b/deps/rabbitmq_peer_discovery_aws/test/unit_SUITE.erl
@@ -48,27 +48,27 @@ maybe_add_tag_filters(_Config) ->
     ?assertEqual(Expectation, Result).
 
 get_hostname_name_from_reservation_set(_Config) ->
-    {
+    ok = eunit:test({
       foreach,
       fun on_start/0,
       fun on_finish/1,
       [{"from private DNS",
         fun() ->
-                Expectation = ["ip-10-0-16-31.eu-west-1.compute.internal",
-                               "ip-10-0-16-29.eu-west-1.compute.internal"],
+                Expectation = ["ip-10-0-16-29.eu-west-1.compute.internal",
+                               "ip-10-0-16-31.eu-west-1.compute.internal"],
                 ?assertEqual(Expectation,
                              rabbit_peer_discovery_aws:get_hostname_name_from_reservation_set(
                                reservation_set(), []))
         end},
        {"from private IP",
         fun() ->
                 os:putenv("AWS_USE_PRIVATE_IP", "true"),
-                Expectation = ["10.0.16.31", "10.0.16.29"],
+                Expectation = ["10.0.16.29", "10.0.16.31"],
                 ?assertEqual(Expectation,
                              rabbit_peer_discovery_aws:get_hostname_name_from_reservation_set(
                                reservation_set(), []))
         end}]
-    }.
+    }).
 
 registration_support(_Config) ->
     ?assertEqual(false, rabbit_peer_discovery_aws:supports_registration()).

diff --git a/deps/rabbitmq_peer_discovery_common/include/rabbit_peer_discovery.hrl b/deps/rabbitmq_peer_discovery_common/include/rabbit_peer_discovery.hrl
@@ -16,7 +16,14 @@
 % by `httpc`
 -define(DEFAULT_HTTP_TIMEOUT, 2250).
 
--type peer_discovery_config_value() :: atom() | integer() | string() | list() | map() | any() | undefined.
+-type peer_discovery_config_value() :: port()
+                                     | atom()
+                                     | integer()
+                                     | string()
+                                     | proplists:proplist()
+                                     | map()
+                                     | list()
+                                     | undefined.
 
 -record(peer_discovery_config_entry_meta,
         {env_variable  :: string(),

diff --git a/deps/rabbitmq_peer_discovery_common/src/rabbit_peer_discovery_config.erl b/deps/rabbitmq_peer_discovery_common/src/rabbit_peer_discovery_config.erl
@@ -120,14 +120,14 @@ get_integer_from_env_variable_or_map(Map, OSKey, AppKey, Default) ->
 %%--------------------------------------------------------------------
 -spec normalize(Type  :: atom(),
                 Value :: atom() | boolean() | integer() | string() | list()) ->
-  atom() | integer() | string().
+  peer_discovery_config_value().
 %% TODO: switch these to use delegate to rabbit_data_coercion:*
 normalize(Type, Value) when Type =:= port ->
   rabbit_peer_discovery_util:parse_port(Value);
 normalize(Type, Value) when Type =:= atom ->
   rabbit_peer_discovery_util:as_atom(Value);
 normalize(Type, Value) when Type =:= list ->
-  rabbit_data_coercion:to_list(Value);
+  rabbit_peer_discovery_util:as_list(Value);
 normalize(Type, Value) when Type =:= integer ->
   rabbit_peer_discovery_util:as_integer(Value);
 normalize(Type, Value) when Type =:= string ->

diff --git a/deps/rabbitmq_peer_discovery_common/src/rabbit_peer_discovery_util.erl b/deps/rabbitmq_peer_discovery_common/src/rabbit_peer_discovery_util.erl
@@ -410,14 +410,18 @@ as_list([]) -> [];
 as_list(Value) when is_atom(Value) ; is_integer(Value) ; is_binary(Value) ->
   [Value];
 as_list(Value) when is_list(Value) ->
+  Parse = fun(T) -> 
+    S = string:strip(T),
+    case string:to_float(S) of
+      {Float, []} -> Float;
+      _ -> case string:to_integer(S) of
+             {Integer, []} -> Integer;
+             _ -> S
+           end
+    end
+  end,
   case io_lib:printable_list(Value) or io_lib:printable_unicode_list(Value) of
-    true -> [case string:to_float(S) of
-               {Float, []} -> Float;
-               _ -> case string:to_integer(S) of
-                      {Integer, []} -> Integer;
-                      _ -> string:strip(S)
-                    end
-             end || S <- string:tokens(Value, ",")];
+    true -> [Parse(T) || T <- string:tokens(Value, ",")];
     false -> Value
   end;
 as_list(Value) ->