diff options
Diffstat (limited to 'spec/lib/gitlab/with_feature_category_spec.rb')
-rw-r--r-- | spec/lib/gitlab/with_feature_category_spec.rb | 69 |
1 files changed, 69 insertions, 0 deletions
diff --git a/spec/lib/gitlab/with_feature_category_spec.rb b/spec/lib/gitlab/with_feature_category_spec.rb new file mode 100644 index 00000000000..b6fe1c84b26 --- /dev/null +++ b/spec/lib/gitlab/with_feature_category_spec.rb @@ -0,0 +1,69 @@ +# frozen_string_literal: true + +require 'fast_spec_helper' +require_relative "../../../lib/gitlab/with_feature_category" + +RSpec.describe Gitlab::WithFeatureCategory do + describe ".feature_category_for_action" do + let(:base_controller) do + Class.new do + include ::Gitlab::WithFeatureCategory + end + end + + let(:controller) do + Class.new(base_controller) do + feature_category :foo, %w(update edit) + feature_category :bar, %w(index show) + feature_category :quux, %w(destroy) + end + end + + let(:subclass) do + Class.new(controller) do + feature_category :baz, %w(subclass_index) + end + end + + it "is nil when nothing was defined" do + expect(base_controller.feature_category_for_action("hello")).to be_nil + end + + it "returns the expected category", :aggregate_failures do + expect(controller.feature_category_for_action("update")).to eq(:foo) + expect(controller.feature_category_for_action("index")).to eq(:bar) + expect(controller.feature_category_for_action("destroy")).to eq(:quux) + end + + it "returns the expected category for categories defined in subclasses" do + expect(subclass.feature_category_for_action("subclass_index")).to eq(:baz) + end + + it "raises an error when defining for the controller and for individual actions" do + expect do + Class.new(base_controller) do + feature_category :hello + feature_category :goodbye, [:world] + end + end.to raise_error(ArgumentError, "hello is defined for all actions, but other categories are set") + end + + it "raises an error when multiple calls define the same action" do + expect do + Class.new(base_controller) do + feature_category :hello, [:world] + feature_category :goodbye, ["world"] + end + end.to raise_error(ArgumentError, "Actions have multiple feature categories: world") + end + + it "does not raise an error when multiple calls define the same action and feature category" do + expect do + Class.new(base_controller) do + feature_category :hello, [:world] + feature_category :hello, ["world"] + end + end.not_to raise_error + end + end +end |