163 lines
3.8 KiB
Erlang
163 lines
3.8 KiB
Erlang
%%--------------------------------------------------------------------
|
|
%% Copyright (c) 2022-2024 EMQ Technologies Co., Ltd. All Rights Reserved.
|
|
%%--------------------------------------------------------------------
|
|
|
|
-module(emqx_license_SUITE).
|
|
|
|
-compile(nowarn_export_all).
|
|
-compile(export_all).
|
|
|
|
-include_lib("emqx/include/emqx_mqtt.hrl").
|
|
|
|
-include_lib("eunit/include/eunit.hrl").
|
|
-include_lib("common_test/include/ct.hrl").
|
|
|
|
all() ->
|
|
emqx_common_test_helpers:all(?MODULE).
|
|
|
|
init_per_suite(Config) ->
|
|
emqx_license_test_lib:mock_parser(),
|
|
Apps = emqx_cth_suite:start(
|
|
[
|
|
emqx,
|
|
emqx_conf,
|
|
{emqx_license, "license { key = \"default\" }"}
|
|
],
|
|
#{work_dir => emqx_cth_suite:work_dir(Config)}
|
|
),
|
|
[{suite_apps, Apps} | Config].
|
|
|
|
end_per_suite(Config) ->
|
|
emqx_license_test_lib:unmock_parser(),
|
|
ok = emqx_cth_suite:stop(?config(suite_apps, Config)).
|
|
|
|
init_per_testcase(Case, Config) ->
|
|
setup_test(Case, Config) ++ Config.
|
|
|
|
end_per_testcase(Case, Config) ->
|
|
teardown_test(Case, Config).
|
|
|
|
setup_test(_TestCase, _Config) ->
|
|
[].
|
|
|
|
teardown_test(_TestCase, _Config) ->
|
|
ok.
|
|
|
|
%%------------------------------------------------------------------------------
|
|
%% Tests
|
|
%%------------------------------------------------------------------------------
|
|
|
|
t_update_value(_Config) ->
|
|
?assertMatch(
|
|
{error, #{parse_results := [_ | _]}},
|
|
emqx_license:update_key("invalid.license")
|
|
),
|
|
|
|
LicenseValue = emqx_license_test_lib:default_test_license(),
|
|
|
|
?assertMatch(
|
|
{ok, #{}},
|
|
emqx_license:update_key(LicenseValue)
|
|
).
|
|
|
|
t_check_exceeded(_Config) ->
|
|
{_, License} = mk_license(
|
|
[
|
|
"220111",
|
|
"0",
|
|
"10",
|
|
"Foo",
|
|
"contact@foo.com",
|
|
"bar",
|
|
"20220111",
|
|
"100000",
|
|
"10"
|
|
]
|
|
),
|
|
#{} = emqx_license_checker:update(License),
|
|
|
|
ok = lists:foreach(
|
|
fun(_) ->
|
|
{ok, C} = emqtt:start_link(),
|
|
{ok, _} = emqtt:connect(C)
|
|
end,
|
|
lists:seq(1, 12)
|
|
),
|
|
|
|
?assertEqual(
|
|
{stop, {error, ?RC_QUOTA_EXCEEDED}},
|
|
emqx_license:check(#{}, #{})
|
|
).
|
|
|
|
t_check_ok(_Config) ->
|
|
{_, License} = mk_license(
|
|
[
|
|
"220111",
|
|
"0",
|
|
"10",
|
|
"Foo",
|
|
"contact@foo.com",
|
|
"bar",
|
|
"20220111",
|
|
"100000",
|
|
"10"
|
|
]
|
|
),
|
|
#{} = emqx_license_checker:update(License),
|
|
|
|
ok = lists:foreach(
|
|
fun(_) ->
|
|
{ok, C} = emqtt:start_link(),
|
|
{ok, _} = emqtt:connect(C)
|
|
end,
|
|
lists:seq(1, 11)
|
|
),
|
|
|
|
?assertEqual(
|
|
{ok, #{}},
|
|
emqx_license:check(#{}, #{})
|
|
).
|
|
|
|
t_check_expired(_Config) ->
|
|
{_, License} = mk_license(
|
|
[
|
|
"220111",
|
|
%% Official customer
|
|
"1",
|
|
%% Small customer
|
|
"0",
|
|
"Foo",
|
|
"contact@foo.com",
|
|
"bar",
|
|
%% Expired long ago
|
|
"20211101",
|
|
"10",
|
|
"10"
|
|
]
|
|
),
|
|
#{} = emqx_license_checker:update(License),
|
|
|
|
?assertEqual(
|
|
{stop, {error, ?RC_QUOTA_EXCEEDED}},
|
|
emqx_license:check(#{}, #{})
|
|
).
|
|
|
|
t_check_not_loaded(_Config) ->
|
|
ok = emqx_license_checker:purge(),
|
|
?assertEqual(
|
|
{stop, {error, ?RC_QUOTA_EXCEEDED}},
|
|
emqx_license:check(#{}, #{})
|
|
).
|
|
|
|
%%------------------------------------------------------------------------------
|
|
%% Helpers
|
|
%%------------------------------------------------------------------------------
|
|
|
|
mk_license(Fields) ->
|
|
EncodedLicense = emqx_license_test_lib:make_license(Fields),
|
|
{ok, License} = emqx_license_parser:parse(
|
|
EncodedLicense,
|
|
emqx_license_test_lib:public_key_pem()
|
|
),
|
|
{EncodedLicense, License}.
|