1 files changed, 70 insertions, 0 deletions
diff --git a/deps/rabbitmq_cli/test/ctl/enable_feature_flag_test.exs b/deps/rabbitmq_cli/test/ctl/enable_feature_flag_test.exs
new file mode 100644
index 0000000000..f8a3e62920
--- /dev/null
+++ b/deps/rabbitmq_cli/test/ctl/enable_feature_flag_test.exs
@@ -0,0 +1,70 @@
+## This Source Code Form is subject to the terms of the Mozilla Public
+## License, v. 2.0. If a copy of the MPL was not distributed with this
+## file, You can obtain one at https://mozilla.org/MPL/2.0/.
+##
+## Copyright (c) 2018-2020 VMware, Inc. or its affiliates.  All rights reserved.
+
+
+defmodule EnableFeatureFlagCommandTest do
+  use ExUnit.Case, async: false
+  import TestHelper
+
+  @command RabbitMQ.CLI.Ctl.Commands.EnableFeatureFlagCommand
+  @feature_flag :ff_from_enable_ff_testsuite
+
+  setup_all do
+    RabbitMQ.CLI.Core.Distribution.start()
+
+    # Define an arbitrary feature flag for the test.
+    node = get_rabbit_hostname()
+    new_feature_flags = %{
+      @feature_flag =>
+      %{desc: "My feature flag",
+        provided_by: :EnableFeatureFlagCommandTest,
+        stability: :stable}}
+    :ok = :rabbit_misc.rpc_call(
+      node, :rabbit_feature_flags, :initialize_registry, [new_feature_flags])
+
+    {
+      :ok,
+      opts: %{node: get_rabbit_hostname()},
+      feature_flag: @feature_flag
+    }
+  end
+
+  test "validate: wrong number of arguments results in arg count errors" do
+    assert @command.validate([], %{}) == {:validation_failure, :not_enough_args}
+    assert @command.validate(["ff_from_enable_ff_testsuite", "whoops"], %{}) == {:validation_failure, :too_many_args}
+  end
+
+  test "validate: passing an empty string for feature_flag name is an arg error", context do
+    assert match?({:validation_failure, {:bad_argument, _}}, @command.validate([""], context[:opts]))
+  end
+
+  test "run: passing a valid feature_flag name to a running RabbitMQ node succeeds", context do
+    assert @command.run([Atom.to_string(context[:feature_flag])], context[:opts]) == :ok
+    assert list_feature_flags(:enabled) |> Map.has_key?(context[:feature_flag])
+  end
+
+  test "run: attempt to use an unreachable node returns a nodedown" do
+    opts = %{node: :jake@thedog, timeout: 200}
+    assert match?({:badrpc, _}, @command.run(["na"], opts))
+  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
+    assert list_feature_flags(:enabled) |> Map.has_key?(context[:feature_flag])
+  end
+
+  test "run: enabling all feature flags succeeds", context do
+    enable_feature_flag context[:feature_flag]
+    assert @command.run(["all"], context[:opts]) == :ok
+    assert list_feature_flags(:enabled) |> Map.has_key?(context[:feature_flag])
+  end
+
+  test "banner", context do
+    assert @command.banner([context[:feature_flag]], context[:opts])
+      =~ ~r/Enabling feature flag \"#{context[:feature_flag]}\" \.\.\./
+  end
+end