fix specs

omkarmoghe · Apr 19, 2024 · 3a3a6d4 · 3a3a6d4
1 parent 3f692e8
commit 3a3a6d4
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/test/test_experiment.rb b/test/test_experiment.rb
@@ -47,13 +47,16 @@ def publish!(result)
   def setup
     @experiment = TestExperiment.new("test-experiment")
     @control = LabCoat::Observation.new("control", @experiment) do
-      @experiment.control(2)
+      @experiment.control
     end
   end
 
   def test_enabled?
-    refute(@experiment.enabled?(1))
-    assert(@experiment.enabled?(2))
+    @experiment.instance_variable_set(:@context, { num: 1 })
+    refute(@experiment.enabled?)
+    @experiment.instance_variable_set(:@context, { num: 2 })
+    assert(@experiment.enabled?)
+    @experiment.instance_variable_set(:@context, {})
   end
 
   def test_compare