|
|
@@ -5,27 +5,28 @@
|
|
|
-module(emqx_conf_schema_tests).
|
|
|
|
|
|
-include_lib("eunit/include/eunit.hrl").
|
|
|
+
|
|
|
+%% erlfmt-ignore
|
|
|
+-define(BASE_CONF,
|
|
|
+ """
|
|
|
+ node {
|
|
|
+ name = \"emqx1@127.0.0.1\"
|
|
|
+ cookie = \"emqxsecretcookie\"
|
|
|
+ data_dir = \"data\"
|
|
|
+ }
|
|
|
+ cluster {
|
|
|
+ name = emqxcl
|
|
|
+ discovery_strategy = static
|
|
|
+ static.seeds = ~p
|
|
|
+ core_nodes = ~p
|
|
|
+ }
|
|
|
+ """).
|
|
|
+
|
|
|
array_nodes_test() ->
|
|
|
ExpectNodes = ['emqx1@127.0.0.1', 'emqx2@127.0.0.1'],
|
|
|
- BaseConf =
|
|
|
- ""
|
|
|
- "\n"
|
|
|
- " node {\n"
|
|
|
- " name = \"emqx1@127.0.0.1\"\n"
|
|
|
- " cookie = \"emqxsecretcookie\"\n"
|
|
|
- " data_dir = \"data\"\n"
|
|
|
- " }\n"
|
|
|
- " cluster {\n"
|
|
|
- " name = emqxcl\n"
|
|
|
- " discovery_strategy = static\n"
|
|
|
- " static.seeds = ~p\n"
|
|
|
- " core_nodes = ~p\n"
|
|
|
- " }\n"
|
|
|
- " "
|
|
|
- "",
|
|
|
lists:foreach(
|
|
|
fun(Nodes) ->
|
|
|
- ConfFile = iolist_to_binary(io_lib:format(BaseConf, [Nodes, Nodes])),
|
|
|
+ ConfFile = to_bin(?BASE_CONF, [Nodes, Nodes]),
|
|
|
{ok, Conf} = hocon:binary(ConfFile, #{format => richmap}),
|
|
|
ConfList = hocon_tconf:generate(emqx_conf_schema, Conf),
|
|
|
ClusterDiscovery = proplists:get_value(
|
|
|
@@ -46,6 +47,75 @@ array_nodes_test() ->
|
|
|
),
|
|
|
ok.
|
|
|
|
|
|
+%% erlfmt-ignore
|
|
|
+-define(BASE_AUTHN_ARRAY,
|
|
|
+ """
|
|
|
+ authentication = [
|
|
|
+ {backend = \"http\"
|
|
|
+ body {password = \"${password}\", username = \"${username}\"}
|
|
|
+ connect_timeout = \"15s\"
|
|
|
+ enable_pipelining = 100
|
|
|
+ headers {\"content-type\" = \"application/json\"}
|
|
|
+ mechanism = \"password_based\"
|
|
|
+ method = \"~p\"
|
|
|
+ pool_size = 8
|
|
|
+ request_timeout = \"5s\"
|
|
|
+ ssl {enable = ~p, verify = \"verify_peer\"}
|
|
|
+ url = \"~ts\"
|
|
|
+ }
|
|
|
+ ]
|
|
|
+ """
|
|
|
+).
|
|
|
+
|
|
|
+-define(ERROR(Reason),
|
|
|
+ {emqx_conf_schema, [
|
|
|
+ #{
|
|
|
+ kind := validation_error,
|
|
|
+ reason := integrity_validation_failure,
|
|
|
+ result := _,
|
|
|
+ validation_name := Reason
|
|
|
+ }
|
|
|
+ ]}
|
|
|
+).
|
|
|
+
|
|
|
+authn_validations_test() ->
|
|
|
+ BaseConf = to_bin(?BASE_CONF, ["emqx1@127.0.0.1", "emqx1@127.0.0.1"]),
|
|
|
+
|
|
|
+ OKHttps = to_bin(?BASE_AUTHN_ARRAY, [post, true, <<"https://127.0.0.1:8080">>]),
|
|
|
+ Conf0 = <<BaseConf/binary, OKHttps/binary>>,
|
|
|
+ {ok, ConfMap0} = hocon:binary(Conf0, #{format => richmap}),
|
|
|
+ ?assert(is_list(hocon_tconf:generate(emqx_conf_schema, ConfMap0))),
|
|
|
+
|
|
|
+ OKHttp = to_bin(?BASE_AUTHN_ARRAY, [post, false, <<"http://127.0.0.1:8080">>]),
|
|
|
+ Conf1 = <<BaseConf/binary, OKHttp/binary>>,
|
|
|
+ {ok, ConfMap1} = hocon:binary(Conf1, #{format => richmap}),
|
|
|
+ ?assert(is_list(hocon_tconf:generate(emqx_conf_schema, ConfMap1))),
|
|
|
+
|
|
|
+ DisableSSLWithHttps = to_bin(?BASE_AUTHN_ARRAY, [post, false, <<"https://127.0.0.1:8080">>]),
|
|
|
+ Conf2 = <<BaseConf/binary, DisableSSLWithHttps/binary>>,
|
|
|
+ {ok, ConfMap2} = hocon:binary(Conf2, #{format => richmap}),
|
|
|
+ ?assertThrow(
|
|
|
+ ?ERROR(check_http_ssl_opts),
|
|
|
+ hocon_tconf:generate(emqx_conf_schema, ConfMap2)
|
|
|
+ ),
|
|
|
+
|
|
|
+ BadHeader = to_bin(?BASE_AUTHN_ARRAY, [get, true, <<"https://127.0.0.1:8080">>]),
|
|
|
+ Conf3 = <<BaseConf/binary, BadHeader/binary>>,
|
|
|
+ {ok, ConfMap3} = hocon:binary(Conf3, #{format => richmap}),
|
|
|
+ ?assertThrow(
|
|
|
+ ?ERROR(check_http_headers),
|
|
|
+ hocon_tconf:generate(emqx_conf_schema, ConfMap3)
|
|
|
+ ),
|
|
|
+
|
|
|
+ BadHeaderWithTuple = binary:replace(BadHeader, [<<"[">>, <<"]">>], <<"">>, [global]),
|
|
|
+ Conf4 = <<BaseConf/binary, BadHeaderWithTuple/binary>>,
|
|
|
+ {ok, ConfMap4} = hocon:binary(Conf4, #{format => richmap}),
|
|
|
+ ?assertThrow(
|
|
|
+ ?ERROR(check_http_headers),
|
|
|
+ hocon_tconf:generate(emqx_conf_schema, ConfMap4)
|
|
|
+ ),
|
|
|
+ ok.
|
|
|
+
|
|
|
doc_gen_test() ->
|
|
|
%% the json file too large to encode.
|
|
|
{
|
|
|
@@ -66,3 +136,6 @@ doc_gen_test() ->
|
|
|
ok
|
|
|
end
|
|
|
}.
|
|
|
+
|
|
|
+to_bin(Format, Args) ->
|
|
|
+ iolist_to_binary(io_lib:format(Format, Args)).
|
zhongwencool