Merge pull request #296 from ekorneeff/metadata-helper

Push metadata to ab_test block

lib/split/helper.rb

@@ -29,11 +29,11 @@ def ab_test(metric_descriptor, control = nil, *alternatives)
 
       if block_given?
         if defined?(capture) # a block in a rails view
-          block = Proc.new { yield(alternative) }
+          block = Proc.new { yield(alternative, (trial.metadata if trial)) }
           concat(capture(alternative, &block))
           false
         else
-          yield(alternative)
+          yield(alternative, (trial.metadata if trial))
         end
       else
         alternative

lib/split/trial.rb

@@ -15,7 +15,7 @@ def initialize(attrs = {})
     end
 
     def metadata
-      @metadata ||= experiment.metadata[alternative.name]
+      @metadata ||= experiment.metadata[alternative.name] if experiment.metadata
     end
 
     def alternative

spec/helper_spec.rb

@@ -198,6 +198,26 @@
     end
   end
 
+  describe 'metadata' do
+    before do
+      Split.configuration.experiments = {
+        :my_experiment => {
+          :alternatives => ["one", "two"],
+          :resettable => false,
+          :metadata => { 'one' => 'Meta1', 'two' => 'Meta2' }
+        }
+      }
+    end
+
+    it 'should be passed to helper block' do
+      @params = {'my_experiment' => 'one'}
+      expect(ab_test('my_experiment')).to eq 'one'
+      expect(ab_test('my_experiment') do |alternative, meta|
+        meta
+      end).to eq('Meta1')
+    end
+  end
+
   describe 'finished' do
     before(:each) do
       @experiment_name = 'link_color'