Add test cases for the '--experimental' flag

Author: the-mikedavis

deps/rabbitmq_cli/test/ctl/enable_feature_flag_test.exs

@@ -10,6 +10,8 @@ defmodule EnableFeatureFlagCommandTest do
 
   @command RabbitMQ.CLI.Ctl.Commands.EnableFeatureFlagCommand
   @feature_flag :ff_from_enable_ff_testsuite
+  @experimental_flag :ff_from_enable_ff_testsuite_experimental
+  @usage_exit_code RabbitMQ.CLI.Core.ExitCodes.exit_usage()
 
   setup_all do
     RabbitMQ.CLI.Core.Distribution.start()
@@ -22,6 +24,11 @@ defmodule EnableFeatureFlagCommandTest do
         desc: ~c"My feature flag",
         provided_by: :EnableFeatureFlagCommandTest,
         stability: :stable
+      },
+      @experimental_flag => %{
+        desc: ~c"An **experimental** feature!",
+        provided_by: :EnableFeatureFlagCommandTest,
+        stability: :experimental
       }
     }
 
@@ -35,7 +42,9 @@ defmodule EnableFeatureFlagCommandTest do
 
     {
       :ok,
-      opts: %{node: get_rabbit_hostname(), experimental: false}, feature_flag: @feature_flag
+      opts: %{node: get_rabbit_hostname(), experimental: false},
+      feature_flag: @feature_flag,
+      experimental_flag: @experimental_flag
     }
   end
 
@@ -63,6 +72,16 @@ defmodule EnableFeatureFlagCommandTest do
     assert match?({:badrpc, _}, @command.run(["na"], opts))
   end
 
+  test "run: enabling an experimental flag requires '--experimental'", context do
+    experimental_flag = Atom.to_string(context[:experimental_flag])
+    assert match?(
+             {:error, @usage_exit_code, _},
+             @command.run([experimental_flag], context[:opts])
+           )
+    opts = Map.put(context[:opts], :experimental, true)
+    assert @command.run([experimental_flag], opts) == :ok
+  end
+
   test "run: enabling the same feature flag twice is idempotent", context do
     enable_feature_flag(context[:feature_flag])
     assert @command.run([Atom.to_string(context[:feature_flag])], context[:opts]) == :ok
@@ -75,6 +94,12 @@ defmodule EnableFeatureFlagCommandTest do
     assert list_feature_flags(:enabled) |> Map.has_key?(context[:feature_flag])
   end
 
+  test "run: enabling all feature flags with '--experimental' returns an error", context do
+    enable_feature_flag(context[:feature_flag])
+    opts = Map.put(context[:opts], :experimental, true)
+    assert match?({:error, @usage_exit_code, _}, @command.run(["all"], opts))
+  end
+
   test "banner", context do
     assert @command.banner([context[:feature_flag]], context[:opts]) =~
              ~r/Enabling feature flag \"#{context[:feature_flag]}\" \.\.\./