Add `Feature.enabled_and_licensed?` helper method

When no license is available (such as in CE), this is equivalent to
calling `enabled?`.

When a license is available but the specified feature isn't
license-gated, this is also equivalent to calling `enabled?`.

When a license is available and the specified feature is license-gated,
this method will check both that the feature is enabled, and that it's
available to the license.

1 files changed, 50 insertions, 0 deletions

diff --git a/spec/lib/feature_spec.rb b/spec/lib/feature_spec.rb
index 48c0ba8a653..6c29c1b1e2c 100644
--- a/spec/lib/feature_spec.rb
+++ b/spec/lib/feature_spec.rb
@@ -159,6 +159,56 @@ describe Feature do
     end
   end
 
+  describe '.enabled_and_licensed?' do
+    it 'passes all arguments to `enabled?`' do
+      expect(described_class).to receive(:enabled?)
+        .with(:foo, 'thing', default_enabled: true)
+        .and_return(false)
+
+      expect(described_class.enabled_and_licensed?(:foo, 'thing', default_enabled: true))
+        .to eq false
+    end
+
+    context 'with an enabled feature' do
+      before do
+        allow(described_class).to receive(:enabled?).and_return(true)
+      end
+
+      # Behavior changes slightly when we're in EE and License is available
+      if defined?(License)
+        context 'with an unregistered license feature' do
+          it 'returns true' do
+            expect(described_class.enabled_and_licensed?(:not_a_real_feature))
+              .to eq true
+          end
+        end
+
+        context 'with a registered license feature' do
+          it 'delegates to `License.feature_available?`' do
+            license = License.current
+
+            expect(license.features).to receive(:include?)
+              .with(:a_very_real_feature)
+              .and_return(true)
+            expect(license).to receive(:feature_available?)
+              .with(:a_very_real_feature)
+              .and_return(true)
+            expect(License).to receive(:current).and_return(license)
+
+            expect(described_class.enabled_and_licensed?(:a_very_real_feature))
+              .to eq true
+          end
+        end
+      else
+        context 'with no License definition' do
+          it 'returns true' do
+            expect(described_class.enabled_and_licensed?(:foo)).to eq true
+          end
+        end
+      end
+    end
+  end
+
   describe '.disable?' do
     it 'returns true for undefined feature' do
       expect(described_class.disabled?(:some_random_feature_flag)).to be_truthy