Test fixes - happy within suite.

librato · Aug 25, 2012 · 02df1a9 · 02df1a9
1 parent 6ba7b48
commit 02df1a9
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/test/unit/middleware_test.rb b/test/unit/middleware_test.rb
@@ -3,14 +3,23 @@
 require 'metrics/rails'
 
 class MetricRackMiddlewareTest < MiniTest::Unit::TestCase
+  include Mocha::API
+
   def setup
     Metrics::Rails.stubs(:forking_server?).returns(false)
     Metrics::Rails.stubs(:measure).returns(true)
     Metrics::Rails.stubs(:increment).returns(true)
 
-    Time.stubs(:now).returns(0.1305, 0.20075)
+    Time.stubs(:now).returns(Time.at(0.1305), Time.at(0.20075))
+
+    @middleware = Metrics::Rack::Middleware.new(
+      stub(:call => [200, {}, []]), Metrics::Rails
+    )
+  end
 
-    @middleware = Metrics::Rack::Middleware.new stub(:call => [200, {}, []])
+  def teardown
+    Metrics::Rails.unstub :forking_server?, :measure, :increment
+    Time.unstub :now
   end
 
   def test_logs_header_metrics