From bc6e1da2fb68fdc8529f32e2feba8e751d4c1405 Mon Sep 17 00:00:00 2001
From: Stefan Strigler <stefan.strigler@emqx.io>
Date: Fri, 29 Sep 2023 15:44:11 +0200
Subject: [PATCH] test: add suite for emqx_connector

---
 .../test/emqx_connector_SUITE.erl             | 136 ++++++++++++++++++
 1 file changed, 136 insertions(+)
 create mode 100644 apps/emqx_connector/test/emqx_connector_SUITE.erl

diff --git a/apps/emqx_connector/test/emqx_connector_SUITE.erl b/apps/emqx_connector/test/emqx_connector_SUITE.erl
new file mode 100644
index 000000000..188e516ff
--- /dev/null
+++ b/apps/emqx_connector/test/emqx_connector_SUITE.erl
@@ -0,0 +1,136 @@
+%%--------------------------------------------------------------------
+%% Copyright (c) 2022-2023 EMQ Technologies Co., Ltd. All Rights Reserved.
+%%
+%% Licensed under the Apache License, Version 2.0 (the "License");
+%% you may not use this file except in compliance with the License.
+%% You may obtain a copy of the License at
+%%
+%%     http://www.apache.org/licenses/LICENSE-2.0
+%%
+%% Unless required by applicable law or agreed to in writing, software
+%% distributed under the License is distributed on an "AS IS" BASIS,
+%% WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+%% See the License for the specific language governing permissions and
+%% limitations under the License.
+%%--------------------------------------------------------------------
+-module(emqx_connector_SUITE).
+
+-compile(nowarn_export_all).
+-compile(export_all).
+
+-include_lib("eunit/include/eunit.hrl").
+-include_lib("common_test/include/ct.hrl").
+
+-define(START_APPS, [emqx, emqx_conf, emqx_connector]).
+-define(CONNECTOR, dummy_connector_impl).
+
+all() ->
+    emqx_common_test_helpers:all(?MODULE).
+
+init_per_suite(Config) ->
+    _ = application:load(emqx_conf),
+    ok = emqx_common_test_helpers:start_apps(?START_APPS),
+    Config.
+
+end_per_suite(_Config) ->
+    emqx_common_test_helpers:stop_apps(?START_APPS).
+
+init_per_testcase(TestCase, Config) ->
+    ?MODULE:TestCase({init, Config}).
+
+end_per_testcase(TestCase, Config) ->
+    ?MODULE:TestCase({'end', Config}).
+
+t_connector_lifecycle({init, Config}) ->
+    meck:new(emqx_connector_ee_schema, [passthrough]),
+    meck:expect(emqx_connector_ee_schema, resource_type, 1, dummy_connector_impl),
+    meck:new(?CONNECTOR, [non_strict]),
+    meck:expect(?CONNECTOR, callback_mode, 0, async_if_possible),
+    meck:expect(?CONNECTOR, on_start, 2, {ok, connector_state}),
+    meck:expect(?CONNECTOR, on_stop, 2, ok),
+    meck:expect(?CONNECTOR, on_get_status, 2, connected),
+    [{mocked_mods, [?CONNECTOR, emqx_connector_ee_schema]} | Config];
+t_connector_lifecycle({'end', Config}) ->
+    MockedMods = ?config(mocked_mods, Config),
+    meck:unload(MockedMods),
+    Config;
+t_connector_lifecycle(_Config) ->
+    ?assertEqual(
+        [],
+        emqx_connector:list()
+    ),
+
+    ?assertMatch(
+        {ok, _},
+        emqx_connector:create(kafka, my_connector, connector_config())
+    ),
+
+    ?assertMatch(
+        {ok, #{name := my_connector, type := kafka}},
+        emqx_connector:lookup(<<"connector:kafka:my_connector">>)
+    ),
+
+    ?assertMatch(
+        {ok, #{name := my_connector, type := kafka}},
+        emqx_connector:lookup(<<"kafka:my_connector">>)
+    ),
+
+    ?assertMatch(
+        {ok, #{name := my_connector, type := kafka}},
+        emqx_connector:lookup(kafka, my_connector)
+    ),
+
+    ?assertMatch(
+        [#{name := <<"my_connector">>, type := <<"kafka">>}],
+        emqx_connector:list()
+    ),
+
+    ?assertMatch(
+        {ok, _},
+        emqx_connector:remove(kafka, my_connector)
+    ),
+
+    ?assertEqual(
+        [],
+        emqx_connector:list()
+    ),
+
+    ?assert(meck:validate(?CONNECTOR)),
+    ?assertMatch(
+        [
+            {_, {?CONNECTOR, callback_mode, []}, _},
+            {_, {?CONNECTOR, on_start, [_, _]}, {ok, connector_state}},
+            {_, {?CONNECTOR, on_get_status, [_, connector_state]}, connected},
+            {_, {?CONNECTOR, on_stop, [_, connector_state]}, ok}
+        ],
+        meck:history(?CONNECTOR)
+    ),
+    ok.
+
+connector_config() ->
+    #{
+        <<"authentication">> => <<"none">>,
+        <<"bootstrap_hosts">> => <<"127.0.0.1:9092">>,
+        <<"connect_timeout">> => <<"5s">>,
+        <<"enable">> => true,
+        <<"metadata_request_timeout">> => <<"5s">>,
+        <<"min_metadata_refresh_interval">> => <<"3s">>,
+        <<"socket_opts">> =>
+            #{
+                <<"recbuf">> => <<"1024KB">>,
+                <<"sndbuf">> => <<"1024KB">>,
+                <<"tcp_keepalive">> => <<"none">>
+            },
+        <<"ssl">> =>
+            #{
+                <<"ciphers">> => [],
+                <<"depth">> => 10,
+                <<"enable">> => false,
+                <<"hibernate_after">> => <<"5s">>,
+                <<"log_level">> => <<"notice">>,
+                <<"reuse_sessions">> => true,
+                <<"secure_renegotiate">> => true,
+                <<"verify">> => <<"verify_peer">>,
+                <<"versions">> => [<<"tlsv1.3">>, <<"tlsv1.2">>]
+            }
+    }.