refactor

Unleash · Jul 4, 2023 · 000d92c · 000d92c
1 parent f2856ac
commit 000d92c
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/lib/unleash/strategy/flexible_rollout.rb b/lib/unleash/strategy/flexible_rollout.rb
@@ -12,7 +12,7 @@ def is_enabled?(params = {}, context = nil)
         return false unless params.is_a?(Hash)
 
         stickiness = params.fetch('stickiness', 'default')
-        return false unless context.instance_of?(Unleash::Context) || ['random', 'default'].include?(stickiness)
+        return false unless context_is_sufficient?(stickiness, context)
 
         stickiness_id = resolve_stickiness(stickiness, context)
 
@@ -33,6 +33,12 @@ def is_enabled?(params = {}, context = nil)
 
       private
 
+      def context_is_sufficient?(stickiness, context)
+        return true if ['random', 'default'].include?(stickiness)
+
+        context.instance_of?(Unleash::Context)
+      end
+
       def random
         Random.rand(0..100)
       end