From 81a104690df1e3429048aea9b063737c413e26d1 Mon Sep 17 00:00:00 2001
From: "Zaiming (Stone) Shi" <zmstone@gmail.com>
Date: Thu, 30 Mar 2023 16:19:22 +0200
Subject: [PATCH] test: fix flaky influxdb test

---
 .../test/emqx_ee_bridge_influxdb_SUITE.erl             | 10 ++--------
 .../src/emqx_ee_connector_influxdb.erl                 |  9 +++------
 2 files changed, 5 insertions(+), 14 deletions(-)

diff --git a/lib-ee/emqx_ee_bridge/test/emqx_ee_bridge_influxdb_SUITE.erl b/lib-ee/emqx_ee_bridge/test/emqx_ee_bridge_influxdb_SUITE.erl
index e8dd970f3..1b4b4aeb2 100644
--- a/lib-ee/emqx_ee_bridge/test/emqx_ee_bridge_influxdb_SUITE.erl
+++ b/lib-ee/emqx_ee_bridge/test/emqx_ee_bridge_influxdb_SUITE.erl
@@ -990,9 +990,7 @@ t_write_failure(Config) ->
                     ?assertMatch([_ | _], Trace),
                     [#{result := Result} | _] = Trace,
                     ?assert(
-                        {error, {error, {closed, "The connection was lost."}}} =:= Result orelse
-                            {error, {error, closed}} =:= Result orelse
-                            {error, {recoverable_error, econnrefused}} =:= Result,
+                        not emqx_ee_connector_influxdb:is_unrecoverable_error(Result),
                         #{got => Result}
                     );
                 async ->
@@ -1000,11 +998,7 @@ t_write_failure(Config) ->
                     ?assertMatch([#{action := nack} | _], Trace),
                     [#{result := Result} | _] = Trace,
                     ?assert(
-                        {error, {recoverable_error, {closed, "The connection was lost."}}} =:=
-                            Result orelse
-                            {error, {error, closed}} =:= Result orelse
-                            {error, {recoverable_error, econnrefused}} =:= Result orelse
-                            {error, {recoverable_error, noproc}} =:= Result,
+                        not emqx_ee_connector_influxdb:is_unrecoverable_error(Result),
                         #{got => Result}
                     )
             end,
diff --git a/lib-ee/emqx_ee_connector/src/emqx_ee_connector_influxdb.erl b/lib-ee/emqx_ee_connector/src/emqx_ee_connector_influxdb.erl
index 5c99a23a8..afe17cae6 100644
--- a/lib-ee/emqx_ee_connector/src/emqx_ee_connector_influxdb.erl
+++ b/lib-ee/emqx_ee_connector/src/emqx_ee_connector_influxdb.erl
@@ -35,6 +35,9 @@
     desc/1
 ]).
 
+%% only for test
+-export([is_unrecoverable_error/1]).
+
 -type ts_precision() :: ns | us | ms | s.
 
 %% influxdb servers don't need parse
@@ -655,12 +658,6 @@ str(S) when is_list(S) ->
 
 is_unrecoverable_error({error, {unrecoverable_error, _}}) ->
     true;
-is_unrecoverable_error({error, {recoverable_error, _}}) ->
-    false;
-is_unrecoverable_error({error, {error, econnrefused}}) ->
-    false;
-is_unrecoverable_error({error, econnrefused}) ->
-    false;
 is_unrecoverable_error(_) ->
     false.