Add MessageLogger so we can look at logs later

Author: koppen

app/services/github_hook/message_logger.rb

@@ -0,0 +1,43 @@
+module GithubHook
+  class MessageLogger
+    attr_reader :messages, :wrapped_logger
+
+    def initialize(wrapped_logger = nil)
+      @messages = []
+      @wrapped_logger = wrapped_logger
+    end
+
+    def debug(message = yield)
+      add_message(:debug, message)
+    end
+
+    def error(message = yield)
+      add_message(:error, message)
+    end
+
+    def fatal(message = yield)
+      add_message(:fatal, message)
+    end
+
+    def info(message = yield)
+      add_message(:info, message)
+    end
+
+    def warn(message = yield)
+      add_message(:warn, message)
+    end
+
+    private
+
+    def add_message(level, message)
+      if wrapped_logger
+        wrapped_logger.send(level, message)
+      end
+
+      @messages << {
+        :level => level.to_s,
+        :message => message
+      }
+    end
+  end
+end

test/unit/github_hook/message_logger_test.rb

@@ -0,0 +1,49 @@
+# require 'test_helper'
+require "test/unit"
+require_relative "../../../app/services/github_hook/message_logger"
+
+class MessageLoggerTest < Test::Unit::TestCase
+  setup do
+    @logger = GithubHook::MessageLogger.new
+  end
+
+  def test_adds_messages_to_an_array
+    logger.info "Testing"
+    assert_equal [
+      {:level => "info", :message => "Testing"}
+    ], logger.messages
+  end
+
+  def test_supports_standard_log_levels
+    levels = ["fatal", "error", "warn", "info", "debug"]
+    levels.each do |level|
+      logger.public_send(level, level)
+    end
+    assert_equal levels, logger.messages.map { |m| m[:level] }
+  end
+
+  def test_supports_blocks
+    logger.debug { "This is my message" }
+    assert_equal [
+      {:level => "debug", :message => "This is my message"}
+    ], logger.messages
+  end
+
+  def test_logs_to_a_wrapped_logger_as_well
+    wrapped_logger = GithubHook::MessageLogger.new
+    logger = GithubHook::MessageLogger.new(wrapped_logger)
+    logger.debug "This goes everywhere"
+    assert_equal [
+      :level => "debug", :message => "This goes everywhere"
+    ], logger.messages
+    assert_equal [
+      :level => "debug", :message => "This goes everywhere"
+    ], wrapped_logger.messages
+  end
+
+  private
+
+  def logger
+    @logger ||= GithubHook::MessageLogger.new
+  end
+end