Fix spec

bolshakov · Nov 6, 2014 · 6aa4abf · 6aa4abf
1 parent c297529
commit 6aa4abf
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/spec/stoplight/light_spec.rb b/spec/stoplight/light_spec.rb
@@ -236,16 +236,17 @@
       subject(:result) do
         begin
           light.run
-        rescue Stoplight::Error::RedLight
+        rescue ZeroDivisionError
           nil
         end
       end
 
+      let(:code_result) { 1 / 0 }
       let(:error_class) { HipChat::Unauthorized }
+      let(:threshold) { 1 }
 
       before do
-        Stoplight.data_store.set_state(
-          light.name, Stoplight::DataStore::STATE_LOCKED_RED)
+        light.with_threshold(threshold)
         allow(room).to receive(:send).with(
           'Stoplight',
           /\A@all /,