test/bbqueue/test_logger.rb in bbqueue-0.0.1 vs test/bbqueue/test_logger.rb in bbqueue-0.0.2

@@ -1,31 +1,40 @@
 
+require "logger"
+
 class BBQueue::TestLogger
+  attr_accessor :tempfile, :logger
+
   def initialize
-    @messages = []
+    self.tempfile = Tempfile.new("logger")
+
+    logger = Logger.new(tempfile.path)
+    logger.formatter = proc { |severity, datetime, progname, msg| "#{severity.downcase}: #{msg}\n" }
+
+    self.logger = logger
   end
 
-  def debug(object)
-    @messages.push object
+  def debug(*args)
+    logger.debug(*args)
   end
 
-  def info(object)
-    @messages.push object
+  def info(*args)
+    logger.info(*args)
   end
 
-  def warn(object)
-    @messages.push object
+  def warn(*args)
+    logger.warn(*args)
   end
 
-  def error(object)
-    @messages.push object
+  def error(*args)
+    logger.error(*args)
   end
 
-  def fatal(object)
-    @messages.push object
+  def fatal(*args)
+    logger.fatal(*args)
   end
 
-  def count
-    @messages.size
+  def count(regex = /.*/)
+    File.read(tempfile.path).lines.grep(regex).count
   end
 end