another test

Author: shivnagarajan

test/test_middleware.rb

@@ -108,4 +108,19 @@ def test_enabled_should_override_mode_if_a_proc
     middleware.call(env)
     assert proc_called
   end
+
+  def test_saviour_should_be_called_when_enabled_with_env
+    proc_called = false
+    env_set = nil
+    results_received = nil
+    middleware = StackProf::Middleware.new(proc {|env| proc_called = true}, enabled: Proc.new{ [true, 'foo'] }, saviour: Proc.new{ |env, results| env_set = env['FOO'] ; results_received = results[:mode] }, save_every: 1)
+    StackProf.expects(:start).with({mode: 'foo', interval: StackProf::Middleware.interval, raw: false})
+    StackProf.expects(:stop)
+    StackProf.stubs(:results).returns({ mode: 'foo' })
+
+    middleware.call({ 'FOO' => 'bar' })
+    assert proc_called
+    assert_equal env_set, 'bar'
+    assert_equal results_received, 'foo'
+  end
 end