test/logger_test.rb in mini_graphite-0.2.1 vs test/logger_test.rb in mini_graphite-0.3.7

@@ -1,10 +1,10 @@
 require "minitest/autorun"
 require "mocha/setup"
 require_relative "../lib/mini_graphite"
 
-class MiniGraphiteTest < MiniTest::Unit::TestCase
+class MiniGraphiteTest < MiniTest::Test
 
   def test_debug
     Kernel.expects(:puts).with(regexp_matches(/MESSAGE/))
     logger = Dalia::MiniGraphite::Logger.new
     logger.debug("MESSAGE")
@@ -15,12 +15,11 @@
     logger = Dalia::MiniGraphite::Logger.new(false)
     logger.debug("MESSAGE")
   end
 
   def test_debug_when_rails_actived
-    Kernel.const_set("Rails", mock(:logger => mock(:info)))
+    Rails.expects(:logger => mock(:info))
     logger = Dalia::MiniGraphite::Logger.new
     logger.debug("MESSAGE")
-    Kernel.send(:remove_const, :Rails)
   end
 
 end