Allow strategy to be called on arbitrary classes

lib/flip/abstract_strategy.rb

@@ -1,6 +1,9 @@
 module Flip
   class AbstractStrategy
 
+    def initialize(_model_class = nil)
+    end
+
     def name
       self.class.name.split("::").last.gsub(/Strategy$/, "").underscore
     end

lib/flip/declarable.rb

@@ -12,7 +12,7 @@ def feature(key, options = {})
 
     # Adds a strategy for determining feature status.
     def strategy(strategy)
-      FeatureSet.instance.add_strategy strategy
+      FeatureSet.instance.add_strategy strategy, self
     end
 
     # The default response, boolean or a Proc to be called.

lib/flip/feature_set.rb

@@ -32,8 +32,8 @@ def << definition
     end
 
     # Adds a strategy for determing feature status.
-    def add_strategy(strategy)
-      strategy = strategy.new if strategy.is_a? Class
+    def add_strategy(strategy, klass = nil)
+      strategy = strategy.new(klass) if strategy.is_a? Class
       @strategies[strategy.name] = strategy
     end
 

spec/declarable_spec.rb

@@ -22,11 +22,26 @@
       it { should_not be_on(:one) }
       it { should be_on(:three) }
     end
+
     context "with default set to true" do
       before { model_class.send(:default, true) }
       it { should be_on(:one) }
       it { should be_on(:three) }
     end
   end
 
+  describe "the .strategy class method" do
+    let!(:model_class) do
+      Class.new do
+        extend Flip::Declarable
+      end
+    end
+
+    it "passes the class it's on" do
+      expect(Flip::DeclarationStrategy).to receive(:new)
+        .with(model_class)
+        .and_call_original
+      model_class.strategy Flip::DeclarationStrategy
+    end
+  end
 end