Better logger

Author: Suszyński Krzysztof

lib/puppet_forge_server/logger.rb

@@ -16,6 +16,7 @@
 
 
 require 'logger'
+require 'logger/colors'
 
 module PuppetForgeServer
   class Logger
@@ -45,7 +46,13 @@ def method_missing (method_name, *args, &block)
                       else
                         method_name
                     end
-      @loggers.each { |logger| logger.send(method_name, args.first) }
+      if args.size > 0
+        # setters
+        @loggers.each { |logger| logger.send(method_name, args.first) }
+      else
+        # getters
+        @loggers.collect { |logger| logger.send(method_name) }
+      end
     end
 
     def respond_to?(method_name, include_private = false)
@@ -70,4 +77,4 @@ def set(loggers= {})
       end
     end
   end
-end
+end

puppet-forge-server.gemspec

@@ -47,6 +47,7 @@ Gem::Specification.new do |spec|
   spec.add_dependency 'redcarpet', '~> 3.3.0'
   spec.add_dependency 'lrucache', '~> 0.1.4'
   spec.add_dependency 'iconv', '~> 1.0.4'
+  spec.add_dependency 'logger-colors', '~> 1.0'
 
   spec.add_development_dependency 'rake', '~> 10.3'
   spec.add_development_dependency 'rspec', '~> 3.1'

spec/unit/http/http_client_spec.rb

@@ -29,10 +29,13 @@
         socket.close
       end
     end
+    @original_level = PuppetForgeServer::Logger.get.level.first
+    PuppetForgeServer::Logger.get.level = Logger::WARN
   end
   after(:each) do
     @server.close
     @thr.kill
+    PuppetForgeServer::Logger.get.level = @original_level
   end
   let(:instance) { described_class.new(cache) }
   describe '#download' do