test/test_helper.rb in ohm-0.1.0.rc4 vs test/test_helper.rb in ohm-0.1.0.rc5

@@ -4,9 +4,50 @@
   require "ruby-debug"
 rescue LoadError
 end
 
 require "contest"
+
+def silence_warnings
+  original_verbose, $VERBOSE = $VERBOSE, nil
+  yield
+ensure
+  $VERBOSE = original_verbose
+end
+
+$VERBOSE = true
+
+class Logger
+  def self.current
+    Thread.current[:logger] ||= new
+  end
+
+  def initialize
+    clear
+  end
+
+  def clear
+    @lines = []
+  end
+
+  def debug(message)
+    @lines << message.to_s
+  end
+
+  def debug?; true; end
+  alias info  debug
+  alias warn  debug
+  alias error debug
+
+  def commands
+    @lines.map { |line| line[/Redis >> ([A-Z].+?)$/, 1] }.compact
+  end
+end
+
 require "ohm"
 
-Ohm.connect(:port => 6379, :db => 15, :timeout => 3)
-Ohm.flush
+class Test::Unit::TestCase
+  setup do
+    Ohm.redis = Redis.connect(:logger => Logger.current)
+    Ohm.flush
+  end
+end