From d5847f33a8c3f634e42021cc42ec80ab89b9b9ce Mon Sep 17 00:00:00 2001
From: Thales Macedo Garitezi <thalesmg@gmail.com>
Date: Wed, 2 Aug 2023 10:20:53 -0300
Subject: [PATCH] test: attempt to stabilize flaky tests

---
 .../test/emqx_bridge_pgsql_SUITE.erl                | 13 +++++++++----
 1 file changed, 9 insertions(+), 4 deletions(-)

diff --git a/apps/emqx_bridge_pgsql/test/emqx_bridge_pgsql_SUITE.erl b/apps/emqx_bridge_pgsql/test/emqx_bridge_pgsql_SUITE.erl
index 89b1c58e0..cd79db43d 100644
--- a/apps/emqx_bridge_pgsql/test/emqx_bridge_pgsql_SUITE.erl
+++ b/apps/emqx_bridge_pgsql/test/emqx_bridge_pgsql_SUITE.erl
@@ -10,6 +10,7 @@
 -include_lib("eunit/include/eunit.hrl").
 -include_lib("common_test/include/ct.hrl").
 -include_lib("snabbkaffe/include/snabbkaffe.hrl").
+-include("emqx_resource_errors.hrl").
 
 % SQL definitions
 -define(SQL_BRIDGE,
@@ -690,10 +691,14 @@ t_table_removed(Config) ->
             connect_and_drop_table(Config),
             Val = integer_to_binary(erlang:unique_integer()),
             SentData = #{payload => Val, timestamp => 1668602148000},
-            ?assertMatch(
-                {error, {unrecoverable_error, {error, error, <<"42P01">>, undefined_table, _, _}}},
-                query_resource_sync(Config, {send_message, SentData, []})
-            ),
+            case query_resource_sync(Config, {send_message, SentData, []}) of
+                {error, {unrecoverable_error, {error, error, <<"42P01">>, undefined_table, _, _}}} ->
+                    ok;
+                ?RESOURCE_ERROR_M(not_connected, _) ->
+                    ok;
+                Res ->
+                    ct:fail("unexpected result: ~p", [Res])
+            end,
             ok
         end,
         []