Refactored MockLogger class.

diff --git a/test/TC_floodgate.rb b/test/TC_floodgate.rb
index 31a2b17..b1263db 100644 (file)
--- a/test/TC_floodgate.rb
+++ b/test/TC_floodgate.rb
@@ -4,37 +4,10 @@ require 'shogi_server'
 require 'shogi_server/player'
 require 'shogi_server/pairing'
 require 'shogi_server/league/floodgate'
+require 'test/mock_log_message'
 
 $topdir = File.expand_path File.dirname(__FILE__)
 
-class MockLogger
-  def debug(str)
-    #puts str
-  end
-  def info(str)
-    #puts str
-  end
-  def warn(str)
-    puts str
-  end
-  def error(str)
-    puts str
-  end
-end
-
-$logger = MockLogger.new
-def log_message(msg)
-  $logger.info(msg)
-end
-
-def log_warning(msg)
-  $logger.warn(msg)
-end
-
-def log_error(msg)
-  $logger.error(msg)
-end
-
 class TestFloodgate < Test::Unit::TestCase
   def setup
     @fg = ShogiServer::League::Floodgate.new(nil)

diff --git a/test/TC_pairing.rb b/test/TC_pairing.rb
index 0f8e7c4..30e9546 100644 (file)
--- a/test/TC_pairing.rb
+++ b/test/TC_pairing.rb
@@ -3,27 +3,8 @@ require 'test/unit'
 require 'shogi_server'
 require 'shogi_server/player'
 require 'shogi_server/pairing'
+require 'test/mock_log_message'
 
-class MockLogger
-  def debug(str)
-  end
-  def info(str)
-    #puts str
-  end
-  def warn(str)
-  end
-  def error(str)
-  end
-end
-
-$logger = MockLogger.new
-def log_message(msg)
-  $logger.info(msg)
-end
-
-def log_warning(msg)
-  $logger.warn(msg)
-end
 
 def same_pair?(a, b)
   unless a.size == 2 && b.size == 2

diff --git a/test/mock_log_message.rb b/test/mock_log_message.rb
index e2e1e1d..5ffeb84 100644 (file)
--- a/test/mock_log_message.rb
+++ b/test/mock_log_message.rb
@@ -1,17 +1,32 @@
-def log_info(str)
-  $stderr.puts str
+class MockLogger
+  def debug(str)
+    #puts str
+  end
+  def info(str)
+    #puts str
+  end
+  def warn(str)
+    puts str
+  end
+  def error(str)
+    puts str
+  end
 end
 
-def log_message(str)
-  $stderr.puts str
+$logger = MockLogger.new
+def log_message(msg)
+  $logger.info(msg)
 end
 
-def log_warning(str)
-  $stderr.puts str
+def log_warning(msg)
+  $logger.warn(msg)
 end
 
-def log_error(str)
-  $stderr.puts str
+def log_error(msg)
+  $logger.error(msg)
 end
 
+def log_info(msg)
+  $logger.info(msg)
+end