%%-------------------------------------------------------------------- %% 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(#{}, #{}) ). t_import_config(_Config) -> %% Import to default license ?assertMatch( {ok, #{root_key := license, changed := _}}, emqx_license:import_config(#{<<"license">> => #{<<"key">> => <<"default">>}}) ), ?assertEqual(default, emqx:get_config([license, key])), ?assertMatch({ok, #{max_connections := 10}}, emqx_license_checker:limits()), %% Import to a new license EncodedLicense = emqx_license_test_lib:make_license(#{max_connections => "100"}), ?assertMatch( {ok, #{root_key := license, changed := _}}, emqx_license:import_config( #{ <<"license">> => #{ <<"key">> => EncodedLicense, <<"connection_low_watermark">> => <<"20%">>, <<"connection_high_watermark">> => <<"50%">> } } ) ), ?assertMatch({ok, #{max_connections := 100}}, emqx_license_checker:limits()), ?assertMatch( #{connection_low_watermark := 0.2, connection_high_watermark := 0.5}, emqx:get_config([license]) ). %%------------------------------------------------------------------------------ %% 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}.