Merge pull request #665 from clowder/patch-1

Update RSpec helper to support block syntax
splitrb · Nov 7, 2021 · 7872301 · 7872301
2 parents 8e4f7f8 + b65e5b2
commit 7872301
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -175,8 +175,10 @@ module SplitHelper
   #   use_ab_test(signup_form: "single_page", pricing: "show_enterprise_prices")
   #
   def use_ab_test(alternatives_by_experiment)
-    allow_any_instance_of(Split::Helper).to receive(:ab_test) do |_receiver, experiment|
-      alternatives_by_experiment.fetch(experiment) { |key| raise "Unknown experiment '#{key}'" }
+    allow_any_instance_of(Split::Helper).to receive(:ab_test) do |_receiver, experiment, &block|
+      variant = alternatives_by_experiment.fetch(experiment) { |key| raise "Unknown experiment '#{key}'" }
+      block.call(variant) unless block.nil?
+      variant
     end
   end
 end