tests

Author: dumbbell

deps/rabbit/test/rabbit_db_binding_SUITE.erl

@@ -8,6 +8,7 @@
 -module(rabbit_db_binding_SUITE).
 
 -include_lib("rabbit_common/include/rabbit.hrl").
+-include_lib("common_test/include/ct.hrl").
 -include_lib("eunit/include/eunit.hrl").
 
 -export([all/0,
@@ -21,8 +22,8 @@
 
          create/1, create1/1,
          exists/1, exists1/1,
-         delete/1, delete1/1,
-         auto_delete/1, auto_delete1/1,
+         delete_v1/1, delete_v2/1, delete1/2,
+         auto_delete_v1/1, auto_delete_v2/1, auto_delete1/2,
          get_all/1, get_all1/1,
          get_all_by_vhost/1, get_all_by_vhost1/1,
          get_all_for_source/1, get_all_for_source1/1,
@@ -33,7 +34,9 @@
          get_all_for_source_and_destination_reverse1/1,
          fold/1, fold1/1,
          match/1, match1/1,
-         match_routing_key/1, match_routing_key1/1
+         match_routing_key/1, match_routing_key1/1,
+         replaced_delete_queue_transaction/1,
+         replaced_delete_queue_transaction1/1
         ]).
 
 -define(VHOST, <<"/">>).
@@ -47,15 +50,18 @@ all() ->
 groups() ->
     [
      {all_tests, [], all_tests()},
-     {feature_flags, [], [replaced_delete_queue_transaction]}
+     {feature_flags, [],
+      [delete_v1,
+       delete_v2,
+       auto_delete_v1,
+       auto_delete_v2,
+       replaced_delete_queue_transaction]}
     ].
 
 all_tests() ->
     [
      create,
      exists,
-     delete,
-     auto_delete,
      get_all,
      get_all_by_vhost,
      get_all_for_source,
@@ -101,7 +107,16 @@ end_per_group(_Group, Config) ->
       rabbit_ct_broker_helpers:teardown_steps()).
 
 init_per_testcase(Testcase, Config) ->
-    rabbit_ct_helpers:testcase_started(Config, Testcase).
+    rabbit_ct_helpers:testcase_started(Config, Testcase),
+    case lists:member({name, feature_flags}, ?config(tc_group_properties, Config)) of
+        true ->
+            ok = rabbit_ct_broker_helpers:stop_broker(Config, 0),
+            ok = rabbit_ct_broker_helpers:reset_node(Config, 0),
+            ok = rabbit_ct_broker_helpers:start_broker(Config, 0);
+        false ->
+            ok
+    end,
+    Config.
 
 end_per_testcase(Testcase, Config) ->
     rabbit_ct_broker_helpers:rpc(Config, 0, rabbit_db_exchange, clear, []),
@@ -150,10 +165,28 @@ exists1(_Config) ->
     ?assertEqual(true, rabbit_db_binding:exists(Binding)),
     passed.
 
-delete(Config) ->
-    passed = rabbit_ct_broker_helpers:rpc(Config, 0, ?MODULE, delete1, [Config]).
-
-delete1(_Config) ->
+delete_v1(Config) ->
+    passed = rabbit_ct_broker_helpers:rpc(Config, 0, ?MODULE, delete1, [Config, v1]).
+
+delete_v2(Config) ->
+    Ret = rabbit_ct_broker_helpers:enable_feature_flag(
+            Config, replaced_delete_queue_transaction),
+    case Ret of
+        ok ->
+            passed = rabbit_ct_broker_helpers:rpc(Config, 0, ?MODULE, delete1, [Config, v2]);
+        {skip, _} = Skip ->
+            Skip
+    end.
+
+delete1(_Config, Version) ->
+    case Version of
+        v1 ->
+            ?assertNot(
+               rabbit_feature_flags:is_enabled(replaced_delete_queue_transaction));
+        v2 ->
+            ?assert(
+               rabbit_feature_flags:is_enabled(replaced_delete_queue_transaction))
+    end,
     XName1 = rabbit_misc:r(?VHOST, exchange, <<"test-exchange1">>),
     XName2 = rabbit_misc:r(?VHOST, exchange, <<"test-exchange2">>),
     Exchange1 = #exchange{name = XName1, durable = true, auto_delete = false, decorators = {[], []}},
@@ -171,10 +204,28 @@ delete1(_Config) ->
     ?assertEqual(false, rabbit_db_binding:exists(Binding)),
     passed.
 
-auto_delete(Config) ->
-    passed = rabbit_ct_broker_helpers:rpc(Config, 0, ?MODULE, auto_delete1, [Config]).
-
-auto_delete1(_Config) ->
+auto_delete_v1(Config) ->
+    passed = rabbit_ct_broker_helpers:rpc(Config, 0, ?MODULE, auto_delete1, [Config, v1]).
+
+auto_delete_v2(Config) ->
+    Ret = rabbit_ct_broker_helpers:enable_feature_flag(
+            Config, replaced_delete_queue_transaction),
+    case Ret of
+        ok ->
+            passed = rabbit_ct_broker_helpers:rpc(Config, 0, ?MODULE, auto_delete1, [Config, v2]);
+        {skip, _} = Skip ->
+            Skip
+    end.
+
+auto_delete1(_Config, Version) ->
+    case Version of
+        v1 ->
+            ?assertNot(
+               rabbit_feature_flags:is_enabled(replaced_delete_queue_transaction));
+        v2 ->
+            ?assert(
+               rabbit_feature_flags:is_enabled(replaced_delete_queue_transaction))
+    end,
     XName1 = rabbit_misc:r(?VHOST, exchange, <<"test-exchange1">>),
     XName2 = rabbit_misc:r(?VHOST, exchange, <<"test-exchange2">>),
     Exchange1 = #exchange{name = XName1, durable = true, auto_delete = true, decorators = {[], []}},